add trace headers

johnbellone · Jun 18, 2021 · eae2756 · eae2756
1 parent 50f26b0
commit eae2756
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/client_interceptors.go b/client_interceptors.go
@@ -2,6 +2,7 @@ package grpc_sentry
 
 import (
 	"context"
+	"google.golang.org/grpc/metadata"
 
 	"github.com/getsentry/sentry-go"
 
@@ -23,6 +24,14 @@ func UnaryClientInterceptor(opts ...Option) grpc.UnaryClientInterceptor {
 			ctx = sentry.SetHubOnContext(ctx, hub)
 		}
 
+		span := sentry.StartSpan(ctx, "grpc.client")
+		ctx = span.Context()
+		md := metadata.Pairs("sentry-trace", span.ToSentryTrace())
+		ctx = metadata.NewOutgoingContext(ctx, md)
+		defer span.Finish()
+
+		hub.Scope().SetTransaction(method)
+
 		err := invoker(ctx, method, req, reply, cc, callOpts...)
 
 		if err != nil && o.ReportOn(err) {
@@ -48,6 +57,14 @@ func StreamClientInterceptor(opts ...Option) grpc.StreamClientInterceptor {
 			ctx = sentry.SetHubOnContext(ctx, hub)
 		}
 
+		span := sentry.StartSpan(ctx, "grpc.client")
+		ctx = span.Context()
+		md := metadata.Pairs("sentry-trace", span.ToSentryTrace())
+		ctx = metadata.NewOutgoingContext(ctx, md)
+		defer span.Finish()
+
+		hub.Scope().SetTransaction(method)
+
 		clientStream, err := streamer(ctx, desc, cc, method, callOpts...)
 
 		if err != nil && o.ReportOn(err) {