代理模式详解

176 阅读12分钟

1. 代理模式

代理模式是一种比较好理解的设计模式。简单来说就是 我们使用代理对象来代替对真实对象(real object)的访问,这样就可以在不修改原目标对象的前提下,提供额外的功能操作,扩展目标对象的功能。

代理模式一半包含三种角色:

抽象主题角色(Subject):抽象主题类的主要职责是声明真实主题与代理的共同接口方法,该类可以是接口也可以是抽象类。;

真实主题角色(RealSubject): 该类定义了代理所表示的真实对象,是负责执行系统真正的逻辑业务对象;

代理主题角色(Proxy):代理类;其内部持有RealSubject的引用,因此具备完全的对RealSubject的代理权。客户端调用代理对象的方法,同时也调用被代理对象的方法,但是会在代理对象前后增加一些代码处理。

在代码中,一般代理会被理解为代码增强,实际上就是扩展目标对象的功能,比如说在目标对象的某个方法执行前后你可以增加一些自定义的操作。

代理模式属于结构性模式:分为静态代理和动态代理。

2. 静态代理

静态代理中,我们对目标对象的每个方法的增强都是手动完成的,非常不灵活(比如接口一旦新增加方法,目标对象和代理对象都要进行修改且麻烦需要对每个目标类都单独写一个代理类)。 实际应用场景非常非常少,日常开发几乎看不到使用静态代理的场景。

上面我们是从实现和应用角度来说的静态代理,从 JVM 层面来说, 静态代理在编译时就将接口、实现类、代理类这些都变成了一个个实际的 class 文件。

静态代理实现步骤:

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

下面通过代码展示!

2.1 简单示例

  • 定义一个接口

    public interface ISubject {
    
        void request();
        
    }
    
  • 实现类

    public class RealSubject implements ISubject {
    
        public void request() {
            System.out.println("real service is called.");
        }
    
    }
    
  • 创建代理类

    public class Proxy implements ISubject {
    
        private ISubject subject;
    
        public Proxy(ISubject subject) {
            this.subject = subject;
        }
    
    
        //代码增强
        public void request() {
            before();
            subject.request();
            after();
        }
    
        public void before() {
            System.out.println("called before request().");
        }
    
        public void after() {
            System.out.println("called after request().");
        }
    }
    
  • 测试类

    public class Client {
    
        public static void main(String[] args) {
    
            Proxy proxy = new Proxy(new RealSubject());
            proxy.request();
    
        }
    
    }
    
  • 输出结果

    image-20211201122801220

2.2 业务应用

在分布式业务场景中,通常会对数据库进行分库分表,我们可以使用设置数据源路由来动态切换数据源。

注意:为了操作方便没有实际创建相关数据库;

  • 首先创建订单类

    public class Order {
    
        private Object orderInfo;
    
        //订单创建时间进行按年分库
        private Long createTime;
        
        private String id;
    
        public Object getOrderInfo() {
            return orderInfo;
        }
    
        public void setOrderInfo(Object orderInfo) {
            this.orderInfo = orderInfo;
        }
    
        public Long getCreateTime() {
            return createTime;
        }
    
        public void setCreateTime(Long createTime) {
            this.createTime = createTime;
        }
    
        public String getId() {
            return id;
        }
    
        public void setId(String id) {
            this.id = id;
        }
    }
    
  • 持久层类

    public class OrderDao {
    
        public int insert(Order order) {
            System.out.println("OrderDao创建Order成功!");
            return 1;
        }
        
    }
    
  • 业务类

    public interface IOrderService {
    
        int createOrder(Order order);
        
    }
    
    public class OrderService implements IOrderService {
        
        private OrderDao orderDao;
    
        public OrderService() {
            //如果使用Spring应该是自动注入的
            //我们为了使用方便,在构造方法中将orderDao直接初始化了
            orderDao = new OrderDao();
        }
    
        public int createOrder(Order order) {
            System.out.println("OrderService调用orderDao创建订单");
            return orderDao.insert(order);
        }
    }
    

接下来使用静态代理,主要完成的功能是:根据订单创建时间自动按年进行分库。

  • 创建数据源路由对象

    //动态切换数据源
    public class DynamicDataSourceEntity {
    
        //默认数据源
        public final static String DEFAULE_SOURCE = null;
    
        private final static ThreadLocal<String> local = new ThreadLocal<String>();
    
        private DynamicDataSourceEntity() {
        }
    
        //获取正在使用的数据源名字
        public static String get() {
            return local.get();
        }
    
        //还原当前切换的数据源
        public static void restore() {
            local.set(DEFAULE_SOURCE);
        }
    
        //设置已知名字的数据源
        public static void set(String source) {
            local.set(source);
        }
    
        //根据年份动态设置数据源
        public static void set(int year) {
            local.set("DB_" + year);
        }
    
        //清空数据源
        public static void clear() {
            local.remove();
        }
    
    }
    
  • 创建代理类

    public class OrderServiceStaticProxy implements IOrderService {
    
        private SimpleDateFormat yearFormat = new SimpleDateFormat("yyyy");
    
        private IOrderService orderService;
    
        public OrderServiceStaticProxy(IOrderService orderService) {
            this.orderService = orderService;
        }
    
        public int createOrder(Order order) {
            before();
    
            Long time = order.getCreateTime();
            Integer dbRouter = Integer.valueOf(yearFormat.format(new Date(time)));
            System.out.println("静态代理类自动分配到【DB_" + dbRouter + "】数据源处理数据");
            DynamicDataSourceEntity.set(dbRouter);
            this.orderService.createOrder(order);
            DynamicDataSourceEntity.restore();
    
            after();
            return 0;
        }
    
        private void before() {
            System.out.println("Proxy before");
        }
    
        private void after() {
            System.out.println("Proxy after");
        }
    }
    
  • 测试类

    public class DbRouteProxyTest {
        public static void main(String[] args) {
            try {
                Order order = new Order();
                SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
                Date date = sdf.parse("2021/12/01");
                order.setCreateTime(date.getTime());
                IOrderService orderService = new OrderServiceStaticProxy(new OrderService());
                orderService.createOrder(order);
            } catch (Exception e) {
                e.printStackTrace();
            }
    
        }
    
    }
    
  • 输出结果

    image-20211201124535485

结果符合我们的预期,根据日期将数据存储到对应的数据库中。

3. 动态代理

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

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

说到动态代理,Spring AOP、RPC 框架应该是两个不得不提的,它们的实现都依赖了动态代理。

动态代理在我们日常开发中使用的相对较少,但是在框架中的几乎是必用的一门技术。学会了动态代理之后,对于我们理解和学习各种框架的原理也非常有帮助。

就 Java 来说,动态代理的实现方式有很多种,比如 JDK 动态代理CGLIB 动态代理等等。

3.1 优化静态代理

我们可以通过修改静态代理中的数据源路由来了解动态代理。

创建动态代理类:

public class OrderServiceDynamicProxy implements InvocationHandler {

    private SimpleDateFormat yearFormat = new SimpleDateFormat("yyyy");

    Object proxyObj;

    public Object getInstance(Object proxyObj) {
        this.proxyObj = proxyObj;
        Class<?> clazz = proxyObj.getClass();
        return Proxy.newProxyInstance(clazz.getClassLoader(), clazz.getInterfaces(), this);
    }

    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        before(args[0]);
        Object object = method.invoke(proxyObj, args);
        after();
        return object;
    }

    private void after() {
        System.out.println("Proxy after method");
        //还原成默认的数据源
        DynamicDataSourceEntity.restore();
    }

    //target 应该是订单对象Order
    private void before(Object target) {
        try {
            //进行数据源的切换
            System.out.println("Proxy before method");

            //约定优于配置
            Long time = (Long) target.getClass().getMethod("getCreateTime").invoke(target);
            Integer dbRouter = Integer.valueOf(yearFormat.format(new Date(time)));
            System.out.println("静态代理类自动分配到【DB_" + dbRouter + "】数据源处理数据");
            DynamicDataSourceEntity.set(dbRouter);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

测试代码如下:

public class DbRouteProxyTest {
    public static void main(String[] args) {
        try {
            Order order = new Order();
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd");
            Date date = sdf.parse("2021/12/01");
            order.setCreateTime(date.getTime());
            IOrderService orderService = (IOrderService) new OrderServiceDynamicProxy().getInstance(new OrderService());
            orderService.createOrder(order);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

}

我们可以看到依然实现了相关效果。但是,使用动态代理之后,我们不仅可以实现Order表的数据源动态切换,还可以实现其他表的路由。

3.2 JDK动态代理机制

3.2.1 介绍

在 Java 动态代理机制中 InvocationHandler 接口和 Proxy 类是核心。

Proxy 类中使用频率最高的方法是:newProxyInstance() ,这个方法主要用来生成一个代理对象。

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

这个方法一共有 3 个参数:

  1. loader :类加载器,用于加载代理对象。
  2. interfaces : 被代理类实现的一些接口;
  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()方法。 你可以在 invoke() 方法中自定义处理逻辑,比如在方法执行前后做什么事情。

3.2.2 使用步骤

  1. 定义一个接口及其实现类;
  2. 自定义 InvocationHandler 并重写invoke方法,在 invoke 方法中我们会调用原生方法(被代理类的方法)并自定义一些处理逻辑;
  3. 通过 Proxy.newProxyInstance(ClassLoader loader,Class<?>[] interfaces,InvocationHandler h) 方法创建代理对象;

3.2.3 简单示例

  • 定义接口

    public interface SmsService {
        String send(String message);
    }
    
  • 实现类

    public class SmsServiceImpl implements SmsService {
        public String send(String message) {
            System.out.println("send message:" + message);
            return message;
        }
    }
    
  • 代理类

    public class JDkInvocationHandler implements InvocationHandler {
    
        /**
         * 代理类中的真实对象
         */
        private final Object target;
    
        public JDkInvocationHandler(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() 方法代替我们去调用了被代理对象的原生方法。

  • 获取代理对象的工厂类

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

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

  • 测试

    public class JDKTest {
    
        public static void main(String[] args) {
            SmsService smsService = (SmsService) JdkProxyFactory.getProxy(new SmsServiceImpl());
            smsService.send("java");
        }
    }
    

    image-20211201131031396

3.3 CGLIB 动态代理机制

3.3.1 介绍

JDK 动态代理有一个最致命的问题是其只能代理实现了接口的类。

为了解决这个问题,我们可以用 CGLIB 动态代理机制来避免。

CGLIB (opens new window)(Code Generation Library)是一个基于ASM (opens new window)的字节码生成库,它允许我们在运行时对字节码进行修改和动态生成。CGLIB 通过继承方式实现代理。很多知名的开源框架都使用到了CGLIB (opens new window), 例如 Spring 中的 AOP 模块中:如果目标对象实现了接口,则默认采用 JDK 动态代理,否则采用 CGLIB 动态代理。

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

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

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

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

3.3.2 使用步骤

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

3.3.3 简单示例

  • 引入依赖

    不同于 JDK 动态代理不需要额外的依赖。CGLIB (opens new window)(Code Generation Library) 实际是属于一个开源项目,如果你要使用它的话,需要手动添加相关依赖。

    <dependency>
      <groupId>cglib</groupId>
      <artifactId>cglib</artifactId>
      <version>3.3.0</version>
    </dependency>
    
  • 定义类

    public class SmsService {
        public String send(String message) {
            System.out.println("send message:" + message);
            return message;
        }
    }
    
  • 自定义MethodInterceptor

    public class CGMethodInterceptor implements MethodInterceptor {
    
        /**
         * @param o           代理对象(增强的对象)
         * @param method      被拦截的方法(需要增强的方法)
         * @param args        方法入参
         * @param methodProxy 用于调用原始方法
         */
        @Override
        public Object intercept(Object o, Method method, Object[] args, MethodProxy methodProxy) throws Throwable {
            //调用方法之前,我们可以添加自己的操作
            System.out.println("before method " + method.getName());
            Object object = methodProxy.invokeSuper(o, args);
            //调用方法之后,我们同样可以添加自己的操作
            System.out.println("after method " + method.getName());
            return object;
        }
    }
    
  • 获取代理类

    public class CglibProxyFactory {
    
        public static Object getProxy(Class<?> clazz) {
            // 创建动态代理增强类
            Enhancer enhancer = new Enhancer();
            // 设置类加载器
            enhancer.setClassLoader(clazz.getClassLoader());
            // 设置被代理类
            enhancer.setSuperclass(clazz);
            // 设置方法拦截器
            enhancer.setCallback(new CGMethodInterceptor());
            // 创建代理类
            return enhancer.create();
        }
    }
    
  • 测试

    public class CGTest {
    
        public static void main(String[] args) {
            SmsService aliSmsService = (SmsService) CglibProxyFactory.getProxy(SmsService.class);
            aliSmsService.send("java");
        }
    }
    

    image-20211201131815626

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

  • JDK 动态代理只能代理实现了接口的类或者直接代理接口,而 CGLIB 可以代理未实现任何接口的类。 另外, CGLIB 动态代理是通过生成一个被代理类的子类来拦截被代理类的方法调用,因此不能代理声明为 final 类型的类和方法。
  • JDK动态代理和CGLIB代理都在运行期间生成字节码,JDK动态代理直接写Class字节码,CGLIB代理使用ASM框架写Class字节码,随意相对复杂,生成代理类的效率比较低。
  • JDK调用代理方法是通过反射机制,CGLIB是通过FastClass机制直接调用方法,所有执行效率较高。
  • 就二者的效率来说,大部分情况都是 JDK 动态代理更优秀,随着 JDK 版本的升级,这个优势更加明显。

4. 静态代理和动态代理的对比

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