chore: error log on query error (#7553)

* chore: error log or query error

* chore: no error on context cancel
This commit is contained in:
Srikanth Chekuri 2025-04-07 20:45:49 +05:30 committed by GitHub
parent 838ef5dcc5
commit e17f63a50c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2,6 +2,8 @@ package telemetrystorehook
import ( import (
"context" "context"
"database/sql"
"errors"
"log/slog" "log/slog"
"time" "time"
@ -30,12 +32,21 @@ func (logging) BeforeQuery(ctx context.Context, event *telemetrystore.QueryEvent
} }
func (hook *logging) AfterQuery(ctx context.Context, event *telemetrystore.QueryEvent) { func (hook *logging) AfterQuery(ctx context.Context, event *telemetrystore.QueryEvent) {
hook.logger.Log( level := hook.level
ctx, args := []any{
hook.level,
"::TELEMETRYSTORE-QUERY::",
"db.query.text", event.Query, "db.query.text", event.Query,
"db.query.args", event.QueryArgs, "db.query.args", event.QueryArgs,
"db.duration", time.Since(event.StartTime).String(), "db.duration", time.Since(event.StartTime).String(),
}
if event.Err != nil && !errors.Is(event.Err, sql.ErrNoRows) && !errors.Is(event.Err, context.Canceled) {
level = slog.LevelError
args = append(args, "db.query.error", event.Err)
}
hook.logger.Log(
ctx,
level,
"::TELEMETRYSTORE-QUERY::",
args...,
) )
} }