367afb544c
* feat: support ptr of ptr of ... in mapping * feat: support ptr of ptr of time.Duration in mapping * feat: support ptr of ptr of json.Number in mapping * chore: improve setting in mapping * feat: support ptr of ptr encoding.TextUnmarshaler in mapping * chore: add more tests * fix: string ptr * chore: update tests
38 lines
814 B
Go
38 lines
814 B
Go
package sqlx
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/zeromicro/go-zero/core/trace"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
oteltrace "go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
var sqlAttributeKey = attribute.Key("sql.method")
|
|
|
|
func startSpan(ctx context.Context, method string) (context.Context, oteltrace.Span) {
|
|
tracer := otel.Tracer(trace.TraceName)
|
|
start, span := tracer.Start(ctx,
|
|
spanName,
|
|
oteltrace.WithSpanKind(oteltrace.SpanKindClient),
|
|
)
|
|
span.SetAttributes(sqlAttributeKey.String(method))
|
|
|
|
return start, span
|
|
}
|
|
|
|
func endSpan(span oteltrace.Span, err error) {
|
|
defer span.End()
|
|
|
|
if err == nil || err == sql.ErrNoRows {
|
|
span.SetStatus(codes.Ok, "")
|
|
return
|
|
}
|
|
|
|
span.SetStatus(codes.Error, err.Error())
|
|
span.RecordError(err)
|
|
}
|