背景：假如我們想打印出某些系統(tǒng)->某些類->某些方法的執(zhí)行耗時，方式有很多召庞，但是想要無侵入的做到這一點破镰，只有Java探針一種方式题画。這也是很多調(diào)用鏈系統(tǒng)依賴的技術(shù)基礎(chǔ)默辨。
什么是Java探針
通俗來講，就是Java提供的一種手段苍息，使我們可以修改并重新加載Class字節(jié)碼缩幸，做到在系統(tǒng)外部來改變類的行為。
Java探針是如何做到的
我們知道只要是java程序竞思，運行的入口就一定是main方法表谊。Java探針技術(shù)相當(dāng)于變相的改變了這個約定，提供了一種外掛盖喷，只要用了這個外掛(即在程序的運行時參數(shù)中加了-javaagent)爆办，就可以先執(zhí)行外掛jar包中的premain方法，然后再執(zhí)行原程序中的main方法课梳。這樣就提供了一個在真正的程序執(zhí)行前距辆，可以修改并重新加載字節(jié)碼的機會。
具體怎么做
我們寫個demo暮刃。
-
假如原程序為：
public class DemoMain { public static void main(String[] args) { System.out.println("\nDemoMain.main開始"); HelloService helloService = new HelloService(); helloService.sayHello("一拳超人"); System.out.println("DemoMain.main結(jié)束"); } } public class HelloService { public void sayHello(String name) { System.out.println(" HelloService.sayHello開始"); System.out.println(" Hello " + name + "!"); System.out.println(" HelloService.sayHello結(jié)束"); } } -
打jar包跨算，新版的idea打包配置有點小問題，默認(rèn)的MANIFEST的文件夾路徑中椭懊，要刪除/main/java诸蚕，如圖：
idea打包配置.jpg 我們想打印HelloService.sayHello的執(zhí)行耗時，使用Java探針來實現(xiàn)氧猬，先寫外掛程序
```java
public class PrintAgent {
public static void premain(String agentArgs, Instrumentation instrumentation) {
System.out.println("\nPrintAgent.premain開始");
System.out.println(" PrintAgent.premain入?yún)ⅲ? + agentArgs);
instrumentation.addTransformer(new PrintTransformer());
System.out.println("PrintAgent.premain結(jié)束");
}
public static void main(String[] args) {
//main方法在idea打包時需要背犯，實際上沒什么作用，不會被執(zhí)行盅抚。
}
}
public class PrintTransformer implements ClassFileTransformer {
final static String startTime = "\nlong startTime = System.nanoTime();\n";
final static String endTime = "\nlong endTime = System.nanoTime();\n";
final static Map<String, List<String>> methodMap = new HashMap<String, List<String>>();
public PrintTransformer() {
addMethod("com.wang.hello.HelloService", "sayHello");
}
private PrintTransformer addMethod(String className, String methodName) {
List<String> list = methodMap.computeIfAbsent(className, cn -> new ArrayList<>());
list.add(methodName);
return this;
}
@Override
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
className = className.replace("/", ".");
if (methodMap.containsKey(className)) {
try {
CtClass ctClass = ClassPool.getDefault().get(className);
for (String methodName : methodMap.get(className)) {
String cost = "\nSystem.out.println(\" Method:"
+ methodName
+ " Cost:\"+(endTime-startTime)+"
+ "\"ns\");\n";
CtMethod ctMethod = ctClass.getDeclaredMethod(methodName);
String wrappedMethodName = methodName + "$wrapped";
ctMethod.setName(wrappedMethodName);
CtMethod newMethod = CtNewMethod.copy(ctMethod, methodName, ctClass, null);
StringBuilder newMethodBody = new StringBuilder();
newMethodBody.append("{");
newMethodBody.append(startTime);
newMethodBody.append(wrappedMethodName + "($$);\n");
newMethodBody.append(endTime);
newMethodBody.append(cost);
newMethodBody.append("}");
newMethod.setBody(newMethodBody.toString());
ctClass.addMethod(newMethod);
}
return ctClass.toBytecode();
} catch (Exception e) {
e.printStackTrace();
}
}
return null;
}
}
```
- PrintAgent的打包配置如圖：
16191343861365.jpg
-
然后需要修改PrintAgent的META-INF/MANIFEST.MF文件媳板，改為(注意結(jié)尾必須加一新行)：
Manifest-Version: 1.0 Premain-Class: com.wang.agent.PrintAgent Can-Redefine-Classes: true **我是空白行占位符** -
打完包后，我們就可以執(zhí)行測試了泉哈，先看下原始程序打印的日志：
//執(zhí)行java -jar demo.jar DemoMain.main開始 HelloService.sayHello開始 Hello 一拳超人! HelloService.sayHello結(jié)束 DemoMain.main結(jié)束 -
外掛探針后蛉幸，再次執(zhí)行打印的日志破讨，注意其中"Method:sayHello Cost:128835ns"為我們增加的耗時日志：
//執(zhí)行java -javaagent:instrumentation.jar=wahaha -jar demo.jar PrintAgent.premain開始 PrintAgent.premain入?yún)ⅲ簑ahaha PrintAgent.premain結(jié)束 DemoMain.main開始 HelloService.sayHello開始 Hello 一拳超人! HelloService.sayHello結(jié)束 Method:sayHello Cost:128835ns DemoMain.main結(jié)束 關(guān)鍵的點解決后，再往深了做奕纫，就能慢慢的實現(xiàn)調(diào)用鏈的邏輯了提陶。
