feat: max send/receive message size settings

Author: KirillKurdyukov

src/Ydb.Sdk/src/Ado/YdbConnectionStringBuilder.cs

@@ -30,6 +30,8 @@ private void InitDefaultValues()
         _keepAlivePingDelay = SocketHttpHandlerDefaults.DefaultKeepAlivePingSeconds;
         _keepAlivePingTimeout = SocketHttpHandlerDefaults.DefaultKeepAlivePingTimeoutSeconds;
         _enableMultipleHttp2Connections = false;
+        _maxSendMessageSize = SocketHttpHandlerDefaults.MaxSendMessageSize;
+        _maxReceiveMessageSize = SocketHttpHandlerDefaults.MaxReceiveMessageSize;
     }
 
     public string Host
@@ -187,6 +189,30 @@ public bool EnableMultipleHttp2Connections
 
     private bool _enableMultipleHttp2Connections;
 
+    public int MaxSendMessageSize
+    {
+        get => _maxSendMessageSize;
+        set
+        {
+            _maxSendMessageSize = value;
+            SaveValue(nameof(MaxSendMessageSize), value);
+        }
+    }
+
+    private int _maxSendMessageSize;
+
+    public int MaxReceiveMessageSize
+    {
+        get => _maxReceiveMessageSize;
+        set
+        {
+            _maxReceiveMessageSize = value;
+            SaveValue(nameof(MaxReceiveMessageSize), value);
+        }
+    }
+
+    private int _maxReceiveMessageSize;
+
     public ILoggerFactory? LoggerFactory { get; init; }
 
     public ICredentialsProvider? CredentialsProvider { get; init; }
@@ -235,24 +261,29 @@ internal Task<Driver> BuildDriver()
     {
         var cert = RootCertificate != null ? X509Certificate.CreateFromCertFile(RootCertificate) : null;
 
-        return Driver.CreateInitialized(new DriverConfig(
-            endpoint: Endpoint,
-            database: Database,
-            credentials: CredentialsProvider,
-            customServerCertificate: cert,
-            customServerCertificates: ServerCertificates
-        )
-        {
-            KeepAlivePingDelay = KeepAlivePingDelay == 0
-                ? Timeout.InfiniteTimeSpan
-                : TimeSpan.FromSeconds(KeepAlivePingDelay),
-            KeepAlivePingTimeout = KeepAlivePingTimeout == 0
-                ? Timeout.InfiniteTimeSpan
-                : TimeSpan.FromSeconds(KeepAlivePingTimeout),
-            User = User,
-            Password = Password,
-            EnableMultipleHttp2Connections = EnableMultipleHttp2Connections
-        }, LoggerFactory);
+        return Driver.CreateInitialized(
+            new DriverConfig(
+                endpoint: Endpoint,
+                database: Database,
+                credentials: CredentialsProvider,
+                customServerCertificate: cert,
+                customServerCertificates: ServerCertificates
+            )
+            {
+                KeepAlivePingDelay = KeepAlivePingDelay == 0
+                    ? Timeout.InfiniteTimeSpan
+                    : TimeSpan.FromSeconds(KeepAlivePingDelay),
+                KeepAlivePingTimeout = KeepAlivePingTimeout == 0
+                    ? Timeout.InfiniteTimeSpan
+                    : TimeSpan.FromSeconds(KeepAlivePingTimeout),
+                User = User,
+                Password = Password,
+                EnableMultipleHttp2Connections = EnableMultipleHttp2Connections,
+                MaxSendMessageSize = MaxSendMessageSize,
+                MaxReceiveMessageSize = MaxReceiveMessageSize
+            },
+            LoggerFactory
+        );
     }
 
     public override void Clear()
@@ -333,6 +364,11 @@ static YdbConnectionOption()
             AddOption(new YdbConnectionOption<bool>(BoolExtractor, (builder, enableMultipleHttp2Connections) =>
                     builder.EnableMultipleHttp2Connections = enableMultipleHttp2Connections),
                 "EnableMultipleHttp2Connections", "Enable Multiple Http2 Connections");
+            AddOption(new YdbConnectionOption<int>(IntExtractor, (builder, maxSendMessageSize) =>
+                builder.MaxSendMessageSize = maxSendMessageSize), "MaxSendMessageSize", "Max Send Message Size");
+            AddOption(new YdbConnectionOption<int>(IntExtractor, (builder, maxReceiveMessageSize) =>
+                    builder.MaxReceiveMessageSize = maxReceiveMessageSize),
+                "MaxReceiveMessageSize", "Max Receive Message Size");
         }
 
         private static void AddOption(YdbConnectionOption option, params string[] keys)

src/Ydb.Sdk/src/DriverConfig.cs

@@ -16,11 +16,14 @@ public class DriverConfig
     public TimeSpan KeepAlivePingTimeout { get; init; } =
         TimeSpan.FromSeconds(SocketHttpHandlerDefaults.DefaultKeepAlivePingTimeoutSeconds);
 
-    public bool EnableMultipleHttp2Connections { get; init; }
-
     public string? User { get; init; }
     public string? Password { get; init; }
 
+    public bool EnableMultipleHttp2Connections { get; init; }
+
+    public int MaxSendMessageSize { get; init; } = SocketHttpHandlerDefaults.MaxSendMessageSize;
+    public int MaxReceiveMessageSize { get; init; } = SocketHttpHandlerDefaults.MaxReceiveMessageSize;
+
     internal X509Certificate2Collection CustomServerCertificates { get; } = new();
     internal TimeSpan EndpointDiscoveryInterval = TimeSpan.FromMinutes(1);
     internal TimeSpan EndpointDiscoveryTimeout = TimeSpan.FromSeconds(10);

src/Ydb.Sdk/src/Pool/ChannelPool.cs

@@ -95,7 +95,9 @@ public GrpcChannel CreateChannel(string endpoint)
         var channelOptions = new GrpcChannelOptions
         {
             LoggerFactory = _loggerFactory,
-            DisposeHttpClient = true
+            DisposeHttpClient = true,
+            MaxSendMessageSize = _config.MaxSendMessageSize,
+            MaxReceiveMessageSize = _config.MaxReceiveMessageSize
         };
 
         var httpHandler = new SocketsHttpHandler

src/Ydb.Sdk/src/SocketHttpHandlerDefaultSettings.cs

@@ -11,4 +11,8 @@ internal static class SocketHttpHandlerDefaults
     /// Default timeout (in seconds) for receiving a response to a keep-alive ping.
     /// </summary>
     internal const int DefaultKeepAlivePingTimeoutSeconds = 10;
+    
+    internal const int MaxSendMessageSize = 64 * 1024 * 1024; // 64 Mb
+    
+    internal const int MaxReceiveMessageSize = 64 * 1024 * 1024; // 64 Mb
 }

src/Ydb.Sdk/tests/Ado/YdbConnectionStringBuilderTests.cs

@@ -8,18 +8,20 @@ public class YdbConnectionStringBuilderTests
     [Fact]
     public void InitDefaultValues_WhenEmptyConstructorInvoke_ReturnDefaultConnectionString()
     {
-        var connectionString = new YdbConnectionStringBuilder();
+        var ydbConnectionStringBuilder = new YdbConnectionStringBuilder();
 
-        Assert.Equal(2136, connectionString.Port);
-        Assert.Equal("localhost", connectionString.Host);
-        Assert.Equal("/local", connectionString.Database);
-        Assert.Equal(100, connectionString.MaxSessionPool);
-        Assert.Null(connectionString.User);
-        Assert.Null(connectionString.Password);
-        Assert.Equal(10, connectionString.KeepAlivePingDelay);
-        Assert.Equal(10, connectionString.KeepAlivePingTimeout);
-        Assert.Equal("", connectionString.ConnectionString);
-        Assert.False(connectionString.EnableMultipleHttp2Connections);
+        Assert.Equal(2136, ydbConnectionStringBuilder.Port);
+        Assert.Equal("localhost", ydbConnectionStringBuilder.Host);
+        Assert.Equal("/local", ydbConnectionStringBuilder.Database);
+        Assert.Equal(100, ydbConnectionStringBuilder.MaxSessionPool);
+        Assert.Null(ydbConnectionStringBuilder.User);
+        Assert.Null(ydbConnectionStringBuilder.Password);
+        Assert.Equal(10, ydbConnectionStringBuilder.KeepAlivePingDelay);
+        Assert.Equal(10, ydbConnectionStringBuilder.KeepAlivePingTimeout);
+        Assert.Equal("", ydbConnectionStringBuilder.ConnectionString);
+        Assert.False(ydbConnectionStringBuilder.EnableMultipleHttp2Connections);
+        Assert.Equal(64 * 1024 * 1024, ydbConnectionStringBuilder.MaxSendMessageSize);
+        Assert.Equal(64 * 1024 * 1024, ydbConnectionStringBuilder.MaxReceiveMessageSize);
     }
 
     [Fact]
@@ -38,7 +40,8 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
         var connectionString =
             new YdbConnectionStringBuilder("Host=server;Port=2135;Database=/my/path;User=Kirill;UseTls=true;" +
                                            "KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
-                                           "EnableMultipleHttp2Connections=true");
+                                           "EnableMultipleHttp2Connections=true;" +
+                                           "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000");
 
         Assert.Equal(2135, connectionString.Port);
         Assert.Equal("server", connectionString.Host);
@@ -49,9 +52,12 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
         Assert.Equal(60, connectionString.KeepAlivePingTimeout);
         Assert.Null(connectionString.Password);
         Assert.True(connectionString.EnableMultipleHttp2Connections);
+        Assert.Equal(1000000, connectionString.MaxSendMessageSize);
+        Assert.Equal(1000000, connectionString.MaxReceiveMessageSize);
         Assert.Equal("Host=server;Port=2135;Database=/my/path;User=Kirill;UseTls=True;" +
                      "KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
-                     "EnableMultipleHttp2Connections=True", connectionString.ConnectionString);
+                     "EnableMultipleHttp2Connections=True;" +
+                     "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000", connectionString.ConnectionString);
     }
 
     [Fact]