java agent
spinrgboot+Javassist
aop运行于应用程序内的方法级别,而agent能够作用于虚拟机级别
对静态方法的改造
<dependency>
<groupId>org.javassist</groupId>
<artifactId>javassist</artifactId>
<version>3.28.0-GA</version> <!-- 版本号可能会更新 -->
</dependency>
@Slf4j
public class AgentUtil {
public static String getToken(){
return "123"+ UUID.randomUUID();
}
public static String doThing(){
log.info("====>>> start dong thing");
return "doing things";
}
}
public class Agenttest001Application {
public static void main(String[] args) throws Exception{
SpringApplication.run(Agenttest001Application.class, args);
ClassPool pool = ClassPool.getDefault();
// 获取目标类的 CtClass 对象
CtClass ctClass = pool.get("com.jin.agenttest001.utils.AgentUtil");
// 获取目标静态方法的 CtMethod 对象
CtMethod method = ctClass.getDeclaredMethod("getToken");
// 修改方法体为返回新的字符串
method.setBody("{ return \"abc\"; }");
// 将修改后的类重新加载到 JVM 中
ctClass.toClass();
}
}
在 Javassist 中, 1 是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地, 1 是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地, 1是一个占位符,用于表示传递给方法或构造函数的第一个参数。它是在修改方法体时用来引用原始方法签名中的参数的一种方式。更一般地,n 表示第 n 个参数(从1开始计数),而 $0 则代表当前对象实例(类似于 Java 中的 this) ,对修改构造方法注意下
public static void mockStudent(ClassPool pool)throws Exception{
CtClass ctClass = pool.get("com.jin.agenttest001.dto.Student");
// 获取目标构造方法的 CtConstructor 对象
CtConstructor[] constructors = ctClass.getDeclaredConstructors();
for (CtConstructor constructor : constructors) {
if (constructor.getParameterTypes().length == 1 &&
constructor.getParameterTypes()[0].getName().equals("int")) {
// 修改构造方法体
constructor.setBody("{ this.name = \"jin\"; this.age = $1; }");
break;
}
}
CtMethod fun2 = ctClass.getDeclaredMethod("fun2");
fun2.setBody("{ return \"fun2\"; }");
ctClass.toClass();
}
对注入的类,顺序稍微调整下
完整代码:
https://gitee.com/jinfei602917226/agent/tree/master/src/main/java/com/jin/agenttest001