mirror of
https://gitee.com/johng/gf
synced 2026-07-03 03:39:35 +08:00
106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
// Package gtrace provides convenience wrapping functionality for tracing feature using OpenTelemetry.
|
|
package gtrace
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/container/gvar"
|
|
"github.com/gogf/gf/net/gipv4"
|
|
"github.com/gogf/gf/text/gstr"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/baggage"
|
|
"go.opentelemetry.io/otel/label"
|
|
"go.opentelemetry.io/otel/trace"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
intranetIps, _ = gipv4.GetIntranetIpArray()
|
|
hostname, _ = os.Hostname()
|
|
)
|
|
|
|
// IsActivated checks and returns if tracing feature is activated.
|
|
func IsActivated(ctx context.Context) bool {
|
|
return GetTraceId(ctx) != ""
|
|
}
|
|
|
|
// CommonLabels returns common used attribute labels:
|
|
// ip.intranet, hostname.
|
|
func CommonLabels() []label.KeyValue {
|
|
return []label.KeyValue{
|
|
label.String(`ip.intranet`, gstr.Join(intranetIps, ",")),
|
|
label.String(`hostname`, hostname),
|
|
}
|
|
}
|
|
|
|
// Tracer is a short function for retrieve Tracer.
|
|
func Tracer(name ...string) trace.Tracer {
|
|
tracerName := ""
|
|
if len(name) > 0 {
|
|
tracerName = name[0]
|
|
}
|
|
return otel.Tracer(tracerName)
|
|
}
|
|
|
|
// GetTraceId retrieves and returns TraceId from context.
|
|
// It returns an empty string is tracing feature is not activated.
|
|
func GetTraceId(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
traceId := trace.SpanContextFromContext(ctx).TraceID
|
|
if traceId.IsValid() {
|
|
return traceId.String()
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// GetSpanId retrieves and returns SpanId from context.
|
|
// It returns an empty string is tracing feature is not activated.
|
|
func GetSpanId(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
spanId := trace.SpanContextFromContext(ctx).SpanID
|
|
if spanId.IsValid() {
|
|
return spanId.String()
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// SetBaggageValue is a convenient function for adding one key-value pair to baggage.
|
|
// Note that it uses label.Any to set the key-value pair.
|
|
func SetBaggageValue(ctx context.Context, key string, value interface{}) context.Context {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
return baggage.ContextWithValues(ctx, label.Any(key, value))
|
|
}
|
|
|
|
// SetBaggageMap is a convenient function for adding map key-value pairs to baggage.
|
|
// Note that it uses label.Any to set the key-value pair.
|
|
func SetBaggageMap(ctx context.Context, data map[string]interface{}) context.Context {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
pairs := make([]label.KeyValue, 0)
|
|
for k, v := range data {
|
|
pairs = append(pairs, label.Any(k, v))
|
|
}
|
|
return baggage.ContextWithValues(ctx, pairs...)
|
|
}
|
|
|
|
// GetBaggageVar retrieves value and returns a *gvar.Var for specified key from baggage.
|
|
func GetBaggageVar(ctx context.Context, key string) *gvar.Var {
|
|
if ctx == nil {
|
|
return gvar.New(nil)
|
|
}
|
|
value := baggage.Value(ctx, label.Key(key))
|
|
return gvar.New(value.AsInterface())
|
|
}
|