mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-18 00:47:48 +08:00
92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
package tracing
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptrace"
|
|
"slices"
|
|
|
|
"github.com/moby/buildkit/util/bklog"
|
|
"github.com/moby/buildkit/util/stack"
|
|
"github.com/pkg/errors"
|
|
"go.opentelemetry.io/contrib/instrumentation/net/http/httptrace/otelhttptrace"
|
|
"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
"go.opentelemetry.io/otel/propagation"
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.26.0"
|
|
"go.opentelemetry.io/otel/trace"
|
|
"go.opentelemetry.io/otel/trace/noop"
|
|
)
|
|
|
|
// StartSpan starts a new span as a child of the span in context.
|
|
// If there is no span in context then this is a no-op.
|
|
func StartSpan(ctx context.Context, operationName string, opts ...trace.SpanStartOption) (trace.Span, context.Context) {
|
|
parent := trace.SpanFromContext(ctx)
|
|
tracer := noop.NewTracerProvider().Tracer("")
|
|
if parent != nil && parent.SpanContext().IsValid() {
|
|
tracer = parent.TracerProvider().Tracer("")
|
|
}
|
|
ctx, span := tracer.Start(ctx, operationName, opts...)
|
|
ctx = bklog.WithLogger(ctx, bklog.GetLogger(ctx).WithField("span", operationName))
|
|
return span, ctx
|
|
}
|
|
|
|
func hasStacktrace(err error) bool {
|
|
switch e := err.(type) {
|
|
case interface{ StackTrace() *stack.Stack }:
|
|
return true
|
|
case interface{ StackTrace() errors.StackTrace }:
|
|
return true
|
|
case interface{ Unwrap() error }:
|
|
return hasStacktrace(e.Unwrap())
|
|
case interface{ Unwrap() []error }:
|
|
if slices.ContainsFunc(e.Unwrap(), hasStacktrace) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// FinishWithError finalizes the span and sets the error if one is passed
|
|
func FinishWithError(span trace.Span, err error) {
|
|
if err != nil {
|
|
span.RecordError(err)
|
|
if hasStacktrace(err) {
|
|
span.SetAttributes(attribute.String(string(semconv.ExceptionStacktraceKey), fmt.Sprintf("%+v", stack.Formatter(err))))
|
|
}
|
|
span.SetStatus(codes.Error, err.Error())
|
|
}
|
|
span.End()
|
|
}
|
|
|
|
// ContextWithSpanFromContext sets the tracing span of a context from other
|
|
// context if one is not already set. Alternative would be
|
|
// context.WithoutCancel() that would copy the context but reset ctx.Done
|
|
func ContextWithSpanFromContext(ctx, ctx2 context.Context) context.Context {
|
|
// if already is a span then noop
|
|
if span := trace.SpanFromContext(ctx); span.SpanContext().IsValid() {
|
|
return ctx
|
|
}
|
|
if span := trace.SpanFromContext(ctx2); span != nil && span.SpanContext().IsValid() {
|
|
return trace.ContextWithSpan(ctx, span)
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
var DefaultTransport = NewTransport(http.DefaultTransport)
|
|
|
|
var DefaultClient = &http.Client{
|
|
Transport: DefaultTransport,
|
|
}
|
|
|
|
func NewTransport(rt http.RoundTripper) http.RoundTripper {
|
|
return otelhttp.NewTransport(rt,
|
|
otelhttp.WithPropagators(propagation.NewCompositeTextMapPropagator(propagation.TraceContext{}, propagation.Baggage{})),
|
|
otelhttp.WithClientTrace(func(ctx context.Context) *httptrace.ClientTrace {
|
|
return otelhttptrace.NewClientTrace(ctx, otelhttptrace.WithoutSubSpans())
|
|
}),
|
|
)
|
|
}
|