javaagent学习

pom.xml

    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    4.0.0
    com.study
    instrustment
    0.0.1-SNAPSHOT
    
 
        
            org.apache.bcel
            bcel
            6.2
        

        
        
            org.javassist
            javassist
            3.19.0-GA
        

    
    
        
            
                com.rimerosolutions.maven.plugins
                wrapper-maven-plugin
                0.0.4
            

            
                org.apache.maven.plugins
                maven-compiler-plugin
                
                    UTF-8
                    1.8
                    1.8
                    
                        
                        ${java.home}/lib/rt.jar
                    

                

            

        

    



 

public class MyMain {

    public static void main(String args[]) {
        myTest();

    }

    private static void myTest() {
        try {
            Thread.currentThread().sleep(3000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

}

import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.Instrumentation;

public class PerfMonAgent {
    private static Instrumentation inst = null;

    public static void premain(String agentArgs, Instrumentation _inst) {
        System.out.println("PerfMonAgent.premain() was called.");
        inst = _inst;
        ClassFileTransformer trans = new PerfMonXformer();
        System.out.println("Adding a PerfMonXformer instance to the JVM.");
        inst.addTransformer(trans);
    }
}

 import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.security.ProtectionDomain;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtBehavior;
import javassist.CtClass;
import javassist.NotFoundException;

public class PerfMonXformer implements ClassFileTransformer {
    public byte[] transform(ClassLoader loader, String className, Class classBeingRedefined,
            ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
        byte[] transformed = null;
//        System.out.println("Transforming " + className);
        ClassPool pool = ClassPool.getDefault();
        CtClass cl = null;
        try {
            cl = pool.makeClass(new java.io.ByteArrayInputStream(classfileBuffer));
            if (cl.isInterface() == false) {
                CtBehavior[] methods = cl.getDeclaredBehaviors();
                for (int i = 0; i < methods.length; i++) {
                    if (methods[i].isEmpty() == false) {
                        doMethod(methods[i]);
                    }
                }
                transformed = cl.toBytecode();
            }
        } catch (Exception e) {
            System.err.println("Could not instrument  " + className + ",  exception : " + e.getMessage());
        } finally {
            if (cl != null) {
                cl.detach();
            }
        }
        return transformed;
    }
 

    private void doMethod(CtBehavior method) throws NotFoundException, CannotCompileException {
        if ("myTest".equalsIgnoreCase(method.getName())) {
            // 添加局部变量,如果不同过addLocalVariable设置,在调用属性时将出现compile error: no such field:
            // startTime
            method.addLocalVariable("startTime", CtClass.longType);
            method.insertBefore("System.out.println(startTime);");
            method.insertBefore("startTime = System.currentTimeMillis();");
//            method.insertBefore("long startTime = System.currentTimeMillis();System.out.println(startTime);");
            method.insertBefore("System.out.println(\"insert before ......\");");
            method.insertAfter("System.out.println(\"leave " + method.getName()
                    + " and time is :\"+System.currentTimeMillis()+\":\" + (System.currentTimeMillis() - startTime));");
        }
    }
 
}

java -cp  /home/your_home/.m2/repository/org/javassist/javassist/3.19.0-GA/javassist-3.19.0-GA.jar:./ -javaagent:/yourpath/target/PerMonAgent.jar MyMain

----------------

MANIFEST.MF

Manifest-Version: 1.0
Premain-Class: PerfMonAgent
Can-Redefine-Classes: true

javaagent学习_第1张图片

你可能感兴趣的:(java)