You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
903 B
Go
36 lines
903 B
Go
package mon
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/zeromicro/go-zero/core/trace"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
oteltrace "go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
var mongoCmdAttributeKey = attribute.Key("mongo.cmd")
|
|
|
|
func startSpan(ctx context.Context, cmd string) (context.Context, oteltrace.Span) {
|
|
tracer := trace.TracerFromContext(ctx)
|
|
ctx, span := tracer.Start(ctx, spanName, oteltrace.WithSpanKind(oteltrace.SpanKindClient))
|
|
span.SetAttributes(mongoCmdAttributeKey.String(cmd))
|
|
|
|
return ctx, span
|
|
}
|
|
|
|
func endSpan(span oteltrace.Span, err error) {
|
|
defer span.End()
|
|
|
|
if err == nil || errors.Is(err, mongo.ErrNoDocuments) ||
|
|
errors.Is(err, mongo.ErrNilValue) || errors.Is(err, mongo.ErrNilDocument) {
|
|
span.SetStatus(codes.Ok, "")
|
|
return
|
|
}
|
|
|
|
span.SetStatus(codes.Error, err.Error())
|
|
span.RecordError(err)
|
|
}
|