dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

src/Npgsql/NpgsqlDataSourceCommand.cs
71 строка2 KB
Nikita Kazmin
Add an ability to filter and enrich tracing activities (#5853)
09 ноя 2024, 09:23
09 ноя 2024, 09:237949c45
using System;
using System.Data;
using System.Data.Common;
using System.Threading;
using System.Threading.Tasks;
using Npgsql.Properties;

namespace Npgsql;

sealed class NpgsqlDataSourceCommand : NpgsqlCommand
{
    internal NpgsqlDataSourceCommand(NpgsqlConnection connection)
        : base(cmdText: null, connection)
    {
    }

    // For NpgsqlBatch only
    internal NpgsqlDataSourceCommand(NpgsqlBatch batch, int batchCommandCapacity, NpgsqlConnection connection)
        : base(batch, batchCommandCapacity, connection)
    {
    }

    internal override async ValueTask<NpgsqlDataReader> ExecuteReader(
        bool async, CommandBehavior behavior,
        CancellationToken cancellationToken)
    {
        await InternalConnection!.Open(async, cancellationToken).ConfigureAwait(false);

        try
        {
            return await base.ExecuteReader(
                    async,
                    behavior | CommandBehavior.CloseConnection,
                    cancellationToken)
                .ConfigureAwait(false);
        }
        catch
        {
            try
            {
                await InternalConnection.Close(async).ConfigureAwait(false);
            }
            catch
            {
                // Swallow to allow the original exception to bubble up
            }

            throw;
        }
    }

    // The below are incompatible with commands executed directly against DbDataSource, since no DbConnection
    // is involved at the user API level and the command owns the DbConnection.
    public override void Prepare()
        => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);

    public override Task PrepareAsync(CancellationToken cancellationToken = default)
        => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);

    protected override DbConnection? DbConnection
    {
        get => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);
        set => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);
    }

    protected override DbTransaction? DbTransaction
    {
        get => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);
        set => throw new NotSupportedException(NpgsqlStrings.NotSupportedOnDataSourceCommand);
    }
}