fix(grpc): fix issue with client without metadata

squzy · Jan 29, 2021 · 41df407 · 41df407
1 parent beba026
commit 41df407
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/integrations/grpc/grpc.go b/integrations/grpc/grpc.go
@@ -82,10 +82,10 @@ func NewServerStreamInterceptor(app *core.Application) grpc.StreamServerIntercep
 		md, ok := metadata.FromIncomingContext(ss.Context())
 		var parent *core.Transaction
 		if ok {
-			id := md.Get(app.GetTracingHeader())[0]
-			if id != "" {
+			metaArr := md.Get(app.GetTracingHeader())
+			if len(metaArr) > 0 && metaArr[0] != "" {
 				parent = &core.Transaction{
-					Id: id,
+					Id: metaArr[0],
 				}
 			}
 		}
@@ -108,10 +108,10 @@ func NewServerUnaryInterceptor(app *core.Application) grpc.UnaryServerIntercepto
 		md, ok := metadata.FromIncomingContext(ctx)
 		var parent *core.Transaction
 		if ok {
-			id := md.Get(app.GetTracingHeader())[0]
-			if id != "" {
+			metaArr := md.Get(app.GetTracingHeader())
+			if len(metaArr) > 0 && metaArr[0] != "" {
 				parent = &core.Transaction{
-					Id: id,
+					Id: metaArr[0],
 				}
 			}
 		}