[编织消息框架][JAVA核心技术]动态代理应用5-javassist

2023-03-18 11:14:18

基础部份：

修改class我们用到javassist，在pom.xml添加

<properties>

    <javassist.version>3.18.2-GA</javassist.version>

</properties>

<dependency>

    <groupId>org.javassist</groupId>

    <artifactId>javassist</artifactId>

    <version>${javassist.version}</version>

</dependency>

1.ClassPool 负责加载CtClass，其中可添加ClassPath，优先级从父ClassPath加载
2.CtClass 处理class信息从ClassPool make
3.CtField 处理field信息
4.CtMethod 处理method信息
5.CtConstructor 处理constructor信息

ClassPool 源码浅读：

AccessController.doPrivileged 方法能够使一段受信任代码获得更大的权限，甚至比调用它的应用程序还要多，可做到临时访问更多的资源，

由于技术内容跟本书无太大关系，我们只关心当抛出java.security.AccessControlException异常时，就考虑添加上去即可

实现部份：

 public class TestJavassist {

     public interface TestObject {

     public void a(int a, String b);

     public void setAge(int value);

     public int getAge();

     }

     public static void main(String[] args) throws Exception {

         String proxyClassName = TestObject.class.getCanonicalName() + "$$$$";

         ClassPool classPool = ClassPool.getDefault();

         CtClass ctClass = classPool.makeClass(proxyClassName);

         // 设置接口

         CtClass[] interfaces = new CtClass[1];

         interfaces[0] = classPool.get(TestObject.class.getName());

         ctClass.setInterfaces(interfaces);

         // 添加字段

         CtField ctField = new CtField(classPool.get(int.class.getName()), "age", ctClass);

         // 设置field属性

         ctField.setModifiers(Modifier.PRIVATE);

         ctClass.addField(ctField);

         // 添加 age getter'setter方法

         String setbody = "{this.age = $1;}";

         String getbody = "{return this.age;}";

         for (Method method : TestObject.class.getDeclaredMethods()) {

             String body = null;

             switch (method.getName()) {

                 case "setAge":

                     body = setbody;

                 break;

                 case "getAge":

                     body = getbody;

                 break;

                 case "a":

                     body = "{System.out.println($1);System.out.println($2);}";

                 break;

                 default:

                     continue;

             }

             Class<?> returnType = method.getReturnType();

             // 转换参数CtClass

             CtClass[] parameterCtClass = new CtClass[method.getParameterTypes().length];

             for (int i = 0; i < method.getParameterTypes().length; i++) {

                 parameterCtClass[i] = classPool.get(method.getParameterTypes()[i].getName());

             }

             CtMethod ctMethod = new CtMethod(classPool.get(returnType.getName()), method.getName(), parameterCtClass, ctClass);

             ctMethod.setModifiers(method.getModifiers());

             ctMethod.setBody(body);

             ctClass.addMethod(ctMethod);

         }

         // 添加构造

         ctClass.addConstructor(CtNewConstructor.defaultConstructor(ctClass));

         Class<?> clz = ctClass.toClass();

         TestObject obj = (TestObject) clz.newInstance();

         obj.setAge(30);

         System.out.println("age : " + obj.getAge());

         obj.a(111, "bbb");

         // ctClass.writeFile("f:/test.class");

     }

 }

源码讲解

1.先通过classPool.makeClass 创建新的类，名称是原来类+自定义标记

2.所有的java类型必须转换成CtClass

3.要给创建的自定义类CtClass 添加接口类或继承父类，保留原类型

4.创建方法时特别注意处理java类型

5.可添加构造方法

下面给出常用的参数对照表，注意：在不同域使用不同有的少点，不过都大同小异

接着在之前的TestProxy 测试添加 testJavassist 看下执行效率，速度与native相差不大，原因是生成的指令很少，感兴趣的话通过 ctClass.writeFile()保存class 然后再用javap查看生成指令数

 public static void testJavassist() throws Exception {

     String proxyClassName = UserService.class.getCanonicalName() + "$$$$";

     ClassPool classPool = ClassPool.getDefault();

     CtClass ctClass = classPool.makeClass(proxyClassName);

     // 设置接口

     CtClass[] interfaces = new CtClass[1];

     interfaces[0] = classPool.get(UserService.class.getName());

     ctClass.setInterfaces(interfaces);

     // 添加 方法

     for (Method method : UserService.class.getDeclaredMethods()) {

         String body = null;

         switch (method.getName()) {

         case "getName":

             body = "{return $1 + \"\";}";

         break;

         case "getAge":

             body = "{return ($w)$1;}";

         break;

         default:

             continue;

         }

         Class<?> returnType = method.getReturnType();

         // 转换参数CtClass

         CtClass[] parameterCtClass = new CtClass[method.getParameterTypes().length];

         for (int i = 0; i < method.getParameterTypes().length; i++) {

             parameterCtClass[i] = classPool.get(method.getParameterTypes()[i].getName());

         }

         CtMethod ctMethod = new CtMethod(classPool.get(returnType.getName()), method.getName(), parameterCtClass, ctClass);

         ctMethod.setModifiers(method.getModifiers());

         ctMethod.setBody(body);

         ctClass.addMethod(ctMethod);

     }

     // 添加构造

     ctClass.addConstructor(CtNewConstructor.defaultConstructor(ctClass));

     Class<?> clz = ctClass.toClass();

     UserService proxy = (UserService) clz.newInstance();

     run("javassist", proxy);

 }

native: 177 native: 183 native: 128 native: 171 native: 127 native: 141 native: 184 native: 126 native: 127 native: 127 native: 127 native: 126 native: 127 native: 126 native: 127
jdk: 214 jdk: 169 jdk: 169 jdk: 169 jdk: 169 jdk: 169 jdk: 170 jdk: 169 jdk: 170 jdk: 169 jdk: 169 jdk: 171 jdk: 170 jdk: 169 jdk: 170
javassist: 146 javassist: 139 javassist: 140 javassist: 140 javassist: 140 javassist: 139 javassist: 140 javassist: 140 javassist: 138 javassist: 140 javassist: 140 javassist: 141 javassist: 139 javassist: 139 javassist: 139

资料来源：http://jboss-javassist.github.io/javassist/tutorial/tutorial2.html#runtime

http://wsmajunfeng.iteye.com/blog/1912983

码农公寓

基础部份：

实现部份：

相关文章