mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 12:17:53 +08:00
76 lines
2.1 KiB
Go
76 lines
2.1 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
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.opentelemetry.io/otel/baggage"
|
|
|
|
"github.com/gogf/gf/v2/container/gmap"
|
|
"github.com/gogf/gf/v2/container/gvar"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// Baggage holds the data through all tracing spans.
|
|
type Baggage struct {
|
|
ctx context.Context
|
|
}
|
|
|
|
// NewBaggage creates and returns a new Baggage object from given tracing context.
|
|
func NewBaggage(ctx context.Context) *Baggage {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
return &Baggage{
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
// Ctx returns the context that Baggage holds.
|
|
func (b *Baggage) Ctx() context.Context {
|
|
return b.ctx
|
|
}
|
|
|
|
// SetValue is a convenient function for adding one key-value pair to baggage.
|
|
// Note that it uses attribute.Any to set the key-value pair.
|
|
func (b *Baggage) SetValue(key string, value interface{}) context.Context {
|
|
member, _ := baggage.NewMember(key, gconv.String(value))
|
|
bag, _ := baggage.New(member)
|
|
b.ctx = baggage.ContextWithBaggage(b.ctx, bag)
|
|
return b.ctx
|
|
}
|
|
|
|
// SetMap is a convenient function for adding map key-value pairs to baggage.
|
|
// Note that it uses attribute.Any to set the key-value pair.
|
|
func (b *Baggage) SetMap(data map[string]interface{}) context.Context {
|
|
members := make([]baggage.Member, 0)
|
|
for k, v := range data {
|
|
member, _ := baggage.NewMember(k, gconv.String(v))
|
|
members = append(members, member)
|
|
}
|
|
bag, _ := baggage.New(members...)
|
|
b.ctx = baggage.ContextWithBaggage(b.ctx, bag)
|
|
return b.ctx
|
|
}
|
|
|
|
// GetMap retrieves and returns the baggage values as map.
|
|
func (b *Baggage) GetMap() *gmap.StrAnyMap {
|
|
m := gmap.NewStrAnyMap()
|
|
members := baggage.FromContext(b.ctx).Members()
|
|
for i := range members {
|
|
m.Set(members[i].Key(), members[i].Value())
|
|
}
|
|
return m
|
|
}
|
|
|
|
// GetVar retrieves value and returns a *gvar.Var for specified key from baggage.
|
|
func (b *Baggage) GetVar(key string) *gvar.Var {
|
|
value := baggage.FromContext(b.ctx).Member(key).Value()
|
|
return gvar.New(value)
|
|
}
|