最近在实现一个功能需要动态自动Java类,刚好了解了一下java buddy这个库。Java buddy是基于ASM实现,所以比ASM好用的多。还有一个名气更大的库mockito,其核心就是基于byte buddy实现的,可以动态生成mock类,非常方便。另外一个大的应用就是 java agent,其主要作用是在class 被加载之前对其拦截,插入自己的代码。
我的需求是在指定一个类,生成一个子类,在子类override父类的接口,返回指定的值。
具体实现代码如下:
类的生成代码:
Interceptor的实现:
Annotation 定义:
@Retention (RetentionPolicy.RUNTIME)
@Target (ElementType.METHOD)
Public @interface DefaultReturnValue {
String type();
String returnValue();
}
父类的的接口实现:
@DefaultReturnValue(type = “int”, returnValue = “8”)
Public int testReturnInt(){
Return 10;
}
首先NamingStrategy是对生成的类的一个命名策略,其中SuffixingRandom是在父类的基础上,在类名前加一个指定的前缀后面再加一个随机数。
调用subclass指定继承的父类,返回的是一个DynamicType.Builder,所以这里用的是一个Builder模式,具体可以做那些事情可以在Builder里面查看。
Method是对类里面的方法做处理,传入的参数是ElementMatchers的匹配器,这个匹配器可以对各种情况进行过滤,比如函数的返回值,参数,名字,annotation,那个类定义的等等情况。
Intercept是拦截器,对匹配到的对象做对应的操作,在这里我用到了两种情况,一种是对函数的返回值类型进行匹配,然后然后用FixedValue返回固定的值;另外一种就是通过对annotation进行匹配,然后重新指定一个拦截器类,转到拦截器类的实现。
下面一段代码定义了annotation的格式和内容,包括两个值。
Interceptor定义了一个拦截器,里面的实现根据不同的annotation值返回做不同的处理;这里面有两个拦截方法,bytebuddy会根据被拦截的函数自动帮你匹配,比如这里两个拦截函数分别返回不同类型的参数,那么对于被拦截的函数就会根据这个参数自动匹配,具体匹配规则可以深入研究一下。
这里传递进来的Method是被拦截的方法,注意这里用@origin标注,表示是被拦截的方法。
Load就是把函数加载进来,byte buddy定义了几种加载器,目前还没用到,没有研究,这里用最简单的系统类加载器。
最后可以用newInstance就可以生成对应的对象。
当然,在make()后,生成DynamicType.Unloaded后可以通过SaveIn()把生成的类写入存储器。
直接改写类,插入接口或属性。
这个库功能很强大,还有一些高级用法,还有java agent,java instrumentation等一些知识点,以后有机会再补充。
很好玩,很强大吧。