fix contextstore access

DataDog · Jul 2, 2024 · 92b2ee8 · 92b2ee8
1 parent b1385ac
commit 92b2ee8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/..._6/src/main/java/datadog/trace/instrumentation/tibcobw6/AbstractTibcoInstrumentation.java b/..._6/src/main/java/datadog/trace/instrumentation/tibcobw6/AbstractTibcoInstrumentation.java
@@ -2,7 +2,7 @@
 
 import datadog.trace.agent.tooling.InstrumenterModule;
 import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
-import java.util.Collections;
+import java.util.HashMap;
 import java.util.Map;
 
 public abstract class AbstractTibcoInstrumentation extends InstrumenterModule.Tracing {
@@ -17,7 +17,11 @@ protected boolean defaultEnabled() {
 
   @Override
   public Map<String, String> contextStore() {
-    return Collections.singletonMap("com.tibco.pvm.api.PmWorkUnit", AgentSpan.class.getName());
+    Map<String, String> stores = new HashMap<>();
+    stores.put("com.tibco.pvm.api.PmWorkUnit", AgentSpan.class.getName());
+    stores.put(
+        "com.tibco.bw.jms.shared.api.receive.JMSMessageCallBackHandler", String.class.getName());
+    return stores;
   }
 
   @Override