考慮下這個問題：怎么知道我寫的方法執(zhí)行了多久锅移？
再考慮下：怎么知道所有方法分別執(zhí)行了多久在抛？
初學(xué)者都會的方法
void methodDemo(){
Date start = new Date();
//...業(yè)務(wù)代碼
Date end = new Date();
Long duration = end.getTime() - start.getTime();
}
上面的方法我的確可以知道methodDemo()的執(zhí)行時間蟹地，但是如果我有一百個這樣的方法怎么辦或悲？寫一百遍么？這時候用過Spring的站出來指著我鼻子說:”你傻啊堪唐，用AOP啊!“
Spring AOP
具體的樣例代碼我就不贅述了巡语，網(wǎng)上一搜一大把教你使用Spring AOP. 但是有一個問題淮菠，考慮下你的代碼是這樣的：
class Test1{
public void test(){
testInside(12345);
System.out.println("the test method executed");
}
private void testInside(int param){
System.out.println("the testInside method executed,param:"+param);
}
}
可以注意到，我們在test()方法中調(diào)用了testInside()這個private的方法合陵。有深入了解Spring AOP的實現(xiàn)原理就知道枢赔，Spring在實現(xiàn)AOP的時候是在bean放入容器前生成的代理類。而test()調(diào)用this.testInside()的情況拥知，是沒有對testInside()進行AOP增強的。怎么辦举庶？這個時候，就進入我們本文的重點。
Java agent 在JVM層面實現(xiàn)"AOP增強"
回顧一下峦嗤，類是怎么被加載到JVM里面的？加載->驗證->準備->解析->初始化替梨。在加載階段，一定會有一個步驟是把類的二進制信息讀入JVM（可能是.class文件副瀑、可能是網(wǎng)絡(luò)等）恋谭。那么我們可不可以在都進來后，開始下一個步驟前狈孔，改一改類的二進制信息呢？（比如往里面添加統(tǒng)計時間）
答案是當然可以均抽。朋友，Java agent和instrumentation了解一下潦蝇？
什么是java agent
看字面意思深寥，agent就是代理的意思啊。從外部視角來看翩迈，它相當于對我的java程序的一個代理。既然提到代理负饲，那我不是可以在運行我的java程序前做些什么小動作？
先來看用法: java -javaagent:the-agent-demo.jar HelloWorld
在命令行中敲入上面的命令妥泉，是說以the-agent-demo.jar為java agent洞坑，運行我的HelloWorld程序。
這個時候迟杂，在運行HelloWorld的main()方法前排拷，會先運行the-agent-demo.jar中的premain方法。
而premain方法的參數(shù)是什么呢监氢？
public static void premain(String agentArgument,
Instrumentation instrumentation){
System.out.println("Java Agent Demo");
SimpleClassTransformer simpleClassTransformer = new SimpleClassTransformer();
instrumentation.addTransformer(simpleClassTransformer);
}
agentArgument這個是自定義的參數(shù)，比如我可以java -javaagent:the-agent-demo.jar=theAgentArumentDemo HelloWorld其中theAgentArgumentDemo就作為這個參數(shù)傳進來了纵揍。而第二個參數(shù)則是下一節(jié)要描述的议街。再次之前，還要注意隔盛，我們需要在the-agent-demo.jar里面打包進去包含Pre-Main參數(shù)的MENIFEST.MF文件。
Manifest-Version: 1.0
Premain-Class: cn.kobelee.test.TestJavaAgent
什么是instrumentation
Instrumentation是一個接口吮炕，定義了字節(jié)碼修改的規(guī)范龙亲，其位于java.lang.instrument包下面。
在這個包下面另外一個關(guān)鍵的接口類是ClassFileTransformer鳄炉，顧名思義類文件轉(zhuǎn)換器。它只有一個接口定義方法：
byte[]
transform( ClassLoader loader,
String className,
Class<?> classBeingRedefined,
ProtectionDomain protectionDomain,
byte[] classfileBuffer)
throws IllegalClassFormatException;
注意到有一個byte[]數(shù)組的參數(shù)classFileBuffer佑女。這就是類的二進制文件buffer谈竿，其返回值也是byte[]數(shù)組，為修改后的類嚎花。那么這個方法的實現(xiàn)就是要transform（轉(zhuǎn)換/修改）類咯呀洲。
完整連起來就是: java -javaagent:xxx.jar HelloWorld指定代理的jar，里面有premain. 我們需要在premain里面對instrumentation添加ClassTransformer. 而這個classTransformer的實現(xiàn)就是你要怎么修改這個類兵罢。
什么是javassist
上面說到我們要修改byte[]來達到修改類的目的滓窍。可是贰您，直接改二進制文件這種騷操作可能只有上古達人才能做到吧拢操。于是，javassist的作用來了杠园。javassist是jboss提供的一個方便我們修改這個byte[]的工具包舔庶。直接上例子：
public class SimpleClassTransformer implements ClassFileTransformer {
@Override
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
try {
if(!className.contains("kobelee")){//我只需要我定義的包路徑下統(tǒng)計陈醒，當然這個也判斷也可以刪了
return null;
}
CtClass ctClass = ClassPool.getDefault().makeClass(new ByteArrayInputStream(classfileBuffer));
CtMethod[] declaredMethods = ctClass.getDeclaredMethods();
for (CtBehavior method : declaredMethods) {
CtClass[] parameterTypes = method.getParameterTypes();
StringBuilder sb = new StringBuilder("{");
for (int i = 0; i< parameterTypes.length; i++) {
sb.append("StringBuilder code = new StringBuilder();");
sb.append("code.append(\""+method.getLongName()+" before.\");");
sb.append("code.append(\""+parameterTypes[i].getName()+"\");");
sb.append("code.append(\":\");");
sb.append("code.append($args["+i+"]);");
sb.append("System.out.println(code.toString());");
}
sb.append("}");
method.insertBefore(sb.toString());
method.insertAfter("System.out.println(\""+method.getLongName()+" end\");");
}
byte[] returnByte = ctClass.toBytecode();
return returnByte;
} catch (IOException e) {
e.printStackTrace();
} catch (CannotCompileException e) {
e.printStackTrace();
} catch (NotFoundException e) {
e.printStackTrace();
}
return classfileBuffer;
}
}
可以看到代碼中有CtClass, CtMethod等類钉跷。這些表示的是CompileTimeXXX 也就是編譯時候的類相關(guān)信息肚逸。我們通過CtClass ctClass = ClassPool.getDefault().makeClass(new ByteArrayInputStream(classfileBuffer));創(chuàng)建了一個ctClass對象朦促，然后就可以對其注入我們需要的代碼了。上面代碼的例子只是輸出了調(diào)用的方法名和方法參數(shù)务冕，至于具體執(zhí)行時間，采用類似的方法也就不難實現(xiàn)了臊旭。
注意在打包agent.jar的時候油湖，不要忘了將javassist.jar也一起打包進去
public class HelloWorld {
public static void main(String[] args) throws ClassNotFoundException, SQLException {
Test1 one = new Test1();
one.test();
System.out.println("Hello World");
}
}
class Test1{
public void test(){
testInside(12345);
System.out.println("the test method executed");
}
private void testInside(int param){
System.out.println("the testInside method executed,param:"+param);
}
}
java -javaagent:JavaAgentDemo-1.0-SNAPSHOT.jar HelloWorld執(zhí)行控制臺輸出:
Java Agent Demo
方法:cn.kobelee.test.HelloWorld.main(java.lang.String[])執(zhí)行前
方法:cn.kobelee.test.Test1.test()執(zhí)行前
方法:cn.kobelee.test.Test1.testInside(int)執(zhí)行前
the testInside method executed,param:12345
方法:cn.kobelee.test.Test1.testInside(int)執(zhí)行后
the test method executed
方法:cn.kobelee.test.Test1.test()執(zhí)行后
Hello World
方法:cn.kobelee.test.HelloWorld.main(java.lang.String[])執(zhí)行后
可以看到乏德，我們的testInside方法也在調(diào)用前執(zhí)行了我們添加的代碼。
總結(jié)
通過使用java agent喊括，代理我們的應(yīng)用。同時對instrument的不同實現(xiàn)府喳，達到我們可以在業(yè)務(wù)代碼執(zhí)行前后插入任何我們想要的邏輯蘑拯；但是我們并沒有修改任何一行業(yè)務(wù)代碼。目前業(yè)界主要用來做分布式系統(tǒng)的鏈路跟蹤日志輸出申窘。將業(yè)務(wù)日志在java agent中按照指定格式輸出，同時輸出分布式環(huán)境下的調(diào)用唯一標識碎捺，然后再將日志放入流處理引擎中進行鏈路生成。比如這篇博客講的阿里鷹眼監(jiān)控：阿里巴巴鷹眼技術(shù)解密晋柱。其中的第一步日志輸出就需要用到本文講的方法诵叁。
