Skip loading agent if traceable agent has been loaded (#180)

Signed-off-by: Pavol Loffay <p.loffay@gmail.com>
hypertrace · Dec 17, 2020 · 17e1dcb · 17e1dcb
1 parent d5e5699
commit 17e1dcb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/javaagent/src/main/java/org/hypertrace/agent/instrument/HypertraceAgent.java b/javaagent/src/main/java/org/hypertrace/agent/instrument/HypertraceAgent.java
@@ -46,6 +46,11 @@ public static void premain(String agentArgs, Instrumentation inst) {
   }
 
   public static void agentmain(String agentArgs, Instrumentation inst) {
+    if (isTraceableAgentLoaded()) {
+      System.out.println("Traceable agent is already running. Will not attempt to attach");
+      return;
+    }
+
     Map<String, String> parsedArgs = parseAgentArgs(agentArgs);
     for (Map.Entry<String, String> argEntry : parsedArgs.entrySet()) {
       System.setProperty(argEntry.getKey(), argEntry.getValue());
@@ -103,4 +108,13 @@ private static Map<String, String> parseAgentArgs(String agentArgs) {
     }
     return argsMap;
   }
+
+  private static boolean isTraceableAgentLoaded() {
+    try {
+      Class.forName("ai.traceable.agent.bootstrap.AgentContext", true, null);
+      return true;
+    } catch (Throwable e) {
+      return false;
+    }
+  }
 }