@@ -55,8 +55,7 @@ func GRPCClientTrace(options ...TracerOption) kitgrpc.ClientOption {
55
55
clientFinalizer := kitgrpc .ClientFinalizer (
56
56
func (ctx context.Context , err error ) {
57
57
if span := trace .FromContext (ctx ); span != nil {
58
- s , ok := status .FromError (err )
59
- if ok {
58
+ if s , ok := status .FromError (err ); ok {
60
59
span .SetStatus (trace.Status {Code : int32 (s .Code ()), Message : s .Message ()})
61
60
} else {
62
61
span .SetStatus (trace.Status {Code : int32 (codes .Unknown ), Message : err .Error ()})
@@ -70,7 +69,6 @@ func GRPCClientTrace(options ...TracerOption) kitgrpc.ClientOption {
70
69
clientBefore (c )
71
70
clientFinalizer (c )
72
71
}
73
-
74
72
}
75
73
76
74
// GRPCServerTrace enables OpenCensus tracing of a Go kit gRPC transport server.
@@ -83,33 +81,33 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
83
81
84
82
serverBefore := kitgrpc .ServerBefore (
85
83
func (ctx context.Context , md metadata.MD ) context.Context {
86
- var (
87
- spanContext trace.SpanContext
88
- ok bool
89
- name string
90
- )
84
+ var name string
91
85
92
86
if cfg .name != "" {
93
87
name = cfg .name
94
88
} else {
95
- name , ok = ctx .Value (kitgrpc .ContextKeyRequestMethod ).(string )
96
- if ! ok || name == "" {
89
+ name , _ = ctx .Value (kitgrpc .ContextKeyRequestMethod ).(string )
90
+ if name == "" {
97
91
// we can't find the gRPC method. probably the
98
92
// unaryInterceptor was not wired up.
99
93
name = "unknown grpc method"
100
94
}
101
95
}
102
96
103
- traceContext := md [propagationKey ]
97
+ var (
98
+ parentContext trace.SpanContext
99
+ traceContext = md [propagationKey ]
100
+ ok bool
101
+ )
104
102
105
103
if len (traceContext ) > 0 {
106
104
traceContextBinary := []byte (traceContext [0 ])
107
- spanContext , ok = propagation .FromBinary (traceContextBinary )
105
+ parentContext , ok = propagation .FromBinary (traceContextBinary )
108
106
if ok && ! cfg .public {
109
107
ctx , _ = trace .StartSpanWithRemoteParent (
110
108
ctx ,
111
109
name ,
112
- spanContext ,
110
+ parentContext ,
113
111
trace .WithSpanKind (trace .SpanKindServer ),
114
112
trace .WithSampler (cfg .sampler ),
115
113
)
@@ -125,8 +123,8 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
125
123
if ok {
126
124
span .AddLink (
127
125
trace.Link {
128
- TraceID : spanContext .TraceID ,
129
- SpanID : spanContext .SpanID ,
126
+ TraceID : parentContext .TraceID ,
127
+ SpanID : parentContext .SpanID ,
130
128
Type : trace .LinkTypeChild ,
131
129
},
132
130
)
@@ -138,8 +136,7 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
138
136
serverFinalizer := kitgrpc .ServerFinalizer (
139
137
func (ctx context.Context , err error ) {
140
138
if span := trace .FromContext (ctx ); span != nil {
141
- s , ok := status .FromError (err )
142
- if ok {
139
+ if s , ok := status .FromError (err ); ok {
143
140
span .SetStatus (trace.Status {Code : int32 (s .Code ()), Message : s .Message ()})
144
141
} else {
145
142
span .SetStatus (trace.Status {Code : int32 (codes .Internal ), Message : err .Error ()})
0 commit comments