JavaAgent技术之添加注解

什么是JavaAgent(Java探针)你可以理解为Java版AOP。只不过这个AOP项目启动时运行一次JavaAgent 只在项目启动时运行一次并且是java文件编译成class文件后才运行。所以不会影响到class文件。JavaAgent 是寄生项目。即需要依赖一个正常项目才能运行我这里演示使用 IDE 为 IDEA 项目为maven普通项目结构可以很方便的帮助我们创建一个 JavaAgent 项目。按住CTRLSHIFTALTS 键进入Project Structure界面选择添加JAR选择项目注意下面的链接要在SRC下。创建好的目录结构其中Premain-Class 是我后期添加上去的。这里指定的为后面的 Agent 类中的 premain 方法。同时也是JavaAgent的启动方法Class-path中的值也是我们在maven中添加的包。你们应该没有编写一个JavaAgent方法packagecom.annie;importjava.lang.instrument.Instrumentation;publicclassAgent{// JavaAgent启动时调用的方法publicstaticvoidpremain(String args,Instrumentation instrumentation){System.out.println(传入的参数为: args);System.out.println(JavaAgent启动了...);}}到这里一个JavaAgent项目就基本启动完成了。我们在创建一个普通项目项目类型无所谓这里我创建的是一个SpringBoot的项目因为比较好演示。先将将JavaAgent编译jar包编译好的jar包就在项目同级目录下的out下面。设置启动配置后面一个Hello是传入的参数。可有可无。不要时不要忘记连 号也去掉。-javaagent:C:\Users\13100\Documents\IdeaProjects\BaseJava\out\artifacts\JavaAgent_jar\JavaAgent.jarHello保存运行到这里一个JavaAgent项目就基本搭建成功了。当然我们不能满足于此。这里我自演示基本的动态注解其实以下就没有和JaveAgent相关内容了。添加javassist包。!-- https://mvnrepository.com/artifact/org.javassist/javassist --dependencygroupIdorg.javassist/groupIdartifactIdjavassist/artifactIdversion3.25.0-GA/version/dependency编写代码写完不要忘记重新编译packagecom.annie;importjavassist.ClassPool;importjavassist.CtClass;importjavassist.bytecode.AnnotationsAttribute;importjavassist.bytecode.AttributeInfo;importjavassist.bytecode.ClassFile;importjavassist.bytecode.ConstPool;importjavassist.bytecode.annotation.Annotation;importjava.io.ByteArrayInputStream;importjava.io.IOException;importjava.lang.instrument.ClassFileTransformer;importjava.lang.instrument.IllegalClassFormatException;importjava.lang.instrument.Instrumentation;importjava.security.ProtectionDomain;importjava.util.List;publicclassAgent{// JavaAgent启动时调用的方法publicstaticvoidpremain(String args,Instrumentation instrumentation){System.out.println(传入的参数为: args);System.out.println(JavaAgent启动了...);// instrumentation 中包含了项目中的全部类。每加载一次.class文件就运行一次instrumentation.addTransformer(newClassFileTransformer(){Overridepublicbyte[]transform(ClassLoader loader,String className,Class?classBeingRedefined,ProtectionDomain protectionDomain,byte[]classfileBuffer)throwsIllegalClassFormatException{// 这里我只捕捉自己写 TestController 类if(className.equals(com/annie/controller/TestController)){// 接下来就是javassist的使用我会粗略的写下备注。不会详解。// 获取一个 class 池。ClassPool classPoolClassPool.getDefault();try{// 创建一个新的 class 类。classfileBuffer 就是当前class的字节码CtClass ctClassclassPool.makeClass(newByteArrayInputStream(classfileBuffer));ClassFile classFilectClass.getClassFile();ConstPool constPoolclassFile.getConstPool();// 从这里取出原本类中的注解 建议DEBUG看下attributes中的数据ListAttributeInfoattributesclassFile.getAttributes();AnnotationsAttribute attributeInfo(AnnotationsAttribute)attributes.get(1);// 导包是javassist的包// 添加新的注解Annotation annotationnewAnnotation(org.springframework.web.bind.annotation.RequestMapping,constPool);attributeInfo.addAnnotation(annotation);// 返回新的字节码returnctClass.toBytecode();}catch(Exceptione){e.printStackTrace();}}returnnewbyte[0];}});}}编写一个 TestController这里我只给了RestController的注解。并没有给RequestMapping注解。RequestMapping由JavaAgent为我添加。packagecom.annie.controller;importorg.springframework.web.bind.annotation.RequestMapping;importorg.springframework.web.bind.annotation.RestController;importjava.lang.annotation.Annotation;RestControllerpublicclassTestController{RequestMapping(/test)publicStringtest(){// 利用反射查看Class注解。Class?extendsTestControlleraClassthis.getClass();Annotation[]declaredAnnotationsaClass.getDeclaredAnnotations();StringBuffer sbnewStringBuffer();// 查看类上全部注解for(Annotation annotation:declaredAnnotations){sb.append(annotation.toString()\n);}returnsb.toString();}}浏览器上输入 127.0.0.1:8080/test 查看TestController上的注解补充内容 linux 上发布项目将javaAgent.jar 和 javaassist.jar 放入到一个文件夹中。运行命令java -javaagent:JavaAgent.jar -jar web-test-0.0.1-SNAPSHOT.jar项目就启动成功。注意一点这个jar包就是你bulid出来的JAR扩展内容github个人博客