npgsql/src/Npgsql/MultiHostDataSourceWrapper.cs at main · npgsql/npgsql

History

46 lines (38 loc) · 2.54 KB

Raw

using Npgsql.Internal;

using Npgsql.Util;

using System.Diagnostics.CodeAnalysis;

using System.Threading;

using System.Threading.Tasks;

using System.Transactions;

namespace Npgsql;

sealed class MultiHostDataSourceWrapper(NpgsqlMultiHostDataSource wrappedSource, TargetSessionAttributes targetSessionAttributes)

: NpgsqlDataSource(CloneSettingsForTargetSessionAttributes(wrappedSource.Settings, targetSessionAttributes), wrappedSource.Configuration, reportMetrics: false)

{

internal NpgsqlMultiHostDataSource WrappedSource { get; } = wrappedSource;

internal override bool OwnsConnectors => false;

public override void Clear() => WrappedSource.Clear();

static NpgsqlConnectionStringBuilder CloneSettingsForTargetSessionAttributes(

NpgsqlConnectionStringBuilder settings,

TargetSessionAttributes targetSessionAttributes)

{

var clonedSettings = settings.Clone();

clonedSettings.TargetSessionAttributesParsed = targetSessionAttributes;

return clonedSettings;

}

internal override (int Total, int Idle, int Busy) Statistics => WrappedSource.Statistics;

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

=> WrappedSource.Get(conn, timeout, async, cancellationToken);

internal override bool TryGetIdleConnector([NotNullWhen(true)] out NpgsqlConnector? connector)

=> throw new NpgsqlException("Npgsql bug: trying to get an idle connector from " + nameof(MultiHostDataSourceWrapper));

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

=> throw new NpgsqlException("Npgsql bug: trying to open a new connector from " + nameof(MultiHostDataSourceWrapper));

internal override void Return(NpgsqlConnector connector)

=> WrappedSource.Return(connector);

internal override void AddPendingEnlistedConnector(NpgsqlConnector connector, Transaction transaction)

=> WrappedSource.AddPendingEnlistedConnector(connector, transaction);

internal override bool TryRemovePendingEnlistedConnector(NpgsqlConnector connector, Transaction transaction)

=> WrappedSource.TryRemovePendingEnlistedConnector(connector, transaction);

internal override bool TryRentEnlistedPending(Transaction transaction, NpgsqlConnection connection,

[NotNullWhen(true)] out NpgsqlConnector? connector)

=> WrappedSource.TryRentEnlistedPending(transaction, connection, out connector);

}

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Expand file tree

Search code, repositories, users, issues, pull requests...

Uh oh!

FilesExpand file tree

MultiHostDataSourceWrapper.cs

Latest commit

History

MultiHostDataSourceWrapper.cs

File metadata and controls

Expand file tree