GenericBatchingBatcher.cs

//------------------------------------------------------------------------------
// <auto-generated>
//     This code was generated by AsyncGenerator.
//
//     Changes to this file may cause incorrect behavior and will be lost if
//     the code is regenerated.
// </auto-generated>
//------------------------------------------------------------------------------


using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Text;
using NHibernate.AdoNet.Util;
using NHibernate.Exceptions;
using NHibernate.SqlCommand;

namespace NHibernate.AdoNet
{
	using System.Threading.Tasks;
	using System.Threading;
	public partial class GenericBatchingBatcher : AbstractBatcher
	{

		public override async Task AddToBatchAsync(IExpectation expectation, CancellationToken cancellationToken)
		{
			cancellationToken.ThrowIfCancellationRequested();
			var batchCommand = CurrentCommand;
			if (_maxNumberOfParameters.HasValue && 
				_currentBatch.CountOfParameters + batchCommand.Parameters.Count > _maxNumberOfParameters)
			{
				await (ExecuteBatchWithTimingAsync(batchCommand, cancellationToken)).ConfigureAwait(false);
			}
			_totalExpectedRowsAffected += expectation.ExpectedRowCount;
			Driver.AdjustCommand(batchCommand);
			LogBatchCommand(batchCommand);
			_currentBatch.Append(batchCommand.Parameters);

			if (_currentBatch.CountOfCommands >= BatchSize)
			{
				await (ExecuteBatchWithTimingAsync(batchCommand, cancellationToken)).ConfigureAwait(false);
			}
		}

		protected override async Task DoExecuteBatchAsync(DbCommand ps, CancellationToken cancellationToken)
		{
			cancellationToken.ThrowIfCancellationRequested();
			if (_currentBatch.CountOfCommands == 0)
			{
				Expectations.VerifyOutcomeBatched(_totalExpectedRowsAffected, 0, ps);
				return;
			}
			try
			{
				Log.Debug("Executing batch");
				await (CheckReadersAsync(cancellationToken)).ConfigureAwait(false);
				if (Factory.Settings.SqlStatementLogger.IsDebugEnabled)
				{
					Factory.Settings.SqlStatementLogger.LogBatchCommand(_currentBatchCommandsLog.ToString());
				}

				int rowsAffected;
				try
				{
					rowsAffected = await (_currentBatch.ExecuteNonQueryAsync(cancellationToken)).ConfigureAwait(false);
				}
				catch (DbException e)
				{
					throw ADOExceptionHelper.Convert(Factory.SQLExceptionConverter, e, "could not execute batch command.");
				}

				Expectations.VerifyOutcomeBatched(_totalExpectedRowsAffected, rowsAffected, ps);
			}
			finally
			{
				ClearCurrentBatch();
			}
		}

		private partial class BatchingCommandSet
		{

			public async Task<int> ExecuteNonQueryAsync(CancellationToken cancellationToken)
			{
				cancellationToken.ThrowIfCancellationRequested();
				if (CountOfCommands == 0)
				{
					return 0;
				}

				using (var batcherCommand = _batcher.Driver.GenerateCommand(
					_commandType,
					_sql.ToSqlString(),
					_sqlTypes.ToArray()))
				{
					for (var i = 0; i < _parameters.Count; i++)
					{
						var parameter = _parameters[i];
						var cmdParam = batcherCommand.Parameters[i];
						cmdParam.Value = parameter.Value;
						cmdParam.Direction = parameter.Direction;
						cmdParam.Precision = parameter.Precision;
						cmdParam.Scale = parameter.Scale;
						cmdParam.Size = parameter.Size;
					}
					await (_batcher.PrepareAsync(batcherCommand, cancellationToken)).ConfigureAwait(false);
					return await (batcherCommand.ExecuteNonQueryAsync(cancellationToken)).ConfigureAwait(false);
				}
			}
		}
	}
}