javassist对已有的类进行操作

article/2025/9/15 4:49:31

准备工作：

首先创建一个java项目，然后导入javassist.jar包

创建一个注解

package com.chengyu.javassist;public @interface Auto {String name();int year();
}

创建一个接口

public interface Earth {
}

创建两个类

public class Pepelo{
}

package com.chengyu.javassist;@Auto(name="KO",year=1999)
public class Student extends Pepelo implements Earth{private int age;private String name;public int myTest(int a){System.out.println("我在学习。。。。" + a);return 6666;}public Student(){}public Student(int age,String name){this.age = age;this.name = 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;}
}

Student类继承Pepelo类实现Earth接口

然后我们开始测试

首先我们来对类进行操作，获取类的信息

 /*** javassist处理类的基本操作*/public static void test01() throws Exception {ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");byte [] bytes = cc.toBytecode(); //获取类的二进制字节码转成byte数组String allName = cc.getName();   //获取类的全限定名  com.chengyu.javassist.StudentString name = cc.getSimpleName(); //获取类的简单名   StudentCtClass superclass = cc.getSuperclass();    //获取类的父类CtClass[] interfaces = cc.getInterfaces();  //获取类的父接口System.out.println(allName);System.out.println(name);System.out.println(superclass.getName());System.out.println(interfaces[0].getName());System.out.println(Arrays.toString(bytes));}

新增一个方法

/*** 新增方法*/public static void test02() throws Exception {ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");//CtMethod make = CtNewMethod.make("public int add(int a,int b){return a+b;}",cc);  创建方法可以使用这种方法 也可以使用下面的方法CtMethod ctMethod = new CtMethod(CtClass.intType,"add",new CtClass[]{CtClass.intType,CtClass.intType},cc);//new CtMethod 的第一个参数为返回类型，第二个参数是方法名，第三个参数为方法的参数列表（这里没有给参数命名），第四个参数为CtClass类ctMethod.setModifiers(Modifier.PUBLIC); //声明方法的访问权限ctMethod.setBody("{return $1+$2;}");    //设置方法体 上面没有给参数命名，这里需要使用$符号进行匹配/*** $0 表示this  $1表示第一个形参   $2 表示第二个形参 以此类推* $args 表示的是一个Object[] 将参数列表放入这个数组中  args[0] 对应的就是 $1 args[1] 对应的就是$2 以此类推* $$ 所有方法参数的简写，主要用在方法调用上* $cflow 一个方法调用的深度，主要用于递归调用上* $r 方法返回值的类型* $_  方法的返回值（修改方法体时不支持）* addCatch() 方法中加入try catch块* $e 表示异常对象* $class this的类型（Class） 也就是$0的类型* $sig 方法参数的类型（Class）数组，数组的顺序为参数的顺序**/cc.addMethod(ctMethod); //将方法添加到类中//下面使用java反射来进行调用验证Class clazz = cc.toClass();Object obj = clazz.newInstance();Method declaredMethod = clazz.getDeclaredMethod("add",int.class, int.class);Object invoke = declaredMethod.invoke(obj, 200, 300);System.out.println("result:" + (int)invoke);}

执行结果

对已有的方法进行修改

 /*** 修改已有的方法*/public static void test03() throws Exception {ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");//根据方法名和参数列表来获取方法CtMethod myTest = cc.getDeclaredMethod("myTest", new CtClass[]{CtClass.intType});myTest.insertBefore("System.out.println($1 + \" 准备打开书\");"); //在方法前进行插入代码myTest.insertAfter("System.out.println(\"已经学完了，出去玩\");"); //在方法后面进行插入代码myTest.insertAt(16,"System.out.println(\"这是在某一行插入的代码！\");"); //在某一行插入代码//使用反射进行调用测试Class clazz = cc.toClass();Object obj = clazz.newInstance();Method declaredMethod = clazz.getDeclaredMethod("myTest",int.class);Object invoke = declaredMethod.invoke(obj, 200);System.out.println("result:" + (int)invoke);}

执行的结果

对属性的操作-新增一个属性并且为其添加set和get方法

 /*** 新增属性并且添加set和get方法*/public static void test04() throws Exception{ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");//CtField make = CtField.make("private String phone;", cc);  可以通过这种方式添加属性，也可以使用下面的方式CtField ctField = new CtField(pool.get("java.lang.String"),"phone",cc);//new CtField 第一个参数为属性的类型，第二个参数为属性名，第三个参数为CtClassctField.setModifiers(Modifier.PRIVATE);  //设置属性的访问权限cc.addField(ctField);                    //将属性添加到方法中//CtField phone = cc.getDeclaredField("phone");  这里可以根据属性名来获取属性//添加set和get方法 第一个参数为方法名 第二个参数为属性cc.addMethod(CtNewMethod.getter("getPhone",ctField));cc.addMethod(CtNewMethod.setter("setPhone",ctField));//通过反射进行调用测试Class clazz = cc.toClass();Object obj = clazz.newInstance();Method setPhone = clazz.getDeclaredMethod("setPhone",String.class);Method getPhone = clazz.getDeclaredMethod("getPhone",null);setPhone.invoke(obj,"13088888888");Object phone = getPhone.invoke(obj);System.out.println(String.valueOf(phone));}

执行结果：

对构造方法的操作

   public static void test05()throws Exception{ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");CtConstructor[] constructors = cc.getConstructors();for(CtConstructor c : constructors){System.out.println(c.getLongName());//打印构造方法的全类限定名c.insertBefore("System.out.println(\"我是构造方法\");"); //当然我们也可以对构造方法进行操作}Class clazz = cc.toClass();Object obj = clazz.newInstance();}

执行结果：

对注解的操作

public static void test06()throws Exception{ClassPool pool = ClassPool.getDefault();CtClass cc = pool.get("com.chengyu.javassist.Student");Object [] all = cc.getAnnotations();  //获取标注在类上的所有注解for(Object a : all){if(a instanceof Auto){Auto b = (Auto) a;System.out.println("name : " + b.name() + " year : " + b.year());}}}

执行结果