【Java高级】深度解析Java动态代理（实战MyBatis手动代理）

本文为原创文章，转载请注明出处
查看[Java]系列内容请点击：https://www.jianshu.com/nb/45938443

我们知道，代理模式一般分为静态代理和动态代理模式，静态代理模式这里还是简单提一下：
我们在以前进行MVC编程的时候，一个好的编程习惯是：从数据库里面查询需要有三个组件：

  • 一个接口A：A是用来定义对数据进行操作的接口
  • 一个数据实现Class AImpl implements A：即MVC中的Model层，用来实际对数据库进行操作和数据处理
  • 一个代理Class AProxy implements A：用来代理AImpl，从而为Model提供数据源、提供更多处理等操作

在真正使用的时候一般使用AProxy来进行数据操作，这样就构成了整个MVC的Model层。

上面的方法是静态代理。

动态代理

动态代理与静态代理不同，想想看，静态代理是自己写的代理类，在程序加载之前代理类就存在了，而动态代理是在运行时动态的生成的一种代理类，它可以为代理的类提供更多的额外服务。

一个简单的例子，拿大家最经常使用的MyBatis来说，在MyBatis里面获取一个Mapper是怎么获取的呢？我们来手动获取一个SqlSession和Mapper：

public static <T> T getMapper(Class<T> var) {
  String resource = "mybatis-config.xml";
  InputStream inputStream = Resources.getResourceAsStream(resource);
  SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
  org.apache.ibatis.session.SqlSession session = sqlSessionFactory.openSession(true); // 自动提交，但不会自动关闭
  return session.getMapper(var);
}

可以看到，这里获取SqlSession只会自动提交SQL，不会自动关闭SqlSession，这里我们仅仅为了执行一次SQL，所以只获取了Mapper而没有获取到SqlSession，所以我们在使用的时候也无法进行关闭SqlSession

MyBatis是如何获取Mapper的？

我们看这一句：

return session.getMapper(var);

这里就是调用了Mybatis的MapperProxy通过动态代理的方式获取的：

package org.apache.ibatis.binding;

public class MapperProxy<T> implements InvocationHandler, Serializable {

  private static final long serialVersionUID = -6424540398559729838L;
  private final SqlSession sqlSession;
  private final Class<T> mapperInterface;
  private final Map<Method, MapperMethod> methodCache;

  public MapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethod> methodCache) {
    this.sqlSession = sqlSession;
    this.mapperInterface = mapperInterface;
    this.methodCache = methodCache;
  }

  /**
   * 这里就是主要的代理方法啦
   **/
  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    try {
      if (Object.class.equals(method.getDeclaringClass())) {
        return method.invoke(this, args);
      } else if (isDefaultMethod(method)) {
        return invokeDefaultMethod(proxy, method, args);
      }
    } catch (Throwable t) {
      throw ExceptionUtil.unwrapThrowable(t);
    }
    final MapperMethod mapperMethod = cachedMapperMethod(method);
    return mapperMethod.execute(sqlSession, args);
  }

// more methods...
}

可以看到，MapperProxy类继承了 InvocationHandler接口，并实现了invoke方法，并在invoke方法中通过反射的方式对需要调用的方法进行了进一步的处理。

动态代理要做什么？

动态代理实际上就是要在一个对象obj1的基础上重新生成一个对象obj2，obj2与obj1具有相同的类型，不同的是，obj2可以在invoke方法中对obj1的方法进行一些其他的操作。

动态代理主要依赖Proxy类和InvocationHandler接口，下面的程序演示了将Apple对象代理到另外一个对象，并在调用Apple对象的方法之前和之后进行一些打印操作。

/**
 * 通用接口
 * */
public interface Fruit {

    String getName();

    String getColor();
}

/**
 * 通用接口
 * */
public class Apple implements Fruit {
    @Override
    public String getName() {
        return "apple";
    }

    @Override
    public String getColor() {
        return "red";
    }
}

/**
 * 代理类
 * */
public class FruitProxy implements InvocationHandler {
    private Object obj1; // obj1

    public FruitProxy(Object o) {
        this.obj1 = o;
    }

    /**
     * @param proxy  被代理后的对象，而不是代理之前的对象，这里就是指的obj2对象
     * @param method 被代理obj1的方法
     * @param args   被代理的方法的参数
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("--- before ---");

        Object result = method.invoke(this.obj1, args); // 调用obj1对象的方法

        System.out.println("method:" + method.getName() + "   result:" + result); // 打印被调用的方法信息
        System.out.println("--- after ---");

        return result;
    }
}

/**
 * 测试程序
 * */
public class Main {
    public static void main(String[] args) {

        Fruit obj1 = new Apple();

        Fruit obj2 = (Fruit) Proxy.newProxyInstance(FruitProxy.class.getClassLoader(), new Class[]{Fruit.class}, new FruitProxy(obj1));

        System.out.println(obj2.getName());
        System.out.println(obj2.getColor());
    }
}

输出结果为：

--- before ---
method:getName   result:apple
--- after ---
apple
--- before ---
method:getColor   result:red
--- after ---
red

可以看到，在进行方法调用的时候，已经在前后进行了更多的处理。由于invoke方法的参数在注释中介绍了，下面着重介绍Proxy的方法：

Proxy.newProxyInstance方法：

public static Object newProxyInstance(ClassLoader loader,
                                          Class<?>[] interfaces,
                                          InvocationHandler h)
  • 该方法第一个参数是类加载器，无论使用当前类的类加载器还是被代理的类的类加载器，一般使用的都是应用类加载器，效果都一样，这里不再过多介绍。
  • 第二个参数是被代理的接口列表，是一个Class数组，是被代理的对象锁继承的接口列表，注意必须是接口才行！！！不是接口必须使用CGLIB代理（这里不介绍），Proxy会通过反射机制来获取其中的方法，并进行代理。这里也是使用了运行时多态。
  • 第三个参数是代理处理器的对象，也就是继承了InvocationHandler接口的对象，Proxy会调用其中的invoke方法。

动态代理MapperProxy

好了，有了上述的基础后，我们基于Mybatis来实现一个可自动关闭的Mapper获取方法，想要实现的效果是：每次都重新获取一个SqlSession，执行完一条SQL后自动关闭：

首先来代理Mybatis的MapperProxy

public class MapperProxy<T> extends org.apache.ibatis.binding.MapperProxy<T> {
    private static final Logger log = Logger.getLogger(MapperProxy.class);

    private SqlSession sqlSession;
    private Class<T> mapperInterface;

    private static Map<Class, Map<String, Object>> methodsMap = new HashMap<>(); // 记录需要close的方法的名称

    public MapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethod> methodCache) {
        super(sqlSession, mapperInterface, methodCache);
        this.sqlSession = sqlSession;
        this.mapperInterface = mapperInterface;

        // 设置需要关闭的接口名称
        if (!methodsMap.containsKey(mapperInterface)) {
            Map<String, Object> methodNames = new HashMap<>();
            Arrays.stream(mapperInterface.getDeclaredMethods()).forEach(m -> methodNames.put(m.getName(), null));
            methodsMap.put(mapperInterface, methodNames);
        }
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        Object obj = null;
        try {
            obj = super.invoke(proxy, method, args);
        } catch (Throwable t) {
            throw t;
        } finally {
            if (methodsMap.containsKey(mapperInterface) && methodsMap.get(mapperInterface).containsKey(method.getName())) {
                // log.info("now close sqlsession");
                sqlSession.close();
            }
        }
        return obj;
    }
}

我们自己写的MapperProxy代理了Mybatis的MapperProxy，在每次调用Mapper的方法完成后会在finally里面关闭SqlSession

使用方法：

/**
 * 一次性执行的Mapper
 */
public static <T> T getMapper(Class<T> var1) {
    if (sqlSessionFactory == null) {
        initSession();
    }
    org.apache.ibatis.session.SqlSession session = sqlSessionFactory.openSession(true);

    // 利用动态代理添加自动关闭 SqlSession
    MapperProxy<T> mapperProxy = new MapperProxy<T>(session, var1, new HashMap<>());
    return (T) Proxy.newProxyInstance(var1.getClassLoader(), new Class[]{var1}, mapperProxy);
}

好了，这样，在每次执行Mapper的方法的时候就会自动关闭了。
实际上这里是有两次代理，一次是Mybatis自己的MapperProxy的代理，生成了Mapper，另外一次是我们自己的代理，将Mapper代理成可以自动关闭的Mapper

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 201,468评论 5赞 473
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 84,620评论 2赞 377
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 148,427评论 0赞 334
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,160评论 1赞 272
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,197评论 5赞 363
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,334评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 37,775评论 3赞 393
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,444评论 0赞 256
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,628评论 1赞 295
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,459评论 2赞 317
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,508评论 1赞 329
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,210评论 3赞 318
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,767评论 3赞 303
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,850评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,076评论 1赞 258
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 42,627评论 2赞 348
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,196评论 2赞 341