Add: hint "session-balancer"

Author: LiamHamsters

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

@@ -236,6 +236,18 @@ public int MaxReceiveMessageSize
 
     private int _maxReceiveMessageSize;
 
+    public bool DisableServerBalancer
+    {  
+        get => _disableServerBalancer;
+        set
+        {
+            _disableServerBalancer = value;
+            SaveValue(nameof(DisableServerBalancer), value);
+        }
+    }
+
+    private bool _disableServerBalancer;
+
     public bool DisableDiscovery
     {
         get => _disableDiscovery;
@@ -334,7 +346,8 @@ internal async Task<IDriver> BuildDriver()
             Password = Password,
             EnableMultipleHttp2Connections = EnableMultipleHttp2Connections,
             MaxSendMessageSize = MaxSendMessageSize,
-            MaxReceiveMessageSize = MaxReceiveMessageSize
+            MaxReceiveMessageSize = MaxReceiveMessageSize,
+            DisableServerBalancer = DisableServerBalancer
         };
         var loggerFactory = LoggerFactory ?? NullLoggerFactory.Instance;
 
@@ -433,6 +446,9 @@ static YdbConnectionOption()
             AddOption(new YdbConnectionOption<int>(IntExtractor,
                     (builder, createSessionTimeout) => builder.CreateSessionTimeout = createSessionTimeout),
                 "CreateSessionTimeout", "Create Session Timeout");
+            AddOption(new YdbConnectionOption<bool>(BoolExtractor, (builder, disableServerBalancer) =>
+                    builder.DisableServerBalancer = disableServerBalancer),
+                "DisableServerBalancer", "Disable Server Balancer");
         }
 
         private static void AddOption(YdbConnectionOption option, params string[] keys)

src/Ydb.Sdk/src/DriverConfig.cs

@@ -26,6 +26,7 @@ public class DriverConfig
 
     public int MaxSendMessageSize { get; init; } = GrpcDefaultSettings.MaxSendMessageSize;
     public int MaxReceiveMessageSize { get; init; } = GrpcDefaultSettings.MaxReceiveMessageSize;
+    public bool DisableServerBalancer { get; init; } = GrpcDefaultSettings.DisableServerBalancer;
 
     internal X509Certificate2Collection CustomServerCertificates { get; } = new();
     internal TimeSpan EndpointDiscoveryInterval = TimeSpan.FromMinutes(1);

src/Ydb.Sdk/src/GrpcDefaultSettings.cs

@@ -21,4 +21,6 @@ internal static class GrpcDefaultSettings
     internal const bool EnableMultipleHttp2Connections = false;
 
     internal const bool DisableDiscovery = false;
+
+    internal const bool DisableServerBalancer = false;
 }

src/Ydb.Sdk/src/IDriver.cs

@@ -182,6 +182,11 @@ protected async ValueTask<CallOptions> GetCallOptions(GrpcRequestSettings settin
             meta.Add(Metadata.RpcTraceIdHeader, settings.TraceId);
         }
 
+        if (!Config.DisableServerBalancer)
+        {
+            meta.Add(Metadata.RpcClientCapabilitiesHeader, "session-balancer");
+        }
+        
         var options = new CallOptions(headers: meta, cancellationToken: settings.CancellationToken);
 
         if (settings.TransportTimeout != TimeSpan.Zero)

src/Ydb.Sdk/src/Metadata.cs

@@ -8,6 +8,7 @@ internal static class Metadata
     public const string RpcTraceIdHeader = "x-ydb-trace-id";
     public const string RpcSdkInfoHeader = "x-ydb-sdk-build-info";
     public const string RpcServerHintsHeader = "x-ydb-server-hints";
+    public const string RpcClientCapabilitiesHeader = "x-ydb-client-capabilities";
 
     public const string GracefulShutdownHint = "session-close";
 }

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

@@ -24,6 +24,7 @@ public void InitDefaultValues_WhenEmptyConstructorInvoke_ReturnDefaultConnection
         Assert.Equal(64 * 1024 * 1024, ydbConnectionStringBuilder.MaxSendMessageSize);
         Assert.Equal(64 * 1024 * 1024, ydbConnectionStringBuilder.MaxReceiveMessageSize);
         Assert.False(ydbConnectionStringBuilder.DisableDiscovery);
+        Assert.False(ydbConnectionStringBuilder.DisableServerBalancer);
         Assert.Equal(5, ydbConnectionStringBuilder.CreateSessionTimeout);
     }
 
@@ -45,7 +46,7 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
             "ConnectTimeout=30;KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
             "EnableMultipleHttp2Connections=true;CreateSessionTimeout=30;" +
             "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000;" +
-            "DisableDiscovery=true"
+            "DisableDiscovery=true;DisableServerBalancer=true"
         );
 
         Assert.Equal(2135, connectionString.Port);
@@ -64,8 +65,9 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
                      "ConnectTimeout=30;KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
                      "EnableMultipleHttp2Connections=True;CreateSessionTimeout=30;" +
                      "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000;" +
-                     "DisableDiscovery=True", connectionString.ConnectionString);
+                     "DisableDiscovery=True;DisableServerBalancer=True", connectionString.ConnectionString);
         Assert.True(connectionString.DisableDiscovery);
+        Assert.True(connectionString.DisableServerBalancer);
         Assert.Equal(30, connectionString.CreateSessionTimeout);
     }