public override Task ConnectionOpeningAsync(ConnectionEventData eventData, CancellationToken cancellationToken)
{
OpeningAsyncCounter++;
return(base.ConnectionOpeningAsync(eventData, cancellationToken));
}
public override void ConnectionOpened(ConnectionEventData eventData)
{
OpenedCounter++;
}
public override async Task <InterceptionResult> ConnectionOpeningAsync(DbConnection connection, ConnectionEventData eventData, InterceptionResult result, CancellationToken cancellationToken = new CancellationToken())
{
if (connection is SqlConnection sqlConnection)
{
sqlConnection.AccessToken = await GetAccessTokenAsync();
}
return(result);
}
public override InterceptionResult ConnectionOpening(DbConnection connection, ConnectionEventData eventData, InterceptionResult result)
{
if (connection is SqlConnection sqlConnection)
{
sqlConnection.AccessToken = GetAccessTokenAsync().Result;
}
return(result);
}
/// <summary>
/// 拦截数据库连接
/// </summary>
/// <param name="connection">数据库连接对象</param>
/// <param name="eventData">数据库连接事件数据</param>
/// <param name="result">拦截结果</param>
/// <returns></returns>
public override InterceptionResult ConnectionOpening(DbConnection connection, ConnectionEventData eventData, InterceptionResult result)
{
// 打印数据库连接信息到 MiniProfiler
PrintConnectionToMiniProfiler(connection, eventData);
return(base.ConnectionOpening(connection, eventData, result));
}
/// <summary>
/// 拦截数据库连接
/// </summary>
/// <param name="connection">数据库连接对象</param>
/// <param name="eventData">数据库连接事件数据</param>
/// <param name="result">拦截器结果</param>
/// <param name="cancellationToken">取消异步Token</param>
/// <returns></returns>
public override ValueTask <InterceptionResult> ConnectionOpeningAsync(DbConnection connection, ConnectionEventData eventData, InterceptionResult result, CancellationToken cancellationToken = default)
{
// 打印数据库连接信息到 MiniProfiler
PrintConnectionToMiniProfiler(connection, eventData);
return(base.ConnectionOpeningAsync(connection, eventData, result, cancellationToken));
}
public InterceptionResult ConnectionOpening(DbConnection connection, ConnectionEventData eventData,
InterceptionResult result)
{
return(result);
}
public InterceptionResult ConnectionClosing(DbConnection connection, ConnectionEventData eventData,
InterceptionResult result)
{
TrackEvent(nameof(ConnectionClosing), connection, eventData.ConnectionId);
return(result);
}
public override async Task <InterceptionResult> ConnectionOpeningAsync(DbConnection connection, ConnectionEventData eventData, InterceptionResult result, CancellationToken cancellationToken = default)
{
const string SqlDatabaseResourceUrl = "https://database.windows.net/";
var sqlConnection = (SqlConnection)connection;
sqlConnection.AccessToken = await new AzureServiceTokenProvider().GetAccessTokenAsync(SqlDatabaseResourceUrl);
return(await base.ConnectionOpeningAsync(connection, eventData, result, cancellationToken));
}
protected virtual void AssertOpening(ConnectionEventData eventData)
{
Assert.NotNull(eventData.Context);
Assert.NotEqual(default, eventData.ConnectionId);
/// <summary>
/// 打印数据库连接信息到 MiniProfiler
/// </summary>
/// <param name="connection">数据库连接对象</param>
/// <param name="eventData">数据库连接事件数据</param>
private void PrintConnectionToMiniProfiler(DbConnection connection, ConnectionEventData eventData)
{
// 打印连接信息消息
App.PrintToMiniProfiler("connection", "Information", $"[Connection Id: {eventData.ConnectionId}] / [Database: {connection.Database}]{(IsPrintDbConnectionInfo ? $" / [Connection String: {connection.ConnectionString}]" : string.Empty)}");
}
public override InterceptionResult ConnectionClosing(DbConnection connection, ConnectionEventData eventData, InterceptionResult result)
{
return(result);
}
public override ValueTask <InterceptionResult> ConnectionOpeningAsync(DbConnection connection, ConnectionEventData eventData, InterceptionResult result, CancellationToken cancellationToken = default)
{
return(new ValueTask <InterceptionResult>(result));
}
public override void ConnectionOpening(ConnectionEventData eventData)
{
instanceMarker = UniqueInstanceIdentifier;
ConnectionOpeningCounter++;
}
public override ValueTask <InterceptionResult> ConnectionOpeningAsync(DbConnection connection, ConnectionEventData eventData,
InterceptionResult result,
CancellationToken cancellationToken = new())
