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

src/Npgsql/UnpooledDataSource.cs

48 строк1 KB

Shay Rojansky

Expose NpgsqlDataSource.Clear() publicly (#5903)

28 окт 2024, 10:53

28 окт 2024, 10:5332fc88f

using System.Diagnostics.CodeAnalysis;
using System.Threading;
using System.Threading.Tasks;
using Npgsql.Internal;
using Npgsql.Util;

namespace Npgsql;

sealed class UnpooledDataSource(NpgsqlConnectionStringBuilder settings, NpgsqlDataSourceConfiguration dataSourceConfig)
    : NpgsqlDataSource(settings, dataSourceConfig)
{
    volatile int _numConnectors;

    internal override (int Total, int Idle, int Busy) Statistics => (_numConnectors, 0, _numConnectors);

    internal override bool OwnsConnectors => true;

    internal override async ValueTask<NpgsqlConnector> Get(
        NpgsqlConnection conn, NpgsqlTimeout timeout, bool async, CancellationToken cancellationToken)
    {
        CheckDisposed();

        var connector = new NpgsqlConnector(this, conn);
        await connector.Open(timeout, async, cancellationToken).ConfigureAwait(false);
        Interlocked.Increment(ref _numConnectors);
        return connector;
    }

    internal override bool TryGetIdleConnector([NotNullWhen(true)] out NpgsqlConnector? connector)
    {
        connector = null;
        return false;
    }

    internal override ValueTask<NpgsqlConnector?> OpenNewConnector(
        NpgsqlConnection conn, NpgsqlTimeout timeout, bool async, CancellationToken cancellationToken)
        => new((NpgsqlConnector?)null);

    internal override void Return(NpgsqlConnector connector)
    {
        Interlocked.Decrement(ref _numConnectors);
        connector.Close();
    }

    public override void Clear()
    {
    }
}