JAVA_javassist学习

javassist
用于对java字节码文件进行操作，可以动态生成新的类、方法、添加属性等，亦可对已有类、方法、属性进行修改

需要导入的jar
javassist.jar

1.使用javassist生成一个新的类

package com.bjsxt.test;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtField;
import javassist.CtMember;
import javassist.CtMethod;

/**
 * 测试使用javassist生成一个新的类
 * @author WL20180732
 *
 */
public class Demo01 {
	public static void main(String[] args) throws Exception {
		ClassPool pool = ClassPool.getDefault();
		CtClass cc = pool.makeClass("com.bjsxt.bean.Emp");
		
		// 创建属性
		CtField f1 = CtField.make("private int empno;", cc);
		CtField f2 = CtField.make("private String ename;", cc);
		cc.addField(f1);
		cc.addField(f2);
		
		// 创建方法
		CtMethod m1 = CtMethod.make("public int getEmpno() {return this.empno;}", cc);
		CtMethod m2 = CtMethod.make("public void setEmpno(int empno) {this.empno = empno;}", cc);
		cc.addMethod(m1);
		cc.addMethod(m2);
		
		// 添加构造器
		CtConstructor constructor = new CtConstructor(
				new CtClass[]{CtClass.intType, pool.get("java.lang.String")}, cc);
		
		constructor.setBody("{this.empno = empno; this.ename = ename;}");
		cc.addConstructor(constructor);
		cc.writeFile("E:/mycode"); // 将上面构造好的类写入指定目录下
		System.out.println("类生成成功");
	}
}

2.测试javassist的API

package com.bjsxt.test;

import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Arrays;

import javax.security.sasl.AuthorizeCallback;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtField;
import javassist.CtMethod;
import javassist.CtNewMethod;
import javassist.Modifier;
import javassist.NotFoundException;

/**
 * 测试javassist的API
 * @author WL20180732
 *
 */
public class Demo02 {
	
	/**
	 * 处理类的基本用法
	 * @throws NotFoundException 
	 * @throws CannotCompileException 
	 * @throws IOException 
	 */
	public static void test01() throws Exception {
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		
		byte[] bytes = cc.toBytecode();
		System.out.println(Arrays.toString(bytes));
		
		System.out.println(cc.getName()); // 获取全类名
		System.out.println(cc.getSimpleName()); // 获取类名
		System.out.println(cc.getSuperclass()); // 获得父类
		System.out.println(cc.getInterfaces()); // 获取接口
	}
	
	/**
	 * 测试产生新的方法
	 * @throws Exception 
	 */
	public static void test02() throws Exception {
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		
//		CtMethod m = CtNewMethod.make("public int add(int a, int b) {return a+b;}", cc);
		CtMethod m = new CtMethod(
				CtClass.intType,   // 返回值类型
				"add", 
				new CtClass[]{CtClass.intType, CtClass.intType}, cc);
		
		m.setModifiers(Modifier.PUBLIC);  //设置方法修饰符
		m.setBody("{System.out.println($1 + $2); return $1 + $2;}");  // $0：this关键字，$$: 方法的所有参数，$1：方法的第一个参数，$2:方法的第二个参数...
		
		cc.addMethod(m);
		
		//通过反射调用生成的方法
		Class clazz = cc.toClass();
		Object obj = clazz.newInstance();  // 通过调用emp的无参构造，创建新的Emp对象
		Method method = clazz.getDeclaredMethod("add", int.class, int.class);
		Object result = method.invoke(obj, 200, 300);
		System.out.println(result);
	}
	
	/**
	 * 修改已有的方法信息
	 * @throws Exception
	 */
	public static void test03() throws Exception{
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		
		CtMethod cm = cc.getDeclaredMethod("sayHello", new CtClass[]{CtClass.intType});
		cm.insertBefore("System.out.println($1);System.out.println(\"start!!\");");
		cm.insertAfter("System.out.println(\"end!!\");");
		cm.insertAt(27, "System.out.println(\"lineNum=\" + 27);"); // 在某一行处加代码
		//通过反射调用生成的方法
		Class clazz = cc.toClass();
		Object obj = clazz.newInstance();  // 通过调用emp的无参构造，创建新的Emp对象
		Method method = clazz.getDeclaredMethod("sayHello", int.class);
		Object result = method.invoke(obj, 200);
		System.out.println(result);
	}
	
	/**
	 * 属性的操作
	 * @throws Exception
	 */
	public static void test04() throws Exception{
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		
//		CtField f1 = CtField.make("private int empno;", cc);
		CtField f1 = new CtField(CtClass.intType, "salary", cc);
		f1.setModifiers(Modifier.PRIVATE);
		cc.addField(f1);
		cc.addMethod(CtNewMethod.getter("getSalary", f1)); // 给f1属性添加getter方法
		cc.addMethod(CtNewMethod.getter("setSalary", f1)); // 给f1属性添加setter方法
//		cc.getDeclaredField("ename");  // 获取指定属性
		
		
	}
	
	/**
	 * 构造方法的操作
	 * @throws Exception
	 */
	public static void test05() throws Exception{
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		
		CtConstructor[] constructors = cc.getConstructors();
		for (CtConstructor ctConstructor : constructors) {
			System.out.println(ctConstructor.getLongName());
//			ctConstructor.insertBefore(src);
		}
	}
	
	public static void test06() throws Exception{
		ClassPool classPool = ClassPool.getDefault();
		CtClass cc = classPool.get("com.bjsxt.test.Emp");  // 获得已有类
		Object[] all = cc.getAnnotations();
		Author a = (Author) all[0];
		String name = a.name();
		int year = a.year();
		System.out.println("name="+name + ",year=" + year);
	}
	
	public static void main(String[] args) throws Exception {
//		test01();
//		test02();
//		test03();
//		test04();
//		test05();
		test06();
	}
}

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(value={ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface Author {
	String name();
	int year();
}

package com.bjsxt.test;

@Author(name="azor", year=2020)
public class Emp {
	
	private int empno;
	private String ename;
	public int getEmpno() {
		return empno;
	}
	public void setEmpno(int empno) {
		this.empno = empno;
	}
	public String getEname() {
		return ename;
	}
	public void setEname(String ename) {
		this.ename = ename;
	}
	public Emp() {
	}
	public Emp(int empno, String ename) {
		this.empno = empno;
		this.ename = ename;
	}
	
	public void sayHello(int a) {
		System.out.println("say Hello," + a);
	}
	
}