unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 331 Wiki Activity

tracer/wrapper: fix observers

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов
2023-01-17 23:31:27 +03:00
parent ccbf23688b
commit 2aaa0a74d8
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tracer/wrapper/wrapper.go
View File

@@ -13,10 +13,10 @@ 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("Call %s.%s", req.Service(), req.Endpoint()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md)+1)
for k, v := range md { for k, v := range md {
labels = append(labels, k, v) labels = append(labels, k, v)
} }
@@ -24,11 +24,12 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client")
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("Stream %s.%s", req.Service(), req.Endpoint()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md))
@@ -39,11 +40,12 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client")
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("Publish %s", msg.Topic()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md))
@@ -54,11 +56,12 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "publisher")
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("Handler %s.%s", req.Service(), req.Method()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromIncomingContext(ctx); ok { if md, ok := metadata.FromIncomingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md))
@@ -69,11 +72,12 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "server")
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("Subscriber %s", msg.Topic()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromIncomingContext(ctx); ok { if md, ok := metadata.FromIncomingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md))
@@ -84,11 +88,12 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "subscriber")
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("Call %s.%s", req.Service(), req.Endpoint()))
var labels []interface{} var labels []interface{}
if md, ok := metadata.FromOutgoingContext(ctx); ok { if md, ok := metadata.FromOutgoingContext(ctx); ok {
labels = make([]interface{}, 0, len(md)) labels = make([]interface{}, 0, len(md))
@@ -99,6 +104,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client")
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }