javassist使用指南

<dependency>
 <groupId>org.javassist</groupId>
 <artifactId>javassist</artifactId>
 <version>3.25.0-GA</version>
</dependency>

package com.rickiyang.learn.javassist;

import javassist.*;

/**
 * @author rickiyang
 * @date 2019-08-06
 * @Desc
 */
public class CreatePerson {

 /**
 * 创建一个Person 对象
 *
 * @throws Exception
 */
 public static void createPseson() throws Exception {
 ClassPool pool = ClassPool.getDefault();

 // 1. 创建一个空类
 CtClass cc = pool.makeClass("com.rickiyang.learn.javassist.Person");

 // 2. 新增一个字段 private String name;
 // 字段名为name
 CtField param = new CtField(pool.get("java.lang.String"), "name", cc);
 // 访问级别是 private
 param.setModifiers(Modifier.PRIVATE);
 // 初始值是 "xiaoming"
 cc.addField(param, CtField.Initializer.constant("xiaoming"));

 // 3. 生成 getter、setter 方法
 cc.addMethod(CtNewMethod.setter("setName", param));
 cc.addMethod(CtNewMethod.getter("getName", param));

 // 4. 添加无参的构造函数
 CtConstructor cons = new CtConstructor(new CtClass[]{}, cc);
 cons.setBody("{name = \"xiaohong\";}");
 cc.addConstructor(cons);

 // 5. 添加有参的构造函数
 cons = new CtConstructor(new CtClass[]{pool.get("java.lang.String")}, cc);
 // $0=this / $1,$2,$3... 代表方法参数
 cons.setBody("{$0.name = $1;}");
 cc.addConstructor(cons);

 // 6. 创建一个名为printName方法，无参数，无返回值，输出name值
 CtMethod ctMethod = new CtMethod(CtClass.voidType, "printName", new CtClass[]{}, cc);
 ctMethod.setModifiers(Modifier.PUBLIC);
 ctMethod.setBody("{System.out.println(name);}");
 cc.addMethod(ctMethod);

 //这里会将这个创建的类对象编译为.class文件
 cc.writeFile("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");
 }

 public static void main(String[] args) {
 try {
  createPseson();
 } catch (Exception e) {
  e.printStackTrace();
 }
 }
}

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//

package com.rickiyang.learn.javassist;

public class Person {
 private String name = "xiaoming";

 public void setName(String var1) {
 this.name = var1;
 }

 public String getName() {
 return this.name;
 }

 public Person() {
 this.name = "xiaohong";
 }

 public Person(String var1) {
 this.name = var1;
 }

 public void printName() {
 System.out.println(this.name);
 }
}

// 这里不写入文件，直接实例化
Object person = cc.toClass().newInstance();
// 设置值
Method setName = person.getClass().getMethod("setName", String.class);
setName.invoke(person, "cunhua");
// 输出值
Method execute = person.getClass().getMethod("printName");
execute.invoke(person);

ClassPool pool = ClassPool.getDefault();
// 设置类路径
pool.appendClassPath("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");
CtClass ctClass = pool.get("com.rickiyang.learn.javassist.Person");
Object person = ctClass.toClass().newInstance();
// ...... 下面和通过反射的方式一样去使用

package com.rickiyang.learn.javassist;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public interface PersonI {

 void setName(String name);

 String getName();

 void printName();

}

ClassPool pool = ClassPool.getDefault();
pool.appendClassPath("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");

// 获取接口
CtClass codeClassI = pool.get("com.rickiyang.learn.javassist.PersonI");
// 获取上面生成的类
CtClass ctClass = pool.get("com.rickiyang.learn.javassist.Person");
// 使代码生成的类，实现 PersonI 接口
ctClass.setInterfaces(new CtClass[]{codeClassI});

// 以下通过接口直接调用 强转
PersonI person = (PersonI)ctClass.toClass().newInstance();
System.out.println(person.getName());
person.setName("xiaolv");
person.printName();

package com.rickiyang.learn.javassist;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public class PersonService {

 public void getPerson(){
 System.out.println("get Person");
 }

 public void personFly(){
 System.out.println("oh my god,I can fly");
 }
}

package com.rickiyang.learn.javassist;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.Modifier;

import java.lang.reflect.Method;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public class UpdatePerson {

 public static void update() throws Exception {
 ClassPool pool = ClassPool.getDefault();
 CtClass cc = pool.get("com.rickiyang.learn.javassist.PersonService");

 CtMethod personFly = cc.getDeclaredMethod("personFly");
 personFly.insertBefore("System.out.println(\"起飞之前准备降落伞\");");
 personFly.insertAfter("System.out.println(\"成功落地。。。。\");");


 //新增一个方法
 CtMethod ctMethod = new CtMethod(CtClass.voidType, "joinFriend", new CtClass[]{}, cc);
 ctMethod.setModifiers(Modifier.PUBLIC);
 ctMethod.setBody("{System.out.println(\"i want to be your friend\");}");
 cc.addMethod(ctMethod);

 Object person = cc.toClass().newInstance();
 // 调用 personFly 方法
 Method personFlyMethod = person.getClass().getMethod("personFly");
 personFlyMethod.invoke(person);
 //调用 joinFriend 方法
 Method execute = person.getClass().getMethod("joinFriend");
 execute.invoke(person);
 }

 public static void main(String[] args) {
 try {
  update();
 } catch (Exception e) {
  e.printStackTrace();
 }
 }
}