public AsyncEnumerator(AsyncQueryingEnumerable <T> queryingEnumerable)
{
_shaperCommandContext = queryingEnumerable._shaperCommandContext;
_valueBufferFactory = _shaperCommandContext.ValueBufferFactory;
_relationalQueryContext = queryingEnumerable._relationalQueryContext;
_shaper = queryingEnumerable._shaper;
}
public Enumerator(QueryingEnumerable <T> queryingEnumerable)
{
_shaperCommandContext = queryingEnumerable._shaperCommandContext;
_valueBufferFactory = _shaperCommandContext.ValueBufferFactory;
_relationalQueryContext = queryingEnumerable._relationalQueryContext;
_shaper = queryingEnumerable._shaper;
_bufferlessMoveNext = BufferlessMoveNext;
}
public async Task <bool> MoveNext(CancellationToken cancellationToken)
{
cancellationToken.ThrowIfCancellationRequested();
try
{
await _queryingEnumerable._relationalQueryContext.Semaphore.WaitAsync(cancellationToken);
if (_buffer == null)
{
if (_dataReader == null)
{
await _queryingEnumerable._relationalQueryContext.Connection
.OpenAsync(cancellationToken);
var relationalCommand
= _queryingEnumerable._shaperCommandContext
.GetRelationalCommand(_queryingEnumerable._relationalQueryContext.ParameterValues);
await _queryingEnumerable._relationalQueryContext
.RegisterValueBufferCursorAsync(this, _queryingEnumerable._queryIndex, cancellationToken);
_dataReader
= await relationalCommand.ExecuteReaderAsync(
_queryingEnumerable._relationalQueryContext.Connection,
_queryingEnumerable._relationalQueryContext.ParameterValues,
manageConnection : false,
cancellationToken : cancellationToken);
_dbDataReader = _dataReader.DbDataReader;
_queryingEnumerable._shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _queryingEnumerable._shaperCommandContext.ValueBufferFactory;
}
var hasNext = await _dbDataReader.ReadAsync(cancellationToken);
_current
= hasNext
? _valueBufferFactory.Create(_dbDataReader)
: default(ValueBuffer);
return(hasNext);
}
if (_buffer.Count > 0)
{
_current = _buffer.Dequeue();
return(true);
}
return(false);
}
finally
{
_queryingEnumerable._relationalQueryContext.Semaphore.Release();
}
}
private async Task <bool> BufferlessMoveNext(bool buffer, CancellationToken cancellationToken)
{
try
{
if (_dataReader == null)
{
await _queryingEnumerable._relationalQueryContext.Connection
.OpenAsync(cancellationToken);
var relationalCommand
= _queryingEnumerable._shaperCommandContext
.GetRelationalCommand(_queryingEnumerable._relationalQueryContext.ParameterValues);
await _queryingEnumerable._relationalQueryContext
.RegisterValueBufferCursorAsync(this, cancellationToken);
_dataReader
= await relationalCommand.ExecuteReaderAsync(
_queryingEnumerable._relationalQueryContext.Connection,
_queryingEnumerable._relationalQueryContext.ParameterValues,
cancellationToken);
_dbDataReader = _dataReader.DbDataReader;
_queryingEnumerable._shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _queryingEnumerable._shaperCommandContext.ValueBufferFactory;
}
var hasNext = await _dbDataReader.ReadAsync(cancellationToken);
_current
= hasNext
? _valueBufferFactory.Create(_dbDataReader)
: default(ValueBuffer);
if (buffer)
{
await BufferAllAsync(cancellationToken);
}
return(hasNext);
}
catch (Exception)
{
_queryingEnumerable._relationalQueryContext.DeregisterValueBufferCursor(this);
_dataReader = null;
_dbDataReader = null;
throw;
}
}
private async Task <bool> BufferlessMoveNext(DbContext _, bool buffer, CancellationToken cancellationToken)
{
if (_dataReader == null)
{
await _relationalQueryContext.Connection.OpenAsync(cancellationToken);
try
{
var relationalCommand
= _shaperCommandContext
.GetRelationalCommand(_relationalQueryContext.ParameterValues, _relationalQueryContext);
await _relationalQueryContext.Connection
.RegisterBufferableAsync(this, cancellationToken);
_dataReader
= await relationalCommand.ExecuteReaderAsync(
_relationalQueryContext.Connection,
_relationalQueryContext.ParameterValues,
_relationalQueryContext.CommandLogger,
cancellationToken);
}
catch
{
// If failure happens creating the data reader, then it won't be available to
// handle closing the connection, so do it explicitly here to preserve ref counting.
_relationalQueryContext.Connection.Close();
throw;
}
_dbDataReader = _dataReader.DbDataReader;
_shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _shaperCommandContext.ValueBufferFactory;
}
var hasNext = await _dataReader.ReadAsync(cancellationToken);
Current
= hasNext
? _shaper.Shape(_relationalQueryContext, _valueBufferFactory.Create(_dbDataReader))
: default;
if (buffer)
{
await BufferAllAsync(cancellationToken);
}
return(hasNext);
}
private bool BufferlessMoveNext(bool buffer)
{
try
{
if (_dataReader == null)
{
_queryingEnumerable._relationalQueryContext.Connection.Open();
var relationalCommand
= _queryingEnumerable._shaperCommandContext
.GetRelationalCommand(_queryingEnumerable._relationalQueryContext.ParameterValues);
_queryingEnumerable._relationalQueryContext
.RegisterValueBufferCursor(this, _queryingEnumerable._queryIndex);
_dataReader
= relationalCommand.ExecuteReader(
_queryingEnumerable._relationalQueryContext.Connection,
_queryingEnumerable._relationalQueryContext.ParameterValues);
_dbDataReader = _dataReader.DbDataReader;
_queryingEnumerable._shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _queryingEnumerable._shaperCommandContext.ValueBufferFactory;
}
var hasNext = _dbDataReader.Read();
_current
= hasNext
? _valueBufferFactory.Create(_dbDataReader)
: default(ValueBuffer);
if (buffer)
{
BufferAll();
}
return(hasNext);
}
catch (Exception)
{
_queryingEnumerable._relationalQueryContext.DeregisterValueBufferCursor(this);
_dataReader = null;
_dbDataReader = null;
throw;
}
}
private bool BufferlessMoveNext(DbContext _, bool buffer)
{
if (_dataReader == null)
{
_relationalQueryContext.Connection.Open();
try
{
var relationalCommand
= _shaperCommandContext
.GetRelationalCommand(_relationalQueryContext.ParameterValues);
_relationalQueryContext.Connection.RegisterBufferable(this);
_dataReader
= relationalCommand.ExecuteReader(
_relationalQueryContext.Connection,
_relationalQueryContext.ParameterValues);
}
catch
{
// If failure happens creating the data reader, then it won't be available to
// handle closing the connection, so do it explicitly here to preserve ref counting.
_relationalQueryContext.Connection.Close();
throw;
}
_dbDataReader = _dataReader.DbDataReader;
_shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _shaperCommandContext.ValueBufferFactory;
}
var hasNext = _dataReader.Read();
Current
= hasNext
? _shaper.Shape(_relationalQueryContext, _valueBufferFactory.Create(_dbDataReader))
: default;
if (buffer)
{
BufferAll();
}
return(hasNext);
}
public bool MoveNext()
{
if (_buffer == null)
{
if (_dataReader == null)
{
_queryingEnumerable._relationalQueryContext.Connection.Open();
var relationalCommand
= _queryingEnumerable._shaperCommandContext
.GetRelationalCommand(_queryingEnumerable._relationalQueryContext.ParameterValues);
_queryingEnumerable._relationalQueryContext
.RegisterValueBufferCursor(this, _queryingEnumerable._queryIndex);
_dataReader
= relationalCommand.ExecuteReader(
_queryingEnumerable._relationalQueryContext.Connection,
_queryingEnumerable._relationalQueryContext.ParameterValues,
manageConnection: false);
_dbDataReader = _dataReader.DbDataReader;
_queryingEnumerable._shaperCommandContext.NotifyReaderCreated(_dbDataReader);
_valueBufferFactory = _queryingEnumerable._shaperCommandContext.ValueBufferFactory;
}
var hasNext = _dbDataReader.Read();
_current
= hasNext
? _valueBufferFactory.Create(_dbDataReader)
: default(ValueBuffer);
return(hasNext);
}
if (_buffer.Count > 0)
{
_current = _buffer.Dequeue();
return(true);
}
return(false);
}
public AsyncEnumerator(AsyncQueryingEnumerable queryingEnumerable)
{
_queryingEnumerable = queryingEnumerable;
_valueBufferFactory = _queryingEnumerable._shaperCommandContext.ValueBufferFactory;
}
/// <summary>
/// 异步恢复
/// </summary>
/// <param name="relationalReader">关系DataReader</param>
/// <param name="cancellationToken">取消令牌</param>
/// <returns></returns>
protected override async Task ConsumeAsync(RelationalDataReader relationalReader, CancellationToken cancellationToken = default(CancellationToken))
{
int commandPosition = 0;
try
{
if (Check.IsTraceEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Trace, OracleTraceTag.Entry, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.ConsumeAsync);
}
while (true)
{
if (commandPosition < CommandResultSet.Count && CommandResultSet[commandPosition] == ResultSetMapping.NoResultSet)
{
commandPosition++;
}
else
{
if (commandPosition < CommandResultSet.Count)
{
if (ModificationCommands[commandPosition].RequiresResultPropagation)
{
int rowsAffected = 0;
int num = 0;
do
{
ModificationCommand tableModification = ModificationCommands[commandPosition];
if (!(await relationalReader.ReadAsync(cancellationToken)))
{
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(ModificationCommands.Count((ModificationCommand m) => m.RequiresResultPropagation), rowsAffected), ModificationCommands[commandPosition].Entries);
}
IRelationalValueBufferFactory relationalValueBufferFactory = CreateValueBufferFactory(tableModification.ColumnModifications);
tableModification.PropagateResults(relationalValueBufferFactory.Create(relationalReader.DbDataReader));
rowsAffected++;
num = commandPosition + 1;
commandPosition = num;
}while (num < CommandResultSet.Count && CommandResultSet[commandPosition - 1] == ResultSetMapping.NotLastInResultSet);
}
else
{
int expectedRowsAffected = 1;
while (true)
{
int num = commandPosition + 1;
commandPosition = num;
if (num >= CommandResultSet.Count || CommandResultSet[commandPosition - 1] != ResultSetMapping.NotLastInResultSet)
{
break;
}
expectedRowsAffected++;
}
if (!relationalReader.Read())
{
break;
}
int rowsAffected = relationalReader.DbDataReader.GetInt32(0);
if (rowsAffected != expectedRowsAffected)
{
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(expectedRowsAffected, rowsAffected), ModificationCommands[commandPosition - 1].Entries);
}
}
}
bool flag = commandPosition < CommandResultSet.Count;
if (flag)
{
flag = await relationalReader.DbDataReader.NextResultAsync(cancellationToken);
}
if (!flag)
{
return;
}
}
}
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(1, 0), ModificationCommands[commandPosition - 1].Entries);
}
catch (DbUpdateException dbEx)
{
if (Check.IsErrorEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Error, OracleTraceTag.Error, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.ConsumeAsync, dbEx.ToString());
}
throw;
}
catch (Exception ex)
{
if (Check.IsErrorEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Error, OracleTraceTag.Error, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.ConsumeAsync, ex.ToString());
}
throw new DbUpdateException(RelationalStrings.UpdateStoreException, ex, ModificationCommands[commandPosition].Entries);
}
}
/// <summary>
/// 恢复
/// </summary>
/// <param name="relationalReader">关系DataReader</param>
protected override void Consume(RelationalDataReader relationalReader)
{
int commandPosition = 0;
int rowsAffected = 0;
try
{
if (Check.IsTraceEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Trace, OracleTraceTag.Entry, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.Consume);
}
while (true)
{
if (commandPosition < CommandResultSet.Count && CommandResultSet[commandPosition] == ResultSetMapping.NoResultSet)
{
commandPosition++;
}
else
{
if (commandPosition < CommandResultSet.Count)
{
if (ModificationCommands[commandPosition].RequiresResultPropagation)
{
rowsAffected = 0;
do
{
ModificationCommand modificationCommand = ModificationCommands[commandPosition];
if (!relationalReader.Read())
{
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(ModificationCommands.Count((ModificationCommand m) => m.RequiresResultPropagation), rowsAffected), ModificationCommands[commandPosition].Entries);
}
IRelationalValueBufferFactory relationalValueBufferFactory = CreateValueBufferFactory(modificationCommand.ColumnModifications);
modificationCommand.PropagateResults(relationalValueBufferFactory.Create(relationalReader.DbDataReader));
rowsAffected++;
foreach (ColumnModification columnModification in modificationCommand.ColumnModifications)
{
if (Check.IsTraceEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Trace, OracleTraceTag.SQL, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.Consume, $"Column Name: {columnModification.ColumnName}");
}
}
}while (++commandPosition < CommandResultSet.Count && CommandResultSet[commandPosition - 1] == ResultSetMapping.NotLastInResultSet);
}
else
{
int expectedRowsAffected = 1;
while (++commandPosition < CommandResultSet.Count && CommandResultSet[commandPosition - 1] == ResultSetMapping.NotLastInResultSet)
{
expectedRowsAffected++;
}
if (!relationalReader.Read())
{
break;
}
rowsAffected = relationalReader.DbDataReader.GetInt32(0);
if (rowsAffected != expectedRowsAffected)
{
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(expectedRowsAffected, rowsAffected), ModificationCommands[commandPosition - 1].Entries);
}
}
}
if (commandPosition >= CommandResultSet.Count || !relationalReader.DbDataReader.NextResult())
{
return;
}
}
}
throw new DbUpdateConcurrencyException(RelationalStrings.UpdateConcurrencyException(1, 0), ModificationCommands[commandPosition - 1].Entries);
}
catch (DbUpdateException dbEx)
{
if (Check.IsErrorEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Error, OracleTraceTag.Error, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.Consume, dbEx.ToString());
}
throw;
}
catch (Exception ex)
{
if (Check.IsErrorEnabled(m_oracleLogger?.Logger))
{
Trace <DbLoggerCategory.Update> .Write(m_oracleLogger, LogLevel.Error, OracleTraceTag.Error, OracleTraceClassName.OracleModificationCommandBatch, OracleTraceFuncName.Consume, ex.ToString());
}
throw new DbUpdateException(RelationalStrings.UpdateStoreException, ex, ModificationCommands[commandPosition - 1].Entries);
}
}
public AsyncEnumerator(AsyncQueryingEnumerable queryingEnumerable)
{
_queryingEnumerable = queryingEnumerable;
_valueBufferFactory = _queryingEnumerable._commandBuilder.ValueBufferFactory;
}
