last fix

Author: LiamHamsters

src/Ydb.Sdk/src/Ado/BulkUpsert/BulkUpsertImporter.cs

@@ -17,19 +17,18 @@ public sealed class BulkUpsertImporter : IBulkUpsertImporter
     private StructType? _structType;
     private int _currentBytes;
 
-    public BulkUpsertImporter(
+    internal BulkUpsertImporter(
         IDriver driver,
         string tableName,
         IReadOnlyList<string> columns,
-        CancellationToken cancellationToken = default,
-        int maxBytes = 64 * 1024 * 1024)
+        int maxBytes,
+        CancellationToken cancellationToken = default)
     {
         _driver = driver;
         _tablePath = tableName;
         _columns = columns;
-        _maxBytes = maxBytes;
-        _cancellationToken = cancellationToken;
         _maxBytes = maxBytes / 2;
+        _cancellationToken = cancellationToken;
     }
 
     public async ValueTask AddRowAsync(object?[] values)

src/Ydb.Sdk/src/Ado/YdbConnection.cs

@@ -56,8 +56,7 @@ public YdbConnection(YdbConnectionStringBuilder connectionStringBuilder)
     public IBulkUpsertImporter BeginBulkUpsertImport(
         string name,
         IReadOnlyList<string> columns,
-        CancellationToken cancellationToken = default
-    )
+        CancellationToken cancellationToken = default)
     {
         ThrowIfConnectionClosed();
         if (CurrentTransaction is { Completed: false })
@@ -66,7 +65,9 @@ public IBulkUpsertImporter BeginBulkUpsertImport(
         var database = ConnectionStringBuilder.Database.TrimEnd('/');
         var tablePath = string.IsNullOrEmpty(database) ? name : $"{database}/{name}";
 
-        return new BulkUpsertImporter(Session.Driver, tablePath, columns, cancellationToken);
+        var maxBytes = ConnectionStringBuilder.BulkUpsertMaxBytes;
+
+        return new BulkUpsertImporter(Session.Driver, tablePath, columns, maxBytes, cancellationToken);
     }
 
     protected override YdbTransaction BeginDbTransaction(IsolationLevel isolationLevel)

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

@@ -42,6 +42,7 @@ private void InitDefaultValues()
         _maxReceiveMessageSize = GrpcDefaultSettings.MaxReceiveMessageSize;
         _disableDiscovery = GrpcDefaultSettings.DisableDiscovery;
         _disableServerBalancer = false;
+        _bulkUpsertMaxBytes = 64 * 1024 * 1024;
     }
 
     public string Host
@@ -291,6 +292,20 @@ public int MaxReceiveMessageSize
     }
 
     private int _maxReceiveMessageSize;
+    
+    private int _bulkUpsertMaxBytes;
+    
+    public int BulkUpsertMaxBytes
+    {
+        get => _bulkUpsertMaxBytes;
+        set
+        {
+            if (value <= 0)
+                throw new ArgumentOutOfRangeException(nameof(value), value, "BulkUpsertMaxBytes must be positive.");
+            _bulkUpsertMaxBytes = value;
+            SaveValue(nameof(BulkUpsertMaxBytes), value);
+        }
+    }
 
     public bool DisableServerBalancer
     {