javassist对已有的类进行操作

准备工作：

首先创建一个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.Student
        String name = cc.getSimpleName(); //获取类的简单名   Student
        CtClass 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 第一个参数为属性的类型，第二个参数为属性名，第三个参数为CtClass
        ctField.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());
            }
        }
    }

执行结果

 在这里就做这些介绍

当然javassist也是有一定的局限性的：

1、泛型、枚举是不支持的，

2、注解的修改也不支持，虽然可以通过底层的javassist类来解决

3、不支持数组的初始化工作，除非数组的容量为1

4、不支持内部类和匿名类

5、不支持continue和break

6、对于复杂的继承也不支持，支持简单的继承关系

原创不易，不喜勿喷，有更好的建议欢迎大家留言