Skip to content

Commit 5766978

Browse files
committed
supported metrics for OnConnBeginTx callback
1 parent 04f5448 commit 5766978

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

metrics/sql.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,19 @@ func databaseSQL(config Config) (t trace.DatabaseSQL) {
6464

6565
return nil
6666
}
67+
t.OnConnBeginTx = func(info trace.DatabaseSQLConnBeginTxStartInfo) func(trace.DatabaseSQLConnBeginTxDoneInfo) {
68+
start := time.Now()
69+
if config.Details()&trace.DatabaseSQLTxEvents != 0 {
70+
return func(info trace.DatabaseSQLConnBeginTxDoneInfo) {
71+
txBegin.With(map[string]string{
72+
"status": errorBrief(info.Error),
73+
}).Inc()
74+
txBeginLatency.With(nil).Record(time.Since(start))
75+
}
76+
}
77+
78+
return nil
79+
}
6780
t.OnTxCommit = func(info trace.DatabaseSQLTxCommitStartInfo) func(trace.DatabaseSQLTxCommitDoneInfo) {
6881
start := time.Now()
6982

0 commit comments

Comments
 (0)