Compare commits
8 Commits
Author | SHA1 | Date | |
---|---|---|---|
bab2833cf7 | |||
564efef60b | |||
cb9d32bf1c | |||
c363033abd | |||
dd914ed9fb | |||
7d2787c3a6 | |||
a81da5bd2d | |||
44cd4a1e16 |
9
.gitignore
vendored
9
.gitignore
vendored
@@ -4,6 +4,7 @@
|
|||||||
*.dll
|
*.dll
|
||||||
*.so
|
*.so
|
||||||
*.dylib
|
*.dylib
|
||||||
|
bin
|
||||||
|
|
||||||
# Test binary, built with `go test -c`
|
# Test binary, built with `go test -c`
|
||||||
*.test
|
*.test
|
||||||
@@ -13,3 +14,11 @@
|
|||||||
|
|
||||||
# Dependency directories (remove the comment below to include it)
|
# Dependency directories (remove the comment below to include it)
|
||||||
# vendor/
|
# vendor/
|
||||||
|
|
||||||
|
# Go workspace file
|
||||||
|
go.work
|
||||||
|
|
||||||
|
# General
|
||||||
|
.DS_Store
|
||||||
|
.idea
|
||||||
|
.vscode
|
@@ -39,7 +39,7 @@ func namedValueToLabels(named []driver.NamedValue) []interface{} {
|
|||||||
|
|
||||||
// getCallerName get the name of the function A where A() -> B() -> GetFunctionCallerName()
|
// getCallerName get the name of the function A where A() -> B() -> GetFunctionCallerName()
|
||||||
func getCallerName() string {
|
func getCallerName() string {
|
||||||
pc, _, _, ok := runtime.Caller(2)
|
pc, _, _, ok := runtime.Caller(3)
|
||||||
details := runtime.FuncForPC(pc)
|
details := runtime.FuncForPC(pc)
|
||||||
var callerName string
|
var callerName string
|
||||||
if ok && details != nil {
|
if ok && details != nil {
|
||||||
|
40
conn.go
40
conn.go
@@ -6,7 +6,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"go.unistack.org/micro/v3/tracer"
|
requestid "go.unistack.org/micro-wrapper-requestid/v4"
|
||||||
|
"go.unistack.org/micro/v4/tracer"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -99,12 +100,11 @@ func (w *wrapperConn) Begin() (driver.Tx, error) {
|
|||||||
// BeginTx implements driver.ConnBeginTx BeginTx
|
// BeginTx implements driver.ConnBeginTx BeginTx
|
||||||
func (w *wrapperConn) BeginTx(ctx context.Context, opts driver.TxOptions) (driver.Tx, error) {
|
func (w *wrapperConn) BeginTx(ctx context.Context, opts driver.TxOptions) (driver.Tx, error) {
|
||||||
nctx, span := w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span := w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
span.AddLabels("method", "BeginTx")
|
span.AddLabels("db.method", "BeginTx")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.query", name)
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
} else {
|
span.AddLabels("x-request-id", id)
|
||||||
name = getCallerName()
|
|
||||||
}
|
}
|
||||||
labels := []string{labelMethod, "BeginTx", labelQuery, name}
|
labels := []string{labelMethod, "BeginTx", labelQuery, name}
|
||||||
|
|
||||||
@@ -178,14 +178,12 @@ func (w *wrapperConn) PrepareContext(ctx context.Context, query string) (driver.
|
|||||||
} else {
|
} else {
|
||||||
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
}
|
}
|
||||||
span.AddLabels("method", "PrepareContext")
|
span.AddLabels("db.method", "PrepareContext")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.query", name)
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
} else {
|
span.AddLabels("x-request-id", id)
|
||||||
name = getCallerName()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
labels := []string{labelMethod, "PrepareContext", labelQuery, name}
|
labels := []string{labelMethod, "PrepareContext", labelQuery, name}
|
||||||
conn, ok := w.conn.(driver.ConnPrepareContext)
|
conn, ok := w.conn.(driver.ConnPrepareContext)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -259,12 +257,11 @@ func (w *wrapperConn) ExecContext(ctx context.Context, query string, args []driv
|
|||||||
} else {
|
} else {
|
||||||
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
}
|
}
|
||||||
span.AddLabels("method", "ExecContext")
|
span.AddLabels("db.method", "ExecContext")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.query", name)
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
} else {
|
span.AddLabels("x-request-id", id)
|
||||||
name = getCallerName()
|
|
||||||
}
|
}
|
||||||
defer span.Finish()
|
defer span.Finish()
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
@@ -382,12 +379,11 @@ func (w *wrapperConn) QueryContext(ctx context.Context, query string, args []dri
|
|||||||
} else {
|
} else {
|
||||||
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
}
|
}
|
||||||
span.AddLabels("method", "QueryContext")
|
span.AddLabels("db.method", "QueryContext")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.statement", name)
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
} else {
|
span.AddLabels("x-request-id", id)
|
||||||
name = getCallerName()
|
|
||||||
}
|
}
|
||||||
defer span.Finish()
|
defer span.Finish()
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
|
7
go.mod
7
go.mod
@@ -1,5 +1,8 @@
|
|||||||
module go.unistack.org/micro-wrapper-sql/v3
|
module go.unistack.org/micro-wrapper-sql/v4
|
||||||
|
|
||||||
go 1.20
|
go 1.20
|
||||||
|
|
||||||
require go.unistack.org/micro/v3 v3.10.25
|
require (
|
||||||
|
go.unistack.org/micro-wrapper-requestid/v4 v4.0.3
|
||||||
|
go.unistack.org/micro/v4 v4.0.8
|
||||||
|
)
|
||||||
|
6
go.sum
6
go.sum
@@ -1,2 +1,4 @@
|
|||||||
go.unistack.org/micro/v3 v3.10.25 h1:A0epdZHOqjnXx103wwFhPKgmvVVbScvfbmn3HmHz1wE=
|
go.unistack.org/micro-wrapper-requestid/v4 v4.0.3 h1:LhloK8yLLsov4XiKw9i20wdBUfvWCG1TaBNgNVtNa8Q=
|
||||||
go.unistack.org/micro/v3 v3.10.25/go.mod h1:ALkeXpqChYDjx8KPi7tz9mmIyOnob6nlNswsg8BnZjQ=
|
go.unistack.org/micro-wrapper-requestid/v4 v4.0.3/go.mod h1:vRIkCMyhbBVus1xQlr/Fef65n07znrBGkpSCpjkEiqk=
|
||||||
|
go.unistack.org/micro/v4 v4.0.8 h1:D5n18+fhyRjJkKSPdIcW1ratWUuzy9TzrS5ud/qImWg=
|
||||||
|
go.unistack.org/micro/v4 v4.0.8/go.mod h1:QT3gOIE4qGgBiQGm2Pad/62Sl5R53QfrgYHD448aX14=
|
||||||
|
18
options.go
18
options.go
@@ -2,12 +2,14 @@ package wrapper
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"go.unistack.org/micro/v3/logger"
|
"go.unistack.org/micro/v4/logger"
|
||||||
"go.unistack.org/micro/v3/meter"
|
"go.unistack.org/micro/v4/meter"
|
||||||
"go.unistack.org/micro/v3/tracer"
|
"go.unistack.org/micro/v4/tracer"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -17,8 +19,8 @@ var (
|
|||||||
DefaultMeterMetricPrefix = "micro_sql_"
|
DefaultMeterMetricPrefix = "micro_sql_"
|
||||||
// DefaultLoggerObserver used to prepare labels for logger
|
// DefaultLoggerObserver used to prepare labels for logger
|
||||||
DefaultLoggerObserver = func(ctx context.Context, method string, query string, td time.Duration, err error) []interface{} {
|
DefaultLoggerObserver = func(ctx context.Context, method string, query string, td time.Duration, err error) []interface{} {
|
||||||
labels := []interface{}{"method", method, "took", fmt.Sprintf("%v", td)}
|
labels := []interface{}{"db.method", method, "took", fmt.Sprintf("%v", td)}
|
||||||
if err != nil {
|
if err != nil && !errors.Is(err, sql.ErrNoRows) {
|
||||||
labels = append(labels, "error", err.Error())
|
labels = append(labels, "error", err.Error())
|
||||||
}
|
}
|
||||||
if query != labelUnknown {
|
if query != labelUnknown {
|
||||||
@@ -44,8 +46,8 @@ var (
|
|||||||
meterRequestDurationSeconds = "request_duration_seconds"
|
meterRequestDurationSeconds = "request_duration_seconds"
|
||||||
|
|
||||||
labelUnknown = "unknown"
|
labelUnknown = "unknown"
|
||||||
labelQuery = "query"
|
labelQuery = "db_statement"
|
||||||
labelMethod = "method"
|
labelMethod = "db_method"
|
||||||
labelStatus = "status"
|
labelStatus = "status"
|
||||||
labelSuccess = "success"
|
labelSuccess = "success"
|
||||||
labelFailure = "failure"
|
labelFailure = "failure"
|
||||||
@@ -180,5 +182,5 @@ func getQueryName(ctx context.Context) string {
|
|||||||
if v, ok := ctx.Value(queryNameKey{}).(string); ok && v != labelUnknown {
|
if v, ok := ctx.Value(queryNameKey{}).(string); ok && v != labelUnknown {
|
||||||
return v
|
return v
|
||||||
}
|
}
|
||||||
return ""
|
return getCallerName()
|
||||||
}
|
}
|
||||||
|
2
stats.go
2
stats.go
@@ -1,4 +1,4 @@
|
|||||||
package wrapper // import "go.unistack.org/micro-wrapper-sql/v3"
|
package wrapper // import "go.unistack.org/micro-wrapper-sql/v4"
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
25
stmt.go
25
stmt.go
@@ -6,7 +6,8 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"go.unistack.org/micro/v3/tracer"
|
requestid "go.unistack.org/micro-wrapper-requestid/v4"
|
||||||
|
"go.unistack.org/micro/v4/tracer"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -140,17 +141,16 @@ func (w *wrapperStmt) ExecContext(ctx context.Context, args []driver.NamedValue)
|
|||||||
} else {
|
} else {
|
||||||
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
}
|
}
|
||||||
span.AddLabels("method", "ExecContext")
|
span.AddLabels("db.method", "ExecContext")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.query", name)
|
|
||||||
} else {
|
|
||||||
name = getCallerName()
|
|
||||||
}
|
|
||||||
defer span.Finish()
|
defer span.Finish()
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
span.AddLabels("db.args", fmt.Sprintf("%v", namedValueToLabels(args)))
|
span.AddLabels("db.args", fmt.Sprintf("%v", namedValueToLabels(args)))
|
||||||
}
|
}
|
||||||
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
|
span.AddLabels("x-request-id", id)
|
||||||
|
}
|
||||||
labels := []string{labelMethod, "ExecContext", labelQuery, name}
|
labels := []string{labelMethod, "ExecContext", labelQuery, name}
|
||||||
|
|
||||||
if conn, ok := w.stmt.(driver.StmtExecContext); ok {
|
if conn, ok := w.stmt.(driver.StmtExecContext); ok {
|
||||||
@@ -214,17 +214,16 @@ func (w *wrapperStmt) QueryContext(ctx context.Context, args []driver.NamedValue
|
|||||||
} else {
|
} else {
|
||||||
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
nctx, span = w.opts.Tracer.Start(ctx, "sdk.database", tracer.WithSpanKind(tracer.SpanKindClient))
|
||||||
}
|
}
|
||||||
span.AddLabels("method", "QueryContext")
|
span.AddLabels("db.method", "QueryContext")
|
||||||
name := getQueryName(ctx)
|
name := getQueryName(ctx)
|
||||||
if name != "" {
|
span.AddLabels("db.statement", name)
|
||||||
span.AddLabels("db.query", name)
|
|
||||||
} else {
|
|
||||||
name = getCallerName()
|
|
||||||
}
|
|
||||||
defer span.Finish()
|
defer span.Finish()
|
||||||
if len(args) > 0 {
|
if len(args) > 0 {
|
||||||
span.AddLabels("db.args", fmt.Sprintf("%v", namedValueToLabels(args)))
|
span.AddLabels("db.args", fmt.Sprintf("%v", namedValueToLabels(args)))
|
||||||
}
|
}
|
||||||
|
if id, ok := ctx.Value(requestid.XRequestIDKey).(string); ok {
|
||||||
|
span.AddLabels("x-request-id", id)
|
||||||
|
}
|
||||||
labels := []string{labelMethod, "QueryContext", labelQuery, name}
|
labels := []string{labelMethod, "QueryContext", labelQuery, name}
|
||||||
if conn, ok := w.stmt.(driver.StmtQueryContext); ok {
|
if conn, ok := w.stmt.(driver.StmtQueryContext); ok {
|
||||||
ts := time.Now()
|
ts := time.Now()
|
||||||
|
Reference in New Issue
Block a user