unistack-org/micro
3
0
Fork 3
Code Issues 16 Pull Requests 2 Actions Packages Projects Releases 239 Wiki Activity

tracer: add span kind

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2023-01-18 00:25:15 +03:00
parent f59023d741
commit db3fc42204
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tracer/wrapper/wrapper.go
View File

@ -24,7 +24,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -40,7 +40,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -56,7 +56,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "publisher") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -72,7 +72,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "server") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -88,7 +88,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "subscriber") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -104,7 +104,7 @@ var (
if err != nil { if err != nil {
labels = append(labels, "error", true) labels = append(labels, "error", true)
} }
labels = append(labels, "type", "client") labels = append(labels, "kind", sp.Kind())
sp.SetLabels(labels...) sp.SetLabels(labels...)
} }
@ -237,7 +237,7 @@ func (ot *tWrapper) Call(ctx context.Context, req client.Request, rsp interface{
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindClient))
} }
defer sp.Finish() defer sp.Finish()
@ -260,7 +260,7 @@ func (ot *tWrapper) Stream(ctx context.Context, req client.Request, opts ...clie
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindClient))
} }
defer sp.Finish() defer sp.Finish()
@ -276,7 +276,7 @@ func (ot *tWrapper) Stream(ctx context.Context, req client.Request, opts ...clie
func (ot *tWrapper) Publish(ctx context.Context, msg client.Message, opts ...client.PublishOption) error { func (ot *tWrapper) Publish(ctx context.Context, msg client.Message, opts ...client.PublishOption) error {
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindProducer))
} }
defer sp.Finish() defer sp.Finish()
@ -299,7 +299,7 @@ func (ot *tWrapper) ServerHandler(ctx context.Context, req server.Request, rsp i
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindServer))
} }
defer sp.Finish() defer sp.Finish()
@ -315,7 +315,7 @@ func (ot *tWrapper) ServerHandler(ctx context.Context, req server.Request, rsp i
func (ot *tWrapper) ServerSubscriber(ctx context.Context, msg server.Message) error { func (ot *tWrapper) ServerSubscriber(ctx context.Context, msg server.Message) error {
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindConsumer))
} }
defer sp.Finish() defer sp.Finish()
@ -362,7 +362,7 @@ func (ot *tWrapper) ClientCallFunc(ctx context.Context, addr string, req client.
sp, ok := tracer.SpanFromContext(ctx) sp, ok := tracer.SpanFromContext(ctx)
if !ok { if !ok {
ctx, sp = ot.opts.Tracer.Start(ctx, "") ctx, sp = ot.opts.Tracer.Start(ctx, "", tracer.WithSpanKind(tracer.SpanKindClient))
} }
defer sp.Finish() defer sp.Finish()