Profiler for any ADO.NET connection and XRay

AWSIDbProfiler.cs

    
    /// <summary>
    /// This must be called NpgsqlConnection for Dapper to detect the connection as being an NpgsqlConnection. 
    /// Otherwise Dapper doesn't handle arrays correctly for Postgresql connections. 
    /// </summary>
    public class NpgsqlConnection : StackExchange.Profiling.Data.ProfiledDbConnection
    {
        public NpgsqlConnection(DbConnection connection, IDbProfiler profiler) : base(connection, profiler)
        {

        }
    }

    public class AWSIDbProfiler : StackExchange.Profiling.Data.IDbProfiler
    {
        public static AWSIDbProfiler Instance { get; } = new AWSIDbProfiler();

        public bool IsActive => true;

        public void ExecuteStart(IDbCommand profiledDbCommand, SqlExecuteType executeType)
        {
            AWSXRayRecorder.Instance.BeginSubsegment("SQL Command");
            AWSXRayRecorder.Instance.AddSqlInformation("sanitized_query", profiledDbCommand.CommandText);
            AWSXRayRecorder.Instance.AddSqlInformation("connection_string", profiledDbCommand.Connection.ConnectionString);                        
        }

        public void ExecuteFinish(IDbCommand profiledDbCommand, SqlExecuteType executeType, DbDataReader reader)
        {
            AWSXRayRecorder.Instance.EndSubsegment();
        }

        public void OnError(IDbCommand profiledDbCommand, SqlExecuteType executeType, Exception exception)
        {
            AWSXRayRecorder.Instance.AddException(exception);
        }

        public void ReaderFinish(IDataReader reader)
        {
            
        }
    }