add MaxAIOContextEvents parameter to DiskAgent config (#2107) (#2111)

ydb-platform · Sep 23, 2024 · 8a896fd · 8a896fd
1 parent b7ecfb4
commit 8a896fd
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/cloud/blockstore/config/disk.proto b/cloud/blockstore/config/disk.proto
@@ -248,6 +248,9 @@ message TDiskAgentConfig
     // NodeBroker. Enabling this option completely disables them if there were
     // no devices found.
     optional bool DisableNodeBrokerRegisterationOnDevicelessAgent = 32;
+
+    // Max. count of concurrently processing operations (io_setup).
+    optional uint32 MaxAIOContextEvents = 33;
 }
 
 ////////////////////////////////////////////////////////////////////////////////

diff --git a/cloud/blockstore/libs/disk_agent/bootstrap.cpp b/cloud/blockstore/libs/disk_agent/bootstrap.cpp
@@ -429,7 +429,8 @@ bool TBootstrap::InitKikimrService()
             }
 
             case NProto::DISK_AGENT_BACKEND_AIO:
-                FileIOService = CreateAIOService();
+                FileIOService =
+                    CreateAIOService(config.GetMaxAIOContextEvents());
                 NvmeManager = CreateNvmeManager(config.GetSecureEraseTimeout());
 
                 AioStorageProvider = CreateAioStorageProvider(

diff --git a/cloud/blockstore/libs/storage/disk_agent/model/config.cpp b/cloud/blockstore/libs/storage/disk_agent/model/config.cpp
@@ -42,6 +42,7 @@ namespace {
     xxx(IOParserActorCount,                 ui32,       0                     )\
     xxx(OffloadAllIORequestsParsingEnabled, bool,       false                 )\
     xxx(DisableNodeBrokerRegisterationOnDevicelessAgent, bool,         false  )\
+    xxx(MaxAIOContextEvents,                ui32,       1024                  )\
 // BLOCKSTORE_AGENT_CONFIG
 
 #define BLOCKSTORE_DECLARE_CONFIG(name, type, value)                           \

diff --git a/cloud/blockstore/libs/storage/disk_agent/model/config.h b/cloud/blockstore/libs/storage/disk_agent/model/config.h
@@ -111,6 +111,7 @@ class TDiskAgentConfig
     ui32 GetIOParserActorCount() const;
     bool GetOffloadAllIORequestsParsingEnabled() const;
     bool GetDisableNodeBrokerRegisterationOnDevicelessAgent() const;
+    ui32 GetMaxAIOContextEvents() const;
 
     void Dump(IOutputStream& out) const;
     void DumpHtml(IOutputStream& out) const;