代码拉取完成,页面将自动刷新
同步操作将从 menuiis/gkit 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
package trace
import (
"context"
"fmt"
"gitee.com/menciis/gkit/errors"
"gitee.com/menciis/gkit/options"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"go.opentelemetry.io/otel/propagation"
"go.opentelemetry.io/otel/trace"
"google.golang.org/protobuf/proto"
)
type Tracer struct {
tracer trace.Tracer
kind trace.SpanKind
opt *config
}
// NewTracer 创建追踪器
func NewTracer(kind trace.SpanKind, opts ...options.Option) *Tracer {
cfg := config{
propagator: propagation.NewCompositeTextMapPropagator(Metadata{}, propagation.Baggage{}, propagation.TraceContext{}),
}
for _, o := range opts {
o(&cfg)
}
if cfg.tracerProvider != nil {
otel.SetTracerProvider(cfg.tracerProvider)
}
switch kind {
case trace.SpanKindClient:
return &Tracer{tracer: otel.Tracer("gkit"), kind: kind, opt: &cfg}
case trace.SpanKindServer:
return &Tracer{tracer: otel.Tracer("gkit"), kind: kind, opt: &cfg}
default:
panic(fmt.Sprintf("unsupported span kind: %v", kind))
}
}
// Start 开始追踪
func (t *Tracer) Start(ctx context.Context, operation string, carrier propagation.TextMapCarrier) (context.Context, trace.Span) {
if t.kind == trace.SpanKindServer {
ctx = t.opt.propagator.Extract(ctx, carrier)
}
ctx, span := t.tracer.Start(ctx,
operation,
trace.WithSpanKind(t.kind),
)
if t.kind == trace.SpanKindClient {
t.opt.propagator.Inject(ctx, carrier)
}
return ctx, span
}
// End 完成追踪
func (t *Tracer) End(ctx context.Context, span trace.Span, m interface{}, err error) {
if err != nil {
span.RecordError(err)
if e := errors.FromError(err); e != nil {
span.SetAttributes(attribute.Key("rpc.status_code").Int64(int64(e.Code)))
}
span.SetStatus(codes.Error, err.Error())
} else {
span.SetStatus(codes.Ok, "OK")
}
if p, ok := m.(proto.Message); ok {
if t.kind == trace.SpanKindServer {
span.SetAttributes(attribute.Key("send_msg.size").Int(proto.Size(p)))
} else {
span.SetAttributes(attribute.Key("recv_msg.size").Int(proto.Size(p)))
}
}
span.End()
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。