Add scope to intercept client response message

opentracing-contrib · May 26, 2020 · cde1dd0 · cde1dd0
1 parent 471bd10
commit cde1dd0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/io/opentracing/contrib/grpc/TracingClientInterceptor.java b/src/main/java/io/opentracing/contrib/grpc/TracingClientInterceptor.java
@@ -196,7 +196,10 @@ public void onMessage(RespT message) {
                             .put(Fields.MESSAGE, "Client received response message")
                             .build());
                   }
-                  super.onMessage(message);
+
+                  try (Scope ignored = tracer.scopeManager().activate(span)) {
+                    super.onMessage(message);
+                  }
                 }
 
                 @Override