ignore double boots

Author: BFergerson

control/src/main/kotlin/spp/probe/SourceProbe.kt

@@ -55,7 +55,11 @@ object SourceProbe {
 
     @JvmStatic
     fun bootAsPlugin(inst: Instrumentation) {
-        if (ProbeConfiguration.isNotQuite) println("SourceProbe initiated")
+        if (isAgentInitialized) {
+            if (ProbeConfiguration.isNotQuite) println("SourceProbe is already initialized")
+            return
+        }
+        if (ProbeConfiguration.isNotQuite) println("SourceProbe initiated via plugin")
 
         //todo: pipe data if in debug mode
         System.setProperty("vertx.logger-delegate-factory-class-name", NopLogDelegateFactory::class.java.canonicalName)
@@ -90,7 +94,11 @@ object SourceProbe {
 
     @JvmStatic
     fun premain(args: String?, inst: Instrumentation) {
-        if (ProbeConfiguration.isNotQuite) println("SourceProbe initiated")
+        if (isAgentInitialized) {
+            if (ProbeConfiguration.isNotQuite) println("SourceProbe is already initialized")
+            return
+        }
+        if (ProbeConfiguration.isNotQuite) println("SourceProbe initiated via agent")
 
         //todo: pipe data if in debug mode
         System.setProperty("vertx.logger-delegate-factory-class-name", NopLogDelegateFactory::class.java.canonicalName)

control/src/main/kotlin/spp/probe/SourceProbePluginDefine.kt

@@ -17,54 +17,65 @@ import java.util.concurrent.atomic.AtomicBoolean
  */
 class SourceProbePluginDefine : ClassInstanceMethodsEnhancePluginDefine() {
 
+    companion object {
+        private val probeStarted = AtomicBoolean()
+        private val cache = HashMap<String, ClassMatch>()
+    }
+
     override fun enhanceClass(): ClassMatch {
-        val defaultClass = Class.forName(
-            "org.apache.skywalking.apm.dependencies.net.bytebuddy.agent.builder.AgentBuilder\$Default"
-        )
-        val dispatcherField = defaultClass.getDeclaredField("DISPATCHER")
-        makeAccessible(dispatcherField)
-        val realDispatcher = dispatcherField.get(null)
-        dispatcherField.set(null, null)
+        if (SourceProbe.isAgentInitialized) {
+            if (ProbeConfiguration.isNotQuite) println("SourceProbe is already initialized")
+            return NameMatch.byName("") //ignore
+        }
 
-        val dispatcherClass = Class.forName(
-            "org.apache.skywalking.apm.dependencies.net.bytebuddy.agent.builder.AgentBuilder\$Default\$Dispatcher"
-        )
-        val addTransformerMethod = dispatcherClass.getDeclaredMethod(
-            "addTransformer",
-            Instrumentation::class.java, ClassFileTransformer::class.java, Boolean::class.java
-        )
-        val setNativeMethodPrefixMethod = dispatcherClass.getDeclaredMethod(
-            "setNativeMethodPrefix",
-            Instrumentation::class.java, ClassFileTransformer::class.java, String::class.java
-        )
-        val isNativeMethodPrefixSupportedMethod = dispatcherClass.getDeclaredMethod(
-            "isNativeMethodPrefixSupported", Instrumentation::class.java
-        )
+        return cache.computeIfAbsent("cache") {
+            val defaultClass = Class.forName(
+                "org.apache.skywalking.apm.dependencies.net.bytebuddy.agent.builder.AgentBuilder\$Default"
+            )
+            val dispatcherField = defaultClass.getDeclaredField("DISPATCHER")
+            makeAccessible(dispatcherField)
+            val realDispatcher = dispatcherField.get(null)
+            dispatcherField.set(null, null)
 
-        val probeStarted = AtomicBoolean()
-        val proxyDispatcher = Proxy.newProxyInstance(
-            dispatcherClass.classLoader,
-            arrayOf(dispatcherClass)
-        ) { _, method, args ->
-            return@newProxyInstance when (method.name) {
-                "addTransformer" -> {
-                    if (probeStarted.compareAndSet(false, true)) {
-                        SourceProbe.bootAsPlugin(args?.get(0) as Instrumentation)
+            val dispatcherClass = Class.forName(
+                "org.apache.skywalking.apm.dependencies.net.bytebuddy.agent.builder.AgentBuilder\$Default\$Dispatcher"
+            )
+            val addTransformerMethod = dispatcherClass.getDeclaredMethod(
+                "addTransformer",
+                Instrumentation::class.java, ClassFileTransformer::class.java, Boolean::class.java
+            )
+            val setNativeMethodPrefixMethod = dispatcherClass.getDeclaredMethod(
+                "setNativeMethodPrefix",
+                Instrumentation::class.java, ClassFileTransformer::class.java, String::class.java
+            )
+            val isNativeMethodPrefixSupportedMethod = dispatcherClass.getDeclaredMethod(
+                "isNativeMethodPrefixSupported", Instrumentation::class.java
+            )
+
+            val proxyDispatcher = Proxy.newProxyInstance(
+                dispatcherClass.classLoader,
+                arrayOf(dispatcherClass)
+            ) { _, method, args ->
+                return@newProxyInstance when (method.name) {
+                    "addTransformer" -> {
+                        if (probeStarted.compareAndSet(false, true)) {
+                            SourceProbe.bootAsPlugin(args?.get(0) as Instrumentation)
+                        }
+                        addTransformerMethod.invoke(realDispatcher, args[0], args[1], args[2])
                     }
-                    addTransformerMethod.invoke(realDispatcher, args[0], args[1], args[2])
-                }
-                "setNativeMethodPrefix" -> {
-                    setNativeMethodPrefixMethod.invoke(realDispatcher, args[0], args[1], args[2])
-                }
-                "isNativeMethodPrefixSupported" -> {
-                    isNativeMethodPrefixSupportedMethod.invoke(realDispatcher, args[0])
+                    "setNativeMethodPrefix" -> {
+                        setNativeMethodPrefixMethod.invoke(realDispatcher, args[0], args[1], args[2])
+                    }
+                    "isNativeMethodPrefixSupported" -> {
+                        isNativeMethodPrefixSupportedMethod.invoke(realDispatcher, args[0])
+                    }
+                    else -> throw IllegalStateException("Unknown method: ${method.name}")
                 }
-                else -> throw IllegalStateException("Unknown method: ${method.name}")
             }
-        }
-        dispatcherField.set(null, proxyDispatcher)
+            dispatcherField.set(null, proxyDispatcher)
 
-        return NameMatch.byName("") //ignore
+            return@computeIfAbsent NameMatch.byName("") //ignore
+        }
     }
 
     @Throws(Exception::class)