设计模型之代理模式

代理模式

代理模式是一种比较好理解的设计模式。使用代理对象来替代对真实对象的访问,这样就可以在不修改原目标对象的前提下,提供额外的功能操作,扩展目标对象的功能。代理模式的主要作用是 扩展目标对象的功能,比如在目标对象的某个方法执行前后可以增加一些自定义的操作。

例如 : 你找小红来帮你问话,小红就可看作你的代理对象,代理的行为(方法)就是问话。

代理模式有静态代理和动态代理两种实现方式

静态代理

静态代理中,我们对目标对象的每个方法的增强都是手动完成的,非常不灵活,接口一旦新增方法,目标对象和代理对象都要进行修改,且麻烦,需要对每个目标类都单独写一个代理类。

从JVM层面来说,静态代理在编译时就将接口、实现类、代理类这些变成了一个个实际的class文件

静态代理的实现步骤

  1. 定义一个接口及其实现类
  2. 创建一个代理类同样实现这个接口
  3. 将目标对象注入进代理类,然后在代理类的对应方法调用目标类中的对应方法。这样就可以通过代理类屏蔽对目标对象的访问,并且在目标方法执行前后做一些自己想做的事

demo

  1. 定义发送短信的接口
    public interface SmsService{
        String send(String message);
    }
  2. 实现发送短信接口的实现类
    public class SmsServiceImpl implements SmsService{
        public String send(String message){
            System.out.println("send message" + message);
            return message;
        }
    }
  3. 创建代理类并同样实现发送短信的接口
    public class SmsProxy implements SmsService{
        private final SmsService smsService;
    
        public SmsProxy(SmsService smsService){
            this.smsService = smsServicel
        }
    
        @Override
        public String send(String message){
            //使用方法之前,添加自己的操作
            System.out.println("before method send()");
            smsService.send(message);
            //调用方法之后,我们同样可以添加自己的操作
            System.out.println("after method send()");
            return null;
        }
    }
  4. 实际使用
    public class Main(){
        public static void main(String[]args){
            SmsService smsService = new SmsServiceImpl();
            SmsProxy smsProxy = new SmsProxy(smsService);
            smsProxy.send("java");
        }
    }
    输出结果
    before method send()
    send message java
    after method send()

动态代理

相比于静态代理,动态代理更灵魂,不需要针对目标类单独创建一个代理类,并且也不需要必须实现接口,可以直接代理实现类(CGLIB动态代理机制)

从jvm角度,动态代理是在运行时动态生成类字节码,并加载到jvm中

Spring AOP、RPC框架的实现都依赖了动态代理。

动态代理在框架中几乎是必用的一门技术。动态代理的实现方法:JDK动态代理,CGLIB动态代理等等

JDK动态代理机制

在Java动态代理机制中,InvocationHandler接口和Proxy类是核心,Proxy类使用频繁最高的方法是newProxyInstance(),这个方法主要是生成一个代理对象

public static Object newProxyInstance(ClassLoader loader,Class<?>[] interface,InvocationHandler h) throws IllegalArgumentException{
    ....
}

三个参数

  1. loader:类加载器,用于加载代理对象
  2. interface:被代理类是实现的一些接口
  3. h:实现了InvocationHandler接口的对象

要实现动态代理的话,还必须实现 InvocationHandler来自定义出来逻辑。当动态代理对象调用一个方法时候,这个方法的调用给就会被转发到实现InvocationHandler接口类的invoke方法来调用

public interface InvocationHandler{
    /**
    当你使用代理对象调用方法时实际会调用到这个方法
    */
    public Object invoke(Object proxy,Method method,Object[]args) throws Throwable;
}

invoke()方法有下面三个参数:

  1. proxy:动态代理生成的代理类
  2. method:与代理类对象调用的方法相对应
  3. args:当前method方法的参数

通过proxy类的newProxyInstance()创建的代理对象在调用方法的时候,实际上调用到实现InvocationHandler接口的类的invoke()方法。

JDK动态代理类使用步骤

  1. 定义一个接口及其实现类

  2. 自定义InvocationHandler并重写invoke()方法,在invoke()方法中调用原生方法(被代理类的方法)并自定义一些处理逻辑

  3. 通过Proxy.newProxyInstance(ClassLoader loader,Class<?> interface,InvocationHandler h)方法创建代理对象

  4. 定义发送短信的接口

    public interface SmsService{
        String send(String message);
    }
  5. 实现发送短信的接口

    public class SmsServiceImpl implements SmsService {
        public String send(String message) {
            System.out.println("send message:" + message);
            return message;
        }
    }
  6. 定义一个JDK动态代理类

    public class DebugInvocationHandler implements InvocationHandler{
        private final Object target;
        class DebugInvocationHandler(Object target){
            this.target = target;
        }
    
        public Object invoke(Object proxy,Method method,Object[]args) throws InvocationTargetException,IllegalAccessException{
             //调用方法之前,我们可以添加自己的操作
            System.out.println("before method " + method.getName());
            Object result = method.invoke(target,args);
            //调用方法之后,我们同样可以添加自己的操作
            System.out.println("after method " + method.getName());
            return result;
        }
    }

    invoke():当动态代理对象调用原生方法时,实际最终调用的是invoke()方法,然后invoke()方法代替去调用被代理对象的原生方法

  7. 获取代理对象的工厂类

    public class JdkProxyFactory{
        public static Object getProxy(Object target){
            return Proxy.newProxyInstance(
                target.getClass().getClassLoader(), //目标类的类加载
                target.getClass().getInterfaces(),  //代理需要实现的接口,可指定多个
                new DebugInvocationHandler(target)  //代理对象对应的自定义 InvocationHandler
            );
        }
    }

    getProxy():主要通过对Proxy.newProxyInstance()方法获取某个类的代理对象

实际使用

SmsService smsService = (SmsService) JdkProxyFactory.getProxy(new SmsServiceImpl());
smsService.send("java");

结果

before method send
send message:java
after method send

CGLIB动态代理机制

jdk动态代理有一个最致命的问题是其只能代理实现了接口的类,为了解决这个问题,可以使用CGLIB动态代理机制来避免

CGLIB(code generation library)是一个基于ASM的字节码生成库,允许在运行时对字节码进行修改和动态生成.CGLIB通过继承的方式实现代理。Spring中的AOP模块中,如果目标对象实现了接口,则默认采用JDK动态代理,否则就是采用CGLIB动态代理

在CGLIB动态代理机制中 MethodInterceptor接口和Enhancer类是核心

需要自定义MethodInterceptor并重写intercept方法,intercept用于拦截增强被代理类的方法

public interface MethodInterceptor extends Calllback{
    //拦截被代理类中的方法
    public Object intercept(Object obj,Method method,Object[]args,
                            MethodProxy proxy) throws Throwable;
}
  1. obj:被代理的对象(目标对象,需要增强的对象)
  2. method:被拦截的方法(需要增强的方法)
  3. args:方法入参
  4. methodProxy:用于调用原始方法

通过Enhancer类来动态获取被代理类,当代理类调用方法时,实际调用的是MethodInterceptor中的intercept方法。

CGLIB动态代理使用步骤

  1. 定义一个类
  2. 自定义MethodInterceptor并重写intercept方法,用于拦截增强被代理类的方法,和jdk动态代理中的invoke方法类似
  3. 通过Enhancer类的create()创建代理类

demo

不同于JDK动态代理,CGLIB使用需添加相关依赖。

<dependency>
  <groupId>cglib</groupId>
  <artifactId>cglib</artifactId>
  <version>3.3.0</version>
</dependency>
  1. 实现发送短信的类
    public class SmsService{
        public String send(String message){
            System.out.println("send message:"+message);
            return message;
        }
    }
  2. 自定义MethodInterceptor(方法拦截器)
    publlic class DebugMethodInterceptor implements MethodInterceptor{
        @Override
        public Object intercept(Object o, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
            //调用方法之前,我们可以添加自己的操作
            System.out.println("before method:"+method.getName());
            Object result = methodProxy.invokeSuper(o,args);
            //调用方法之后
            System.out.println("after method:"+method.getName());
            return result;
        }
    }
  3. 获取代理类
    public Class CglibProxyFactory{
        public static Object getProxy(Class<?> clazz){
            //创建动态增强类
            Enhancer enhancer = new Enhancer();
            //设置类加载器
            enhancer.setClassLoader(clazz.getClassLoader());
            //设置被代理类
            enhancer.setSuperClass(clazz);
            //设置方法拦截器
            enhancer.setCallback(new DebugMethodInterceptor());
            //创建代理类
            return enhancer.create();
        }
    }
  4. 实际使用
    SmsService smsService = (SmsService) CglibProxyFactory.getProxy(SmsService.getClass()));
    smsService.send("java");
    控制台打印出:
    befor methodsend
    send message:cglib
    after methodsend

JDK动态代理和CGLIB动态代理对比

  • JDK动态代理只能代理实现了接口的类或者直接代理接口,而CGLIB可以代理未实现任何接口的类。另外CGLIB动态代理是通过生成一个被代理类的子类来拦截被代理类的方法调用,因此不能代理声明为final类型的类和方法
  • JDK动态代理更优秀。

静态代理和动态代理对比

  • 灵活性: 动态代理更加灵活,不需要必须实现接口,可以直接代理实现类,并且可以不需要针对每个目标类都创建一个代理实现类。另外静态代理中,接口一旦新增方法,目标对象和代理对象都要修改,比较麻烦
  • JVM层面:静态代理是在编译时就将接口、实现类、代理类都变成了class文件。而动态代理在运行时动态生成字节码,并加载到JVM中