fix: also increment database connection count

This commit is contained in:
sam 2024-12-10 02:34:45 +01:00
parent 5157105c35
commit 27e1903c4b
Signed by: sam
GPG key ID: 5F3C3C1B3166639D
2 changed files with 12 additions and 3 deletions

View file

@ -44,10 +44,17 @@ public class DatabasePool
_dataSource = dataSourceBuilder.Build();
}
public async Task<DatabaseConnection> AcquireAsync(CancellationToken ct = default) =>
new(await _dataSource.OpenConnectionAsync(ct));
public async Task<DatabaseConnection> AcquireAsync(CancellationToken ct = default)
{
IncrementConnections();
return new DatabaseConnection(await _dataSource.OpenConnectionAsync(ct));
}
public DatabaseConnection Acquire() => new(_dataSource.OpenConnection());
public DatabaseConnection Acquire()
{
IncrementConnections();
return new DatabaseConnection(_dataSource.OpenConnection());
}
public async Task ExecuteAsync(
Func<DatabaseConnection, Task> func,