Implementation for GetCurrentBatchSize.

Author: cincuranet

src/FirebirdSql.Data.FirebirdClient/Client/Managed/Version16/GdsBatch.cs

@@ -143,14 +143,25 @@ public override async ValueTask<ExecuteResultItem[]> ExecuteAsync(int count, IDe
 		}
 	}
 
-	public override void Dispose2()
+	public override int ComputeBatchSize(int count, IDescriptorFiller descriptorFiller)
+	{
+		var parametersData = GetParametersData(count, descriptorFiller);
+		return parametersData.Sum(x => x.Length);
+	}
+	public override async ValueTask<int> ComputeBatchSizeAsync(int count, IDescriptorFiller descriptorFiller, CancellationToken cancellationToken = default)
+	{
+		var parametersData = await GetParametersDataAsync(count, descriptorFiller, cancellationToken).ConfigureAwait(false);
+		return parametersData.Sum(x => x.Length);
+	}
+
+	public override void Release()
 	{
 		Database.Xdr.Write(IscCodes.op_batch_rls);
 		Database.Xdr.Write(_statement.Handle);
 		Database.AppendDeferredPacket(ProcessReleaseResponse);
 	}
 
-	public override async ValueTask Dispose2Async(CancellationToken cancellationToken = default)
+	public override async ValueTask ReleaseAsync(CancellationToken cancellationToken = default)
 	{
 		await Database.Xdr.WriteAsync(IscCodes.op_batch_rls, cancellationToken).ConfigureAwait(false);
 		await Database.Xdr.WriteAsync(_statement.Handle, cancellationToken).ConfigureAwait(false);

src/FirebirdSql.Data.FirebirdClient/Common/BatchBase.cs

@@ -35,10 +35,9 @@ public class ExecuteResultItem
 	public abstract ExecuteResultItem[] Execute(int count, IDescriptorFiller descriptorFiller);
 	public abstract ValueTask<ExecuteResultItem[]> ExecuteAsync(int count, IDescriptorFiller descriptorFiller, CancellationToken cancellationToken = default);
 
-	public virtual void Dispose2()
-	{ }
-	public virtual ValueTask Dispose2Async(CancellationToken cancellationToken = default)
-	{
-		return ValueTask2.CompletedTask;
-	}
+	public abstract int ComputeBatchSize(int count, IDescriptorFiller descriptorFiller);
+	public abstract ValueTask<int> ComputeBatchSizeAsync(int count, IDescriptorFiller descriptorFiller, CancellationToken cancellationToken = default);
+
+	public abstract void Release();
+	public abstract ValueTask ReleaseAsync(CancellationToken cancellationToken = default);
 }

src/FirebirdSql.Data.FirebirdClient/FirebirdClient/FbBatchCommand.cs

@@ -43,6 +43,7 @@ public sealed class FbBatchCommand : IFbPreparedCommand, IDescriptorFiller, IDis
 	private FbTransaction _transaction;
 	private FbBatchParameterCollection _batchParameters;
 	private StatementBase _statement;
+	private BatchBase _batch;
 	//private FbDataReader _activeReader;
 	private IReadOnlyList<string> _namedParameters;
 	private string _commandText;
@@ -291,6 +292,7 @@ public void Dispose()
 			_connection = null;
 			_transaction = null;
 			_batchParameters = null;
+			_batch = null;
 			_statement = null;
 			//_activeReader = null;
 			_namedParameters = null;
@@ -319,6 +321,7 @@ public async ValueTask DisposeAsync()
 			_connection = null;
 			_transaction = null;
 			_batchParameters = null;
+			_batch = null;
 			_statement = null;
 			//_activeReader = null;
 			_namedParameters = null;
@@ -549,6 +552,23 @@ public Task<string> GetCommandExplainedPlanAsync(CancellationToken cancellationT
 		return _statement.GetExecutionExplainedPlanAsync(cancellationToken).AsTask();
 	}
 
+	public int GetCurrentBatchSize()
+	{
+		if (_batch == null)
+		{
+			throw new InvalidOperationException("Batch must be prepared.");
+		}
+		return _batch.ComputeBatchSize(_batchParameters.Count, this);
+	}
+	public async Task<int> GetCurrentBatchSizeAsync(CancellationToken cancellationToken = default)
+	{
+		if (_batch == null)
+		{
+			throw new InvalidOperationException("Batch must be prepared.");
+		}
+		return await _batch.ComputeBatchSizeAsync(_batchParameters.Count, this, cancellationToken).ConfigureAwait(false);
+	}
+
 	#endregion
 
 	#region Internal Methods
@@ -855,6 +875,8 @@ internal void Release()
 			_connection.InnerConnection.RemovePreparedCommand(this);
 		}
 
+		_batch = null;
+
 		if (_statement != null)
 		{
 			_statement.Dispose2();
@@ -873,6 +895,8 @@ internal async Task ReleaseAsync(CancellationToken cancellationToken = default)
 			_connection.InnerConnection.RemovePreparedCommand(this);
 		}
 
+		_batch = null;
+
 		if (_statement != null)
 		{
 			await _statement.Dispose2Async(cancellationToken).ConfigureAwait(false);
@@ -1131,6 +1155,7 @@ private void Prepare(bool returnsSet)
 		if (_statement == null)
 		{
 			_statement = innerConn.Database.CreateStatement(_transaction.Transaction);
+			_batch = _statement.CreateBatch();
 		}
 
 		// Prepare the statement if	needed
@@ -1150,6 +1175,7 @@ private void Prepare(bool returnsSet)
 			}
 			catch
 			{
+				_batch = null;
 				// Release the statement and rethrow the exception
 				_statement.Release();
 				_statement = null;
@@ -1195,6 +1221,7 @@ private async Task PrepareAsync(bool returnsSet, CancellationToken cancellationT
 		if (_statement == null)
 		{
 			_statement = innerConn.Database.CreateStatement(_transaction.Transaction);
+			_batch = _statement.CreateBatch();
 		}
 
 		// Prepare the statement if	needed
@@ -1214,6 +1241,7 @@ private async Task PrepareAsync(bool returnsSet, CancellationToken cancellationT
 			}
 			catch
 			{
+				_batch = null;
 				// Release the statement and rethrow the exception
 				await _statement.ReleaseAsync(cancellationToken).ConfigureAwait(false);
 				_statement = null;
@@ -1249,17 +1277,16 @@ private FbBatchNonQueryResult ExecuteCommand(bool returnsSet)
 			throw FbException.Create("Must declare command parameters.");
 		}
 
-		var batch = _statement.CreateBatch();
 		try
 		{
-			batch.MultiError = MultiError;
-			batch.BatchBufferSize = BatchBufferSize;
+			_batch.MultiError = MultiError;
+			_batch.BatchBufferSize = BatchBufferSize;
 			// Execute
-			return new FbBatchNonQueryResult(batch.Execute(_batchParameters.Count, this));
+			return new FbBatchNonQueryResult(_batch.Execute(_batchParameters.Count, this));
 		}
 		finally
 		{
-			batch.Dispose2();
+			_batch.Release();
 		}
 	}
 	private async Task<FbBatchNonQueryResult> ExecuteCommandAsync(bool returnsSet, CancellationToken cancellationToken = default)
@@ -1280,17 +1307,16 @@ private async Task<FbBatchNonQueryResult> ExecuteCommandAsync(bool returnsSet, C
 			throw FbException.Create("Must declare command parameters.");
 		}
 
-		var batch = _statement.CreateBatch();
 		try
 		{
-			batch.MultiError = MultiError;
-			batch.BatchBufferSize = BatchBufferSize;
+			_batch.MultiError = MultiError;
+			_batch.BatchBufferSize = BatchBufferSize;
 			// Execute
-			return new FbBatchNonQueryResult(await batch.ExecuteAsync(_batchParameters.Count, this, cancellationToken).ConfigureAwait(false));
+			return new FbBatchNonQueryResult(await _batch.ExecuteAsync(_batchParameters.Count, this, cancellationToken).ConfigureAwait(false));
 		}
 		finally
 		{
-			await batch.Dispose2Async(cancellationToken).ConfigureAwait(false);
+			await _batch.ReleaseAsync(cancellationToken).ConfigureAwait(false);
 		}
 	}