fix wrapper

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2022-12-24 18:18:42 +03:00
parent 26a2d18766
commit 0be584ef0d

View File

@ -14,95 +14,95 @@ import (
var ( var (
DefaultClientCallObserver = func(ctx context.Context, req client.Request, rsp interface{}, opts []client.CallOption, sp tracer.Span, err error) { DefaultClientCallObserver = func(ctx context.Context, req client.Request, rsp interface{}, opts []client.CallOption, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint())) sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultClientStreamObserver = func(ctx context.Context, req client.Request, opts []client.CallOption, stream client.Stream, sp tracer.Span, err error) { DefaultClientStreamObserver = func(ctx context.Context, req client.Request, opts []client.CallOption, stream client.Stream, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint())) sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultClientPublishObserver = func(ctx context.Context, msg client.Message, opts []client.PublishOption, sp tracer.Span, err error) { DefaultClientPublishObserver = func(ctx context.Context, msg client.Message, opts []client.PublishOption, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("Pub to %s", msg.Topic())) sp.SetName(fmt.Sprintf("Pub to %s", msg.Topic()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultServerHandlerObserver = func(ctx context.Context, req server.Request, rsp interface{}, sp tracer.Span, err error) { DefaultServerHandlerObserver = func(ctx context.Context, req server.Request, rsp interface{}, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint())) sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromIncomingContext(ctx); ok { if md, ok := metadata.FromIncomingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultServerSubscriberObserver = func(ctx context.Context, msg server.Message, sp tracer.Span, err error) { DefaultServerSubscriberObserver = func(ctx context.Context, msg server.Message, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("Sub from %s", msg.Topic())) sp.SetName(fmt.Sprintf("Sub from %s", msg.Topic()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromIncomingContext(ctx); ok { if md, ok := metadata.FromIncomingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultClientCallFuncObserver = func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions, sp tracer.Span, err error) { DefaultClientCallFuncObserver = func(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions, sp tracer.Span, err error) {
sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint())) sp.SetName(fmt.Sprintf("%s.%s", req.Service(), req.Endpoint()))
var labels []tracer.Label var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]tracer.Label, 0, len(md)) labels = make([]interface{}, 0, len(md))
for k, v := range md { for k, v := range md {
labels = append(labels, tracer.LabelString(k, v)) labels = append(labels, k, v)
} }
} }
if err != nil { if err != nil {
labels = append(labels, tracer.LabelBool("error", true)) labels = append(labels, "error", true)
} }
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
DefaultSkipEndpoints = []string{"Meter.Metrics"} DefaultSkipEndpoints = []string{"Meter.Metrics", "Health.Live", "Health.Ready", "Health.Version"}
) )
type tWrapper struct { type tWrapper struct {