/// <summary> /// Executes the specified implementation asynchronously. /// </summary> /// <param name="executionToken">The execution token.</param> /// <param name="implementation">The implementation.</param> /// <param name="cancellationToken">The cancellation token.</param> /// <param name="state">The state.</param> /// <returns>The caller is expected to use the StreamingCommandCompletionToken to close any lingering connections and fire appropriate events.</returns> /// <exception cref="System.NotImplementedException"></exception> public override async Task <StreamingCommandCompletionToken> ExecuteStreamAsync(CommandExecutionToken <SQLiteCommand, SQLiteParameter> executionToken, StreamingCommandImplementationAsync <SQLiteCommand> implementation, CancellationToken cancellationToken, object?state) { if (executionToken == null) { throw new ArgumentNullException(nameof(executionToken), $"{nameof(executionToken)} is null."); } if (implementation == null) { throw new ArgumentNullException(nameof(implementation), $"{nameof(implementation)} is null."); } var mode = DisableLocks ? LockType.None : (executionToken as SQLiteCommandExecutionToken)?.LockType ?? LockType.Write; var startTime = DateTimeOffset.Now; OnExecutionStarted(executionToken, startTime, state); IDisposable?lockToken = null; try { switch (mode) { case LockType.Read: lockToken = await SyncLock.ReaderLockAsync().ConfigureAwait(false); break; case LockType.Write: lockToken = await SyncLock.WriterLockAsync().ConfigureAwait(false); break; } var cmd = new SQLiteCommand(); cmd.Connection = m_Connection; if (m_Transaction != null) { cmd.Transaction = m_Transaction; } executionToken.PopulateCommand(cmd, DefaultCommandTimeout); await implementation(cmd).ConfigureAwait(false); return(new StreamingCommandCompletionToken(this, executionToken, startTime, state, cmd) { LockToken = lockToken }); } catch (Exception ex) { lockToken?.Dispose(); if (cancellationToken.IsCancellationRequested) //convert SQLiteException into a OperationCanceledException { var ex2 = new OperationCanceledException("Operation was canceled.", ex, cancellationToken); OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex2, state); throw ex2; } else { OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex, state); throw; } } }
/// <summary> /// Executes the specified implementation asynchronously. /// </summary> /// <param name="executionToken">The execution token.</param> /// <param name="implementation">The implementation.</param> /// <param name="cancellationToken">The cancellation token.</param> /// <param name="state">The state.</param> /// <returns>The caller is expected to use the StreamingCommandCompletionToken to close any lingering connections and fire appropriate events.</returns> /// <exception cref="System.NotImplementedException"></exception> public override async Task <StreamingCommandCompletionToken> ExecuteStreamAsync(CommandExecutionToken <NpgsqlCommand, NpgsqlParameter> executionToken, StreamingCommandImplementationAsync <NpgsqlCommand> implementation, CancellationToken cancellationToken, object?state) { if (executionToken == null) { throw new ArgumentNullException(nameof(executionToken), $"{nameof(executionToken)} is null."); } if (implementation == null) { throw new ArgumentNullException(nameof(implementation), $"{nameof(implementation)} is null."); } var startTime = DateTimeOffset.Now; OnExecutionStarted(executionToken, startTime, state); try { var cmd = new NpgsqlCommand(); cmd.Connection = m_Connection; if (m_Transaction != null) { cmd.Transaction = m_Transaction; } executionToken.PopulateCommand(cmd, DefaultCommandTimeout); if (((PostgreSqlCommandExecutionToken)executionToken).DereferenceCursors) { await DereferenceCursorsAsync(cmd, implementation).ConfigureAwait(false); } else { await implementation(cmd).ConfigureAwait(false); } return(new StreamingCommandCompletionToken(this, executionToken, startTime, state, cmd)); } catch (Exception ex) { if (cancellationToken.IsCancellationRequested) //convert Exception into a OperationCanceledException { var ex2 = new OperationCanceledException("Operation was canceled.", ex, cancellationToken); OnExecutionCanceled(executionToken, startTime, DateTimeOffset.Now, state); throw ex2; } else { OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex, state); throw; } } }
/// <summary> /// Dereferences cursors returned by a stored procedure. /// </summary> /// <param name="cmd">The command.</param> /// <param name="implementation">The implementation.</param> protected static async Task <NpgsqlTransaction?> DereferenceCursorsAsync(NpgsqlCommand cmd, StreamingCommandImplementationAsync <NpgsqlCommand> implementation) { if (cmd == null) { throw new ArgumentNullException(nameof(cmd), $"{nameof(cmd)} is null."); } if (cmd.Connection == null) { throw new ArgumentNullException($"{nameof(cmd)}.{nameof(cmd.Connection)}", $"{nameof(cmd)}.{nameof(cmd.Connection)} is null."); } if (implementation == null) { throw new ArgumentNullException(nameof(implementation), $"{nameof(implementation)} is null."); } var closeTransaction = false; if (cmd.Transaction == null) { cmd.Transaction = cmd.Connection.BeginTransaction(); closeTransaction = true; } var sql = new StringBuilder(); using (var reader = await cmd.ExecuteReaderAsync().ConfigureAwait(false)) while (await reader.ReadAsync().ConfigureAwait(false)) { sql.AppendLine($"FETCH ALL IN \"{reader.GetString(0)}\";"); } using (var cmd2 = new NpgsqlCommand()) { cmd2.Connection = cmd.Connection; cmd2.Transaction = cmd.Transaction; cmd2.CommandTimeout = cmd.CommandTimeout; cmd2.CommandText = sql.ToString(); cmd2.CommandType = CommandType.Text; await implementation(cmd2).ConfigureAwait(false); } return(closeTransaction ? cmd.Transaction : null); }
/// <summary> /// Execute stream as an asynchronous operation. /// </summary> /// <param name="executionToken">The execution token.</param> /// <param name="implementation">The implementation.</param> /// <param name="cancellationToken">The cancellation token that can be used by other objects or threads to receive notice of cancellation.</param> /// <param name="state">The state.</param> /// <returns>A Task<StreamingCommandCompletionToken> representing the asynchronous operation.</returns> /// <exception cref="System.ArgumentNullException">executionToken</exception> /// <exception cref="System.ArgumentNullException">implementation</exception> /// <exception cref="System.ArgumentNullException">executionToken - only AccessCommandExecutionToken is supported.</exception> /// <exception cref="System.InvalidOperationException">currentToken.ExecutionMode is ExecuteScalarAndForward, but currentToken.ForwardResult is null.</exception> public override async Task <StreamingCommandCompletionToken> ExecuteStreamAsync(CommandExecutionToken <OleDbCommand, OleDbParameter> executionToken, StreamingCommandImplementationAsync <OleDbCommand> implementation, CancellationToken cancellationToken, object?state) { if (executionToken == null) { throw new ArgumentNullException(nameof(executionToken), $"{nameof(executionToken)} is null."); } if (implementation == null) { throw new ArgumentNullException(nameof(implementation), $"{nameof(implementation)} is null."); } var currentToken = executionToken as AccessCommandExecutionToken; if (currentToken == null) { throw new ArgumentNullException(nameof(executionToken), "only AccessCommandExecutionToken is supported."); } var startTime = DateTimeOffset.Now; OleDbConnection?con = null; try { con = await CreateConnectionAsync(cancellationToken).ConfigureAwait(false); OleDbCommand?cmdToReturn = null; while (currentToken != null) { OnExecutionStarted(currentToken, startTime, state); using (var cmd = new OleDbCommand()) { cmd.Connection = con; currentToken.PopulateCommand(cmd, DefaultCommandTimeout); if (currentToken.ExecutionMode == AccessCommandExecutionMode.Materializer) { await implementation(cmd).ConfigureAwait(false); cmdToReturn = cmd; } else if (currentToken.ExecutionMode == AccessCommandExecutionMode.ExecuteScalarAndForward) { if (currentToken.ForwardResult == null) { throw new InvalidOperationException("currentToken.ExecutionMode is ExecuteScalarAndForward, but currentToken.ForwardResult is null."); } currentToken.ForwardResult(await cmd.ExecuteScalarAsync().ConfigureAwait(false)); } else { await cmd.ExecuteNonQueryAsync().ConfigureAwait(false); } } currentToken = currentToken.NextCommand; } return(new StreamingCommandCompletionToken(this, executionToken, startTime, state, cmdToReturn, con)); } catch (Exception ex) { #if NET6_0_OR_GREATER if (con != null) { await con.DisposeAsync().ConfigureAwait(false); } #else con?.Dispose(); #endif if (cancellationToken.IsCancellationRequested) //convert Exception into a OperationCanceledException { var ex2 = new OperationCanceledException("Operation was canceled.", ex, cancellationToken); OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex2, state); throw ex2; } else { OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex, state); throw; } } }
/// <summary> /// Executes the specified implementation asynchronously. /// </summary> /// <param name="executionToken">The execution token.</param> /// <param name="implementation">The implementation.</param> /// <param name="cancellationToken">The cancellation token.</param> /// <param name="state">The state.</param> /// <returns>The caller is expected to use the StreamingCommandCompletionToken to close any lingering connections and fire appropriate events.</returns> public abstract Task <StreamingCommandCompletionToken> ExecuteStreamAsync(CommandExecutionToken <TCommand, TParameter> executionToken, StreamingCommandImplementationAsync <TCommand> implementation, CancellationToken cancellationToken, object?state);
/// <summary> /// Executes the specified implementation asynchronously. /// </summary> /// <param name="executionToken">The execution token.</param> /// <param name="implementation">The implementation.</param> /// <param name="cancellationToken">The cancellation token.</param> /// <param name="state">The state.</param> /// <returns>The caller is expected to use the StreamingCommandCompletionToken to close any lingering connections and fire appropriate events.</returns> /// <exception cref="System.NotImplementedException"></exception> public override async Task <StreamingCommandCompletionToken> ExecuteStreamAsync(CommandExecutionToken <SqlCommand, SqlParameter> executionToken, StreamingCommandImplementationAsync <SqlCommand> implementation, CancellationToken cancellationToken, object?state) { if (executionToken == null) { throw new ArgumentNullException(nameof(executionToken), $"{nameof(executionToken)} is null."); } if (implementation == null) { throw new ArgumentNullException(nameof(implementation), $"{nameof(implementation)} is null."); } var startTime = DateTimeOffset.Now; OnExecutionStarted(executionToken, startTime, state); SqlConnection?con = null; try { con = await CreateConnectionAsync(cancellationToken).ConfigureAwait(false); var cmd = new SqlCommand(); cmd.Connection = con; executionToken.PopulateCommand(cmd, DefaultCommandTimeout); await implementation(cmd).ConfigureAwait(false); return(new StreamingCommandCompletionToken(this, executionToken, startTime, state, cmd, con)); } catch (Exception ex) { #if NET6_0_OR_GREATER if (con != null) { await con.DisposeAsync().ConfigureAwait(false); } #else con?.Dispose(); #endif if (cancellationToken.IsCancellationRequested) //convert Exception into a OperationCanceledException { var ex2 = new OperationCanceledException("Operation was canceled.", ex, cancellationToken); OnExecutionCanceled(executionToken, startTime, DateTimeOffset.Now, state); throw ex2; } else { OnExecutionError(executionToken, startTime, DateTimeOffset.Now, ex, state); throw; } } }