fix(instrument): fix dubbo client trace inject adapter.

open-telemetry · Jun 29, 2021 · f0837af · f0837af
1 parent c46761b
commit f0837af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...y/src/main/java/io/opentelemetry/instrumentation/apachedubbo/v2_7/DubboInjectAdapter.java b/...y/src/main/java/io/opentelemetry/instrumentation/apachedubbo/v2_7/DubboInjectAdapter.java
@@ -13,7 +13,7 @@ class DubboInjectAdapter implements TextMapSetter<RpcInvocation> {
   static final DubboInjectAdapter SETTER = new DubboInjectAdapter();
 
   @Override
-  public void set(RpcInvocation rpcInvocation, String key, String value) {
-    rpcInvocation.setAttachment(key, value);
+  public void set(RpcContext rpcContext, String key, String value) {
+    rpcContext.setAttachment(key, value);
   }
 }
diff --git a/.../src/main/java/io/opentelemetry/instrumentation/apachedubbo/v2_7/OpenTelemetryFilter.java b/.../src/main/java/io/opentelemetry/instrumentation/apachedubbo/v2_7/OpenTelemetryFilter.java
@@ -40,7 +40,7 @@ public Result invoke(Invoker<?> invoker, Invocation invocation) {
     final Context context;
     if (kind.equals(CLIENT)) {
       context = tracer.startClientSpan(interfaceName, methodName);
-      tracer.inject(context, (RpcInvocation) invocation, DubboInjectAdapter.SETTER);
+      tracer.inject(context, rpcContext, DubboInjectAdapter.SETTER);
     } else {
       context = tracer.startServerSpan(interfaceName, methodName, (RpcInvocation) invocation);
     }