JavaAgent技术之添加注解

什么是JavaAgent(Java探针)?你可以理解为Java版AOP。只不过这个AOP项目启动时运行一次

JavaAgent 只在项目启动时运行一次并且是java文件编译成class文件后才运行。所以不会影响到class文件。
JavaAgent 是寄生项目。即需要依赖一个正常项目才能运行

我这里演示使用 IDE 为 IDEA ,项目为maven普通项目结构,可以很方便的帮助我们创建一个 JavaAgent 项目。
  1. 按住CTRL+SHIFT+ALT+S 键,进入Project Structure界面
  2. 选择添加JARJavaAgent技术之添加注解
  3. 选择项目,注意下面的链接要在SRC下。JavaAgent技术之添加注解
  4. 创建好的目录结构,其中Premain-Class 是我后期添加上去的。这里指定的为后面的 Agent 类中的 premain 方法。同时也是JavaAgent的启动方法,Class-path中的值也是我们在maven中添加的包。你们应该没有JavaAgent技术之添加注解
  5. 编写一个JavaAgent方法
package com.annie;

import java.lang.instrument.Instrumentation;

public class Agent {

    // JavaAgent启动时调用的方法
    public static void premain(String args, Instrumentation instrumentation) {
        System.out.println("传入的参数为: " + args);
        System.out.println("JavaAgent启动了...");
    }

}

到这里一个JavaAgent项目就基本启动完成了。我们在创建一个普通项目(项目类型无所谓)

这里我创建的是一个SpringBoot的项目,因为比较好演示。
  1. 先将将JavaAgent编译jar包,编译好的jar包就在项目同级目录下的out下面。 JavaAgent技术之添加注解JavaAgent技术之添加注解
  2. 设置启动配置,后面一个Hello是传入的参数。可有可无。不要时不要忘记连 = 号也去掉。
-javaagent:C:\Users\13100\Documents\IdeaProjects\BaseJava\out\artifacts\JavaAgent_jar\JavaAgent.jar=Hello
  1. 保存运行JavaAgent技术之添加注解

到这里一个JavaAgent项目就基本搭建成功了。当然我们不能满足于此。这里我自演示基本的动态注解其实以下就没有和JaveAgent相关内容了。

  1. 添加javassist包。
  <!-- https://mvnrepository.com/artifact/org.javassist/javassist -->
    <dependency>
      <groupId>org.javassist</groupId>
      <artifactId>javassist</artifactId>
      <version>3.25.0-GA</version>
    </dependency>
  1. 编写代码(写完不要忘记重新编译)
package com.annie;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.bytecode.AnnotationsAttribute;
import javassist.bytecode.AttributeInfo;
import javassist.bytecode.ClassFile;
import javassist.bytecode.ConstPool;
import javassist.bytecode.annotation.Annotation;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.lang.instrument.Instrumentation;
import java.security.ProtectionDomain;
import java.util.List;

public class Agent {

    // JavaAgent启动时调用的方法
    public static void premain(String args, Instrumentation instrumentation) {
        System.out.println("传入的参数为: " + args);
        System.out.println("JavaAgent启动了...");

        // instrumentation 中包含了项目中的全部类。每加载一次.class文件就运行一次
        instrumentation.addTransformer(new ClassFileTransformer() {
            @Override
            public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {

                // 这里我只捕捉自己写 TestController 类
                if(className.equals("com/annie/controller/TestController")){
                    // 接下来就是javassist的使用我会粗略的写下备注。不会详解。
                    // 获取一个 class 池。
                    ClassPool classPool = ClassPool.getDefault();

                    try {
                        // 创建一个新的 class 类。classfileBuffer 就是当前class的字节码
                        CtClass ctClass = classPool.makeClass(new ByteArrayInputStream(classfileBuffer));
                        ClassFile classFile = ctClass.getClassFile();
                        ConstPool constPool = classFile.getConstPool();

                        // 从这里取出原本类中的注解 建议DEBUG看下attributes中的数据
                        List<AttributeInfo> attributes = classFile.getAttributes();
                        AnnotationsAttribute attributeInfo = (AnnotationsAttribute) attributes.get(1);
                        // 导包是javassist的包
                        // 添加新的注解
                        Annotation annotation = new Annotation("org.springframework.web.bind.annotation.RequestMapping", constPool);
                        attributeInfo.addAnnotation(annotation);

                        // 返回新的字节码
                        return ctClass.toBytecode();
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
                return new byte[0];
            }
        });
    }
}

  1. 编写一个 TestController,这里我只给了RestController的注解。并没有给RequestMapping注解。RequestMapping由JavaAgent为我添加。
package com.annie.controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.lang.annotation.Annotation;
@RestController
public class TestController {
    @RequestMapping("/test")
    public String test() {
        // 利用反射查看Class注解。
        Class<? extends TestController> aClass = this.getClass();
        Annotation[] declaredAnnotations = aClass.getDeclaredAnnotations();
        StringBuffer sb = new StringBuffer();
        // 查看类上全部注解
        for (Annotation annotation : declaredAnnotations) {
            sb.append(annotation.toString() + "\n");
        }
        return sb.toString();
    }
}
  1. 浏览器上输入 127.0.0.1:8080/test 查看TestController上的注解JavaAgent技术之添加注解

补充内容 linux 上发布项目

将javaAgent.jar 和 javaassist.jar 放入到一个文件夹中。
JavaAgent技术之添加注解
运行命令

java -javaagent:JavaAgent.jar -jar web-test-0.0.1-SNAPSHOT.jar

项目就启动成功。注意一点这个jar包就是你bulid出来的JAR


扩展内容
github
个人博客

上一篇:HDU-1052(贪心策略)


下一篇:100年idea注册码