您的位置: 首页  >  文章  >  CGLib之Enhancer

CGLib之Enhancer

分类: 文章 • 2024-09-19 15:39:10

Enhancer允许为非接口类型创建一个Java代理。Enhancer动态创建了给定类型的子类但是拦截了所有的方法。和Proxy不一样的是，不管是接口还是类他都能正常工作。

来个场景模拟一下AOP

[java] view plain copy
  1. package cglib.enhancer;  
  2.   
  3. public class Hello {  
  4.     public String sayHello(boolean throwException) throws Exception {  
  5.         System.out.println("hello everyone!");  
  6.         if(throwException)  
  7.             throw new Exception("test exception");  
  8.         return "123";  
  9.     }  
  10. }  

ProxyFactory用于创建增强代理实现了方法拦截

[java] view plain copy
  1. package cglib.enhancer;  
  2.   
  3. import java.lang.reflect.Method;  
  4.   
  5. import net.sf.cglib.proxy.Enhancer;  
  6. import net.sf.cglib.proxy.MethodInterceptor;  
  7. import net.sf.cglib.proxy.MethodProxy;  
  8.   
  9. public class ProxyFactory implements MethodInterceptor {  
  10.      //要代理的原始对象  
  11.     private Object obj;  
  12.     public Object createProxy(Object target) {  
  13.         this.obj = target;  
  14.         Enhancer enhancer = new Enhancer();  
  15.         enhancer.setSuperclass(this.obj.getClass());// 设置代理目标  
  16.         enhancer.setCallback(this);// 设置回调  
  17.         enhancer.setClassLoader(target.getClass().getClassLoader());  
  18.         return enhancer.create();  
  19.     }  
  20.   
  21.     @Override  
  22.     public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {  
  23.         Object result = null;  
  24.         try {  
  25.             // 前置通知  
  26.             before();  
  27.             result = proxy.invokeSuper(obj, args);  
  28.             // 后置通知  
  29.             after();  
  30.         } catch (Exception e) {  
  31.             exception();  
  32.         }finally{  
  33.             beforeReturning();  
  34.         }  
  35.         return result;  
  36.     }  
  37.   
  38.   
  39.     private void before() {  
  40.         System.out.println("before method invoke");  
  41.     }  
  42.     private void after() {  
  43.         System.out.println("after method invoke");  
  44.     }  
  45.     private void exception() {  
  46.         System.out.println("method invoke exception");  
  47.     }  
  48.     private void beforeReturning() {  
  49.         System.out.println("before returning");  
  50.     }  
  51. }  
场景类：

[java] view plain copy
  1. package cglib.enhancer;  
  2.   
  3. public class EnhancerTest {  
  4.     public static void main(String[] args) throws Exception {  
  5.         Hello hello = new Hello();  
  6.         ProxyFactory cglibProxy = new ProxyFactory();  
  7.         Hello proxy = (Hello) cglibProxy.createProxy(hello);  
  8.         String result=proxy.sayHello(true);  
  9.         System.out.println(result);  
  10.     }  
  11. }  

运行结果：

[java] view plain copy
  1. before method invoke  
  2. hello everyone!  
  3. method invoke exception  
  4. before returning  
  5. null  
当然把true改为false之后结果变为：

[java] view plain copy
  1. before method invoke  
  2. hello everyone!  
  3. after method invoke  
  4. before returning  
  5. 123  

接下来我们了解下Enhancer可以使用的Callback接口。

[java] view plain copy
  1. package net.sf.cglib.proxy;  
  2.   
  3. /** 
  4.  * All callback interfaces used by {@link Enhancer} extend this interface. 
  5.  * @see MethodInterceptor 
  6.  * @see NoOp 
  7.  * @see LazyLoader 
  8.  * @see Dispatcher 
  9.  * @see InvocationHandler 
  10.  * @see FixedValue 
  11.  */  
  12. public interface Callback  
  13. {  
  14. }  
Callback接口是一个空接口，没有任何契约方法，它只表示这是一个回调。接下来看它的子接口：

CGLib之Enhancer

接口都很简单。

先来一个基础POJO类：

[java] view plain copy
  1. public class SampleClass {  
  2.   public String test(String input) {  
  3.     return "Hello world!";  
  4.   }  
  5. }  

FixedValue 

使用FixedValue可以很容易的替换掉方法的返回值。

[java] view plain copy
  1. @Test  
  2. public void testFixedValue() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new FixedValue() {  
  6.     @Override  
  7.     public Object loadObject() throws Exception {  
  8.       return "Hello cglib!";  
  9.     }  
  10.   });  
  11.   SampleClass proxy = (SampleClass) enhancer.create();  
  12.   assertEquals("Hello cglib!", proxy.test(null));  
  13. }  
在上面的例子中，enhancer返回一个SampleClass的子类实例，当所有的方法被调用的时候返回匿名FixedValue回调返回的值。当然了如果方法返回的不是String类型例如，hashcode()将抛出一个类型转换异常，因为"Hello cglib!"不能转换为java.lang.Number类型。

InvocationHandler

[java] view plain copy
  1. @Test  
  2. public void testInvocationHandler() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new InvocationHandler() {  
  6.     @Override  
  7.     public Object invoke(Object proxy, Method method, Object[] args)  
  8.         throws Throwable {  
  9.       if(method.getDeclaringClass() != Object.class && method.getReturnType() == String.class) {  
  10.         return "Hello cglib!";  
  11.       } else {  
  12.         throw new RuntimeException("Do not know what to do.");  
  13.       }  
  14.     }  
  15.   });  
  16.   SampleClass proxy = (SampleClass) enhancer.create();  
  17.   assertEquals("Hello cglib!", proxy.test(null));  
  18.   assertNotEquals("Hello cglib!", proxy.toString());  
  19. }  
这个回调允许你去回答调用的方法。但是你要小心使用这个回调，因为这个代理对象的所有的方法调用都会使用InvocationHandler#invoke方法，这样的话可能导致死循环。为了避免这种情况你可以使用MethodInterceptor。

MethodInterceptor

[java] view plain copy
  1. @Test  
  2. public void testMethodInterceptor() throws Exception {  
  3.   Enhancer enhancer = new Enhancer();  
  4.   enhancer.setSuperclass(SampleClass.class);  
  5.   enhancer.setCallback(new MethodInterceptor() {  
  6.     @Override  
  7.     public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy)  
  8.         throws Throwable {  
  9.       if(method.getDeclaringClass() != Object.class && method.getReturnType() == String.class) {  
  10.         return "Hello cglib!";  
  11.       } else {  
  12.         return proxy.invokeSuper(obj, args);  
  13.       }  
  14.     }  
  15.   });  
  16.   SampleClass proxy = (SampleClass) enhancer.create();  
  17.   assertEquals("Hello cglib!", proxy.test(null));  
  18.   assertNotEquals("Hello cglib!", proxy.toString());  
  19.   proxy.hashCode(); // Does not throw an exception or result in an endless loop.  
  20. }  

唯一需要注意的就是proxy.invokeSuper和proxy.invoke的区别。invokeSuper是退出当前interceptor的处理，进入下一个callback处理，invoke则会继续回调该方法，如果传递给invoke的obj参数出错容易造成递归调用。

LazyLoader

尽管LazyLoader唯一的方法和FixedValue的一模一样，但是他们还是有一个根本上的区别。
相比于Dispatcher,lazyLoader在第一次获取了loadObject后，会进行缓存，后续的请求调用都会直接调用该缓存的属性。
这个是有道理的,如果你的对象是昂贵的在其创建，不知道对象什么时候被使用。请注意,一些增强的类的构造函数必须被代理对象和延迟加载的对象。因此,确保还有另一个廉价的(可能保护)构造函数获得或使用一个接口类型的代理。你可以选择提供参数的调用Enhancer#create(Object...)。

Dispatcher

Dispatcher和LazyLoader很想但是每次调用方法的时候不会存储加载的对象。这允许改变一个类的实现但不改变原来的对象。
再者说一些构造函数必须被代理和生成的对象调用。

ProxyRefDispatcher

这个类携带了代理对象的引用，通过它的签名调用。例如这允许委托方法调用代理的另一个方法。要清楚这很容易造成死循环，并且总是导致死循环如果同样的方法被ProxyRefDispatcher#loadObject(Object)调用。

NoOp

像名字所说的一样，这个类不做任何操作。相反地，它委托每个方法的调用给被增强类的方法实现。

相关推荐