micro/tracer/context.go
Vasiliy Tolstov 6f6f850af6
Some checks failed
lint / lint (pull_request) Failing after 1m31s
pr / test (pull_request) Failing after 2m37s
move options to dedicated package
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2023-07-29 00:40:58 +03:00

49 lines
1.0 KiB
Go

// Package tracer provides an interface for distributed tracing
package tracer
import (
"context"
)
type tracerKey struct{}
// FromContext returns a tracer from context
func FromContext(ctx context.Context) (Tracer, bool) {
if ctx == nil {
return nil, false
}
if tracer, ok := ctx.Value(tracerKey{}).(Tracer); ok {
return tracer, true
}
return nil, false
}
// NewContext saves the tracer in the context
func NewContext(ctx context.Context, tracer Tracer) context.Context {
if ctx == nil {
ctx = context.Background()
}
return context.WithValue(ctx, tracerKey{}, tracer)
}
type spanKey struct{}
// SpanFromContext returns a span from context
func SpanFromContext(ctx context.Context) (Span, bool) {
if ctx == nil {
return nil, false
}
if span, ok := ctx.Value(spanKey{}).(Span); ok {
return span, true
}
return nil, false
}
// NewSpanContext saves the span in the context
func NewSpanContext(ctx context.Context, span Span) context.Context {
if ctx == nil {
ctx = context.Background()
}
return context.WithValue(ctx, spanKey{}, span)
}