Java动态字节技术之Javassist

createh53周前 (12-11)技术教程20

概述

Javassist是一个开源的分析、编辑和创建Java字节码的类库,可以直接编辑和生成Java生成的字节码。相对于bcel, asm等这些工具,开发者不需要了解虚拟机指令,就能动态改变类的结构,或者动态生成类。javassist简单易用, 快速。

重要的类

1. ClassPool:javassist的类池,使用ClassPool 类可以跟踪和控制所操作的类,它的工作方式与 JVM 类装载器非常相似

2. CtClass: CtClass提供了类的操作,如在类中动态添加新字段、方法和构造函数、以及改变类、父类和接口的方法。

3. CtField:类的属性,通过它可以给类创建新的属性,还可以修改已有的属性的类型,访问修饰符等

4. CtMethod:类中的方法,通过它可以给类创建新的方法,还可以修改返回类型,访问修饰符等, 甚至还可以修改方法体内容代码

5. CtConstructor:与CtMethod类似

API运用

ClassPool

1	// 类库, jvm中所加载的class
2	ClassPool pool = ClassPool.getDefault();
3	// 加载一个已知的类, 注:参数必须为全量类名
4	CtClass ctClass = pool.get("com.itheima.Student");
5	// 创建一个新的类, 类名必须为全量类名
6	CtClass tClass = pool.makeClass("com.itheima.Calculator");

CtField

1	// 获取已知类的属性
2	CtField ctField = ctClass.getDeclaredField("name");
3	// 构建新的类的成员变量
4	CtField ctFieldNew = new CtField(CtClass.intType,"age",ctClass);
5	// 设置类的访问修饰符为public
6	ctFieldNew.setModifiers(Modifier.PUBLIC);
7	// 将属性添加到类中
8	ctClass.addField(ctFieldNew);

CtMethod

1	// 获取已有方法
4	//创建新的方法, 参数1:方法的返回类型,参数2:名称,参数3:方法的参数,参数4:方法所属的类
5	CtMethod ctMethod = new CtMethod(CtClass.intType, "calc", new CtClass[]
{CtClass.intType,CtClass.intType}, tClass);
6	// 设置方法的访问修饰
7	ctMethod.setModifiers(Modifier.PUBLIC);
8	// 将新建的方法添加到类中
9	ctClass.addMethod(ctMethod);
10	// 方法体内容代码 $1代表第一个参数,$2代表第二个参数
11	ctMethod.setBody("return $1 + $2;"); 

2	CtMethod ctMethod = ctClass.getDeclaredMethod("sayHello");

CtConstructor

// 获取已有的构造方法, 参数为构建方法的参数类型数组
CtConstructor ctConstructor = ctClass.getDeclaredConstructor(new CtClass[]{});
// 创建新的构造方法
CtConstructor ctConstructor = new CtConstructor(new CtClass[]{CtClass.intType},ctClass); ctConstructor.setModifiers(Modifier.PUBLIC);
ctConstructor.setBody("this.age = $1;");
ctClass.addConstructor(ctConstructor);
// 也可直接创建
ctConstructor = CtNewConstructor.make("public Student(int age){this.age=age;}", ctClass);

案例

1. 创建maven工程并添加依赖

添加依赖

添加Student类

package com.itheima;
public class Student {
private int age; private String name;
public int getAge() { return age;
}
public void setAge(int age) { this.age = age;
}
public String getName() { return name;
}
public void setName(String name) { this.name = name;
}
public void sayHello(){
System.out.println("Hello, My name is " + this.name);
}
}

2. 修改已有方法体,插入新的代码

对已有的student类中的sayHello方法,当调用时,控制台会输出: Hello, My name is 张三(name=张三)

需求:通过动态修改sayHello方法,当调用sayHello时,除了输出已经的内容外,再输出当前学生的age信息 创建JavassistDemo测试类,代码实现如下:

package com.itheima.test; import org.junit.Test; import com.itheima.Student;
import javassist.ClassPool; import javassist.CtClass; import javassist.CtMethod;

@SuppressWarnings("unchecked") public class JavassistDemo {
	
	@Test
	public void t1() throws Exception{
	// 类库池, jvm中所加载的class
	ClassPool pool = ClassPool.getDefault();
	// 获取指定的Student类
	CtClass ctClass = pool.get("com.itheima.Student");
	// 获取sayHello方法
	CtMethod ctMethod = ctClass.getDeclaredMethod("sayHello");
	// 在方法的代码后追加 一段代码
	ctMethod.insertAfter("System.out.println(\"I'm \" + this.age + \" years old.\");");
	// 使用当前的ClassLoader加载被修改后的类
	Class<Student> newClass = ctClass.toClass();
	
	Student stu = newClass.newInstance();
	stu.setName("张三");
	stu.setAge(18);
	
	stu.sayHello();
	}
}	

运行结果:

3. 动态添加方法

接下来我们给Student类添加一个计算的方法,但不是直接在Student类中添加,而是使用javassist,动态添加

1	public int calculate(int a, int b){
2	return a+b;
3	}
创建测试方法t2,代码如下
1 @Test
2 public void t2() throws Exception{
3 // 类库池, jvm中所加载的class
4 ClassPool pool = ClassPool.getDefault();
5 // 获取指定的Student类
6 CtClass ctClass = pool.get("com.itheima.Student");
7 // 创建calc方法, 带两个参数,参数的类型都为int类型
8 CtMethod ctMethod = new CtMethod(CtClass.intType,"calc",
9 new CtClass[]{CtClass.intType,CtClass.intType}, ctClass);
10 // 设置方法的访问修饰
11 ctMethod.setModifiers(Modifier.PUBLIC);
12 // 设置方法体代码
13 ctMethod.setBody("return $1 + $2;");
14 // 添加新建的方法到原有的类中
15 ctClass.addMethod(ctMethod);
16 // 加载修改后的类
17 ctClass.toClass();
18 // 创建对象
19 Student stu = new Student();
20 // 获取calc方法
21 Method dMethod = Student.class.getDeclaredMethod("calc", new Class[]
{int.class,int.class});
22 // 反射调用 方法
23 Object result = dMethod.invoke(stu, 10,20);
24 // 打印结果
25 System.out.println(String.format("调用calc方法,传入参数:%d,%d", 10,20));
26 System.out.println("返回结果:" + (int)result);
27 }

控制台输出结果:

4. 动态创建类

下面我们再来个神的魔术,无中生有

1 @Test
2 public void t3() throws Exception{
3 ClassPool pool = ClassPool.getDefault();
4 // 创建teacher类
5 CtClass teacherClass = pool.makeClass("com.itheima.Teacher");
6 // 设置为公有类
7 teacherClass.setModifiers(Modifier.PUBLIC);
8 // 获取String类型
9 CtClass stringClass = pool.get("java.lang.String");
10 // 获取list类型
11 CtClass listClass = pool.get("java.util.List");
12 // 获取学生的类型
13 CtClass studentClass = pool.get("com.itheima.Student");
14 // 给teacher添加name属性
15 CtField nameField = new CtField(stringClass, "name", teacherClass);
16 nameField.setModifiers(Modifier.PUBLIC);
17 teacherClass.addField(nameField);
18 // 给teacher类添加students属性
19 CtField studentList = new CtField(listClass, "students",teacherClass);
20 studentList.setModifiers(Modifier.PUBLIC);
21 teacherClass.addField(studentList); 22
23 // 给teacher类添加无参构造方法
24 CtConstructor ctConstructor = CtNewConstructor.make("public Teacher()
{this.name=\"abc\";this.students = new java.util.ArrayList();}", teacherClass);
25 teacherClass.addConstructor(ctConstructor); 26
27 // 给teacher类添加addStudent方法
28 CtMethod m = new CtMethod(CtClass.voidType, "addStudent", new CtClass[]{studentClass}, teacherClass);
29 m.setModifiers(Modifier.PUBLIC);
30 // 添加学生对象到students属性中, $1代表参数1
31 m.setBody("this.students.add($1);");
32 teacherClass.addMethod(m); 33
34 // 给teacher类添加sayHello方法
35 m = new CtMethod(CtClass.voidType, "sayHello", new CtClass[]{}, teacherClass);
36 m.setModifiers(Modifier.PUBLIC);
37 m.setBody("System.out.println(\"Hello, My name is \" + this.name);");
38 m.insertAfter("System.out.println(\"I have \" + this.students.size() + \" students\");");
39 teacherClass.addMethod(m); 40
41 // 加载修改后的类
42 Class<?> cls = teacherClass.toClass();
43 // 实例teacher对象
44 Object obj = cls.newInstance();
45 // 获取addStudent方法
46 Method method = cls.getDeclaredMethod("addStudent", Student.class);

运行结果:

总结

javassist被用于struts2和hibernate中,都用来做动态字节码修改使用。一般开发中不会用到,但在封装框架时比 较有用。虽然javassist提供了一套简单易用的API,但如果用于平常的开发,会有如下几点不好的地方:

  • 1. 所引用的类型,必须通过ClassPool获取后才可以使用
  • 2. 代码块中所用到的引用类型,使用时必须写全量类名
  • 3. 即使代码块内容写错了,它也不会像eclipse等开发工具一样有提示,它只有在运行时才报错
  • 4. 动态修改的类,必须在修改之前,jvm中不存在这个类的实例对象。修改方法的实现必须在修改的类加载之前进行。

相关文章

Java性能测试利器:JMH入门与实践|得物技术

目录一、JMH概述1. JMH的主要特点2. JMH与其他性能测试工具的比较二、快速开始1. 原型方式生成Maven项目2. 已有项目配置JMH三、JMH的核心概念和注解四、JMH的工作原理1. JV...

深入理解Java虚拟机——JMM(Java内存模型)

1、 为什么要有Java内存模型?1.1、 CPU和缓存一致性1. 缓存一致性问题出现的原因CPU的执行速度和内存的读取速度差距越来越大,导致CPU每次操作内存都要耗费很多等待时间。为解决这个问题,早...

Java底层-JMX

JMX(Java Management Extensions,Java管理扩展)在Java编程语言中定义了应用程序以及网络管理和监控的体系结构、设计模式、应用程序接口以及服务。通常使用JMX来监控系统...

看完这篇,还不懂JAVA内存模型(JMM)算我输

前言开篇一个例子,我看看都有谁会?如果不会的,或者不知道原理的,还是老老实实看完这篇文章吧。@Slf4j(topic = "c.VolatileTest") public class...

Java——API

腾讯云 API 全新升级3.0,该版本进行了性能优化且全地域部署、支持就近和按地域接入、访问时延下降显著,接口描述更加详细、错误码描述更加全面、SDK 增加接口级注释,让您更加方便快捷的使用腾讯云产品...

带你了解“Java新特性——模块化”

阶行 大淘宝技术 2024年07月05日 18:57 浙江Java平台从Java 8向Java 9及更高版本的进化,其中引入了一个重要的新特性——模块系统(Project Jigsaw)。模块系统的目...