javassit是一个处理java字节码的类库;CtClass是一个class文件的抽象表示。
1.动态生成.class文件
public static void main(String[] args) {
//默认的类搜索路径
ClassPool classPool = ClassPool.getDefault();
//创建一个ctClass对象
CtClass ctClass = classPool.makeClass("com.lwj.mom.base.javassist.Student");
try {
//添加age属性
ctClass.addField(CtField.make("private int age;",ctClass));
//添加setAge方法
ctClass.addMethod(CtMethod.make("public void setAge(int age){this.age = age;}",ctClass));
//添加getAge方法
ctClass.addMethod(CtMethod.make("public int getAge(){return this.age;}", ctClass));
//将ctClass转换成字节码并写入文件
byte[] byteArray = ctClass.toBytecode();
FileOutputStream output = new FileOutputStream("D:\\Student.class");
output.write(byteArray);
output.close();
System.out.println("文件写入成功!!!");
} catch (CannotCompileException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
2.修改已经被jvm加载的.class文件
public static void main(String[] args) {
//默认的类搜索路径
ClassPool classPool = ClassPool.getDefault();
//创建一个ctClass对象
CtClass ctClass = classPool.makeClass("com.lwj.mom.base.javassist.Student");
try {
//添加age属性
ctClass.addField(CtField.make("private int age;",ctClass));
//添加setAge方法
ctClass.addMethod(CtMethod.make("public void setAge(int age){this.age = age;}",ctClass));
//添加getAge方法
ctClass.addMethod(CtMethod.make("public int getAge(){return this.age;}", ctClass));
//将ctClass转换成字节码并写入文件
byte[] byteArray = ctClass.toBytecode();
FileOutputStream output = new FileOutputStream("D:\\Student.class");
output.write(byteArray);
output.close();
System.out.println("文件写入成功!!!");
/*if(ctClass.isFrozen()){
ctClass.defrost();
}*/
//这里用pool.get()去获取ctClass对象,表示默认JVM已经加载此类.
ctClass = classPool.get("com.lwj.mom.base.javassist.Student");
ctClass.addField(CtField.make("private String sex;", ctClass));
ctClass.addField(CtField.make("private String name;", ctClass));
byteArray = ctClass.toBytecode();
output = new FileOutputStream("D:\\Student.class");
output.write(byteArray);
output.close();
System.out.println("文件修改成功!!!!");
} catch (CannotCompileException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} catch (NotFoundException e) {
e.printStackTrace();
}
}
发现报错,log如下
2.1 class is frozen(冻结class)原因
如果一个CtClass对象通过writeFile(),toClass()或者toByteCode()转换成class文件,那么javassist会冻结这个CtClass对象。后面就不能修改这个CtClass对象了。这样是为了警告开发者不要修改已经被JVM加载的class文件,因为JVM不允许重新加载一个类。
把这段代码打开,问题便得到了解决。
/*if(ctClass.isFrozen()){
ctClass.defrost();
}*/