attach jdk 11

Author: songlonqi-java

lib/tools.jar

@@ -17,8 +17,8 @@
         <groupId>org.apache.maven.plugins</groupId>
         <artifactId>maven-compiler-plugin</artifactId>
         <configuration>
-          <source>1.8</source>
-          <target>1.8</target>
+          <source>11</source>
+          <target>11</target>
         </configuration>
       </plugin>
       <plugin>
@@ -67,14 +67,22 @@
       <version>3.2</version>
     </dependency>
 
-    <dependency>
+<!--    <dependency>
       <groupId>io.earcam.wrapped</groupId>
       <artifactId>com.sun.tools.attach</artifactId>
       <version>1.8.0_jdk8u131-b11</version>
       <scope>compile</scope>
       <type>jar</type>
+    </dependency>-->
+    <dependency>
+      <groupId>com.sun</groupId>
+      <artifactId>tools</artifactId>
+      <version>1.8.0</version>
+      <scope>system</scope>
+      <systemPath>${project.basedir}/lib/tools.jar</systemPath>
     </dependency>
 
+
     <dependency>
       <groupId>org.slf4j</groupId>
       <artifactId>jcl-over-slf4j</artifactId>
@@ -90,4 +98,4 @@
 
   </dependencies>
 
-</project>
+</project>

pom.xml

@@ -9,6 +9,10 @@ public class Config {
 
     private  String agentJar;
 
+    private String pid;
+
+    private String displayName;
+
     public String getOptions() {
         return this.options;
     }
@@ -21,17 +25,24 @@ public String getAgentJar() {
         return  this.agentJar;
     }
 
-    private Config(String options,String downloadAddress,String agentJar){
+    public String getPid(){return this.pid;}
+    public String getDisplayName(){return this.displayName;}
+
+    private Config(String options,String downloadAddress,String agentJar,String pid,String displayName){
         this.options = options;
         this.downloadAddress = downloadAddress;
         this.agentJar = agentJar;
+        this.pid = pid;
+        this.displayName = displayName;
     }
 
     static Config parse(String... args){
         String option = "";
         String downloadAddr= "";
         String currentArg = "";
         String agentJar= "";
+        String pid = "";
+        String displayName = "";
         for(String arg : args){
             if (arg.startsWith("-")){
                 currentArg =arg;
@@ -46,6 +57,12 @@ static Config parse(String... args){
                     case "-agent-jar":
                          agentJar = arg;
                         break;
+                    case "-pid":
+                        pid = arg;
+                        break;
+                    case "-displayName":
+                        displayName = arg;
+                        break;
                     case "-h":
                     case "-help":
                         printOut();
@@ -54,18 +71,25 @@ static Config parse(String... args){
                 }
             }
         }
-        return new Config(option,downloadAddr,agentJar);
+        return new Config(option,downloadAddr,agentJar,pid,displayName);
     }
     public static void printOut(){
         PrintStream out =   System.out;
         out.println("    java -jar agent-attach-java.jar [-options <dd options>]");
         out.println("                                    [-agent-jar <agent filepath>]");
+        out.println("                                    [-pid <pid>]");
+        out.println("                                    [-displayName <service name/displayName>]");
         out.println("                                    [-h]");
         out.println("                                    [-help]");
         out.println(" [-options]:");
         out.println(" this is dd-java-agnet.jar env, example:");
-        out.println("    dd.agent.port=9529,dd.agent.host=localhost,dd.service=serviceName");
+        out.println("    dd.agent.port=9529,dd.agent.host=localhost,dd.service=serviceName,...");
         out.println(" [-agent-jar]:");
         out.println(" default is: /usr/local/ddtrace/dd-java-agent.jar");
+        out.println(" [-pid]:");
+        out.println(" service PID String");
+        out.println(" [-displayName]:");
+        out.println(" service name");
+        out.println(" Note: -pid or -displayName must have a non empty !!!");
     }
 }

src/main/java/com/guance/javaagent/Config.java

@@ -12,26 +12,52 @@ public class JavaAgentLoader {
 
     private static final String jarFilePath = "/usr/local/ddtrace/dd-java-agent.jar";
 
-    public static void loadAgent(String agentJar ,String options) {
+    public static void loadAgent(Config config) {
         logger.info("dynamically loading javaagent");
-        logger.info("------------------options-------------"+options);
-        logger.info("------------------agentJar-------------"+agentJar);
+        logger.info("config.options:"+ config.getOptions());
+        logger.info("config.agentJar: "+config.getAgentJar());
+        logger.info("config.pid: "+config.getPid());
+        logger.info("config.displayName: "+config.getDisplayName());
+        boolean idMode = false;
+        boolean nameMode= false;
+        if (config.getPid() != null && !config.getPid().equals("")){
+            idMode = true;
+        }
+        if (config.getDisplayName()!= null && !config.getDisplayName().equals("")){
+            nameMode =true;
+        }
+        if (!idMode && !nameMode){
+            logger.warn("-pid or -displayName must have a non empty");
+            return;
+        }
         try {
             List<VirtualMachineDescriptor> list = VirtualMachine.list();
             for (int i = 0; i < list.size(); i++) {
                 VirtualMachineDescriptor virtualMachineDescriptor = list.get(i);
-                String version = virtualMachineDescriptor.id();
-                VirtualMachine attach = VirtualMachine.attach(version);
-                if (agentJar != null && !agentJar.equals("")){
-                    attach.loadAgent(agentJar, options);
+                String pid = virtualMachineDescriptor.id();
+		        logger.info(pid);
+		    //    logger.info(virtualMachineDescriptor.displayName());
+                VirtualMachine attach = null;
+                if (idMode && config.getPid().equals(pid) ){
+                     attach = VirtualMachine.attach(pid);
+                }
+                if (nameMode && virtualMachineDescriptor.displayName().equals(config.getDisplayName())){
+                    attach = VirtualMachine.attach(pid);
+                }
+                if (attach == null){
+                    continue;
+                }
+                if (config.getAgentJar() != null && !config.getAgentJar().equals("")){
+                    attach.loadAgent(config.getAgentJar(), config.getOptions());
                 }else {
-                    attach.loadAgent(jarFilePath, options);
+                    attach.loadAgent(jarFilePath, config.getOptions());
                 }
                 attach.detach();
                 logger.info(String.format("attach agent into [%s]",virtualMachineDescriptor.displayName()));
+                return;
             }
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
     }
-}
+}

src/main/java/com/guance/javaagent/JavaAgentLoader.java

@@ -17,8 +17,6 @@ public class MyMainClass {
     public static void main(String[] args) {
         Config config = Config.parse(args);
         // todo load agent ddtrace
-        JavaAgentLoader.loadAgent(config.getAgentJar(),config.getOptions());
-        logger.info("main method invoked with args: {}", Arrays.asList(args));
+        JavaAgentLoader.loadAgent(config);
     }
-
 }