Skip to content

Commit 7777513

Browse files
committed
Add sessionID support and BeforeExecuteWithTime to telemetry
- Add sessionID field to metricContext struct - Update BeforeExecute to accept sessionID parameter - Add BeforeExecuteWithTime method for custom start times - Update connection.go to pass sessionID in BeforeExecute call This enables proper session tracking in telemetry and allows capturing accurate execution times by providing a custom start time.
1 parent 68e9c4e commit 7777513

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

connection.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ func (c *conn) QueryContext(ctx context.Context, query string, args []driver.Nam
200200
var statementID string
201201
if c.telemetry != nil && exStmtResp != nil && exStmtResp.OperationHandle != nil && exStmtResp.OperationHandle.OperationId != nil {
202202
statementID = client.SprintGuid(exStmtResp.OperationHandle.OperationId.GUID)
203-
ctx = c.telemetry.BeforeExecute(ctx, statementID)
203+
ctx = c.telemetry.BeforeExecute(ctx, c.id, statementID)
204204
defer func() {
205205
c.telemetry.AfterExecute(ctx, err)
206206
c.telemetry.CompleteStatement(ctx, statementID, err != nil)

0 commit comments

Comments
 (0)