fix lint

glebnaz · Nov 13, 2023 · 9baf86e · 9baf86e
1 parent be86c26
commit 9baf86e
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/log/grpc_client_mw.go b/log/grpc_client_mw.go
@@ -2,6 +2,7 @@ package log
 
 import (
 	"context"
+
 	"google.golang.org/grpc"
 )
 

diff --git a/metrics/grpc_mw_client.go b/metrics/grpc_mw_client.go
@@ -2,9 +2,10 @@ package metrics
 
 import (
 	"context"
+	"time"
+
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/status"
-	"time"
 )
 
 // ClientMetricsUnaryInterceptor returns a new unary client interceptor for

diff --git a/trace/client_mw.go b/trace/client_mw.go
@@ -2,6 +2,7 @@ package trace
 
 import (
 	"context"
+
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/metadata"
 )
@@ -27,6 +28,7 @@ func ClientBaseUnaryInterceptor(from string) grpc.UnaryClientInterceptor {
 			md, ok := metadata.FromOutgoingContext(ctx)
 			if ok {
 				md = metadata.Join(md, metadata.New(map[string]string{string(simpleReqID): reqID}))
+				ctx = metadata.NewOutgoingContext(ctx, md)
 			} else {
 				ctx = metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{string(simpleReqID): reqID}))
 			}
@@ -35,14 +37,15 @@ func ClientBaseUnaryInterceptor(from string) grpc.UnaryClientInterceptor {
 			md, ok := metadata.FromOutgoingContext(ctx)
 			if ok {
 				md = metadata.Join(md, metadata.New(map[string]string{string(simpleReqID): reqID}))
+				ctx = metadata.NewOutgoingContext(ctx, md)
 			}
 			ctx = metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{string(simpleReqID): reqID}))
 		}
 
 		md, ok := metadata.FromOutgoingContext(ctx)
 		if !ok {
-			md := metadata.New(map[string]string{string(FromKey): from})
-			ctx = metadata.NewOutgoingContext(ctx, md)
+			mdNew := metadata.New(map[string]string{string(FromKey): from})
+			ctx = metadata.NewOutgoingContext(ctx, mdNew)
 		} else {
 			md = metadata.Join(md, metadata.New(map[string]string{string(FromKey): from}))
 			ctx = metadata.NewOutgoingContext(ctx, md)

diff --git a/trace/context.go b/trace/context.go
@@ -2,6 +2,7 @@ package trace
 
 import (
 	"context"
+
 	"github.com/google/uuid"
 	"google.golang.org/grpc/metadata"
 )