Skip to content
Snippets Groups Projects
Commit f85d07bc authored by 邹毅贤's avatar 邹毅贤
Browse files

mod grpc tracing

parent 50b57f95
No related branches found
No related tags found
No related merge requests found
...@@ -53,7 +53,10 @@ func NewClient(url common.URL) *Client { ...@@ -53,7 +53,10 @@ func NewClient(url common.URL) *Client {
if opentracing.IsGlobalTracerRegistered() { if opentracing.IsGlobalTracerRegistered() {
tracer = opentracing.GlobalTracer() tracer = opentracing.GlobalTracer()
conn, err = grpc.Dial(url.Location, grpc.WithInsecure(), grpc.WithBlock(), conn, err = grpc.Dial(url.Location, grpc.WithInsecure(), grpc.WithBlock(),
grpc.WithUnaryInterceptor(otgrpc.OpenTracingClientInterceptor(tracer))) grpc.WithUnaryInterceptor(
otgrpc.OpenTracingClientInterceptor(tracer, otgrpc.LogPayloads())),
grpc.WithStreamInterceptor(
otgrpc.OpenTracingStreamClientInterceptor(tracer, otgrpc.LogPayloads())))
if err != nil { if err != nil {
panic(err) panic(err)
} }
......
...@@ -73,7 +73,9 @@ func (s *Server) Start(url common.URL) { ...@@ -73,7 +73,9 @@ func (s *Server) Start(url common.URL) {
// if global trace instance was set, then server tracer instance can be get, and span context can also be get // if global trace instance was set, then server tracer instance can be get, and span context can also be get
if opentracing.IsGlobalTracerRegistered() { if opentracing.IsGlobalTracerRegistered() {
tracer := opentracing.GlobalTracer() tracer := opentracing.GlobalTracer()
server = grpc.NewServer(grpc.UnaryInterceptor(otgrpc.OpenTracingServerInterceptor(tracer))) server = grpc.NewServer(
grpc.UnaryInterceptor(otgrpc.OpenTracingServerInterceptor(tracer)),
grpc.StreamInterceptor(otgrpc.OpenTracingStreamServerInterceptor(tracer)))
} else { } else {
server = grpc.NewServer() server = grpc.NewServer()
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment