Mybatis源码解析，一步一步从浅入深（六）：映射代理类的获取

在文章：Mybatis源码解析，一步一步从浅入深（二）：按步骤解析源码中我们提到了两个问题：

　　1，为什么在以前的代码流程中从来没有addMapper，而这里却有getMapper？

　　2，UserDao明明是我们定义的一个接口类，根本没有定义实现类，那这个userMapper是什么？是mybatis自动为我们生成的实现类吗？

　　为了更好的解释着两个问题，我们需要重新认识Configuration这个类。

　　但是在这之前，你需要了解一个概念（设计模式）：JAVA设计模式-动态代理(Proxy)示例及说明。否则你可能对接下来的流程一头雾水。

一，再次认识Configuration

public class Configuration {
  //映射注册表
  protected MapperRegistry mapperRegistry = new MapperRegistry(this);

  // 获取映射注册表
  public MapperRegistry getMapperRegistry() {
    return mapperRegistry;
  }

  //添加到映射注册表
  public void addMappers(String packageName, Class<?> superType) {
    mapperRegistry.addMappers(packageName, superType);
  }
  //添加到映射注册表
  public void addMappers(String packageName) {
    mapperRegistry.addMappers(packageName);
  }
  //添加到映射注册表
  public <T> void addMapper(Class<T> type) {
    mapperRegistry.addMapper(type);
  }
  //从映射注册表中获取
  public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
    return mapperRegistry.getMapper(type, sqlSession);
  }
  //判断映射注册表中是否存在
  public boolean hasMapper(Class<?> type) {
    return mapperRegistry.hasMapper(type);
  }
}

　　MapperRegistry源码：

public class MapperRegistry {

  private Configuration config;
  //映射缓存 键：类对象，值：映射代理工厂
  private final Map<Class<?>, MapperProxyFactory<?>> knownMappers = new HashMap<Class<?>, MapperProxyFactory<?>>();

  public MapperRegistry(Configuration config) {
    this.config = config;
  }
  //从映射注册表中获取
  @SuppressWarnings("unchecked")
  public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
    final MapperProxyFactory<T> mapperProxyFactory = (MapperProxyFactory<T>) knownMappers.get(type);
    if (mapperProxyFactory == null)
      throw new BindingException("Type " + type + " is not known to the MapperRegistry.");
    try {
      return mapperProxyFactory.newInstance(sqlSession);
    } catch (Exception e) {
      throw new BindingException("Error getting mapper instance. Cause: " + e, e);
    }
  }
  //判断映射注册表中是否存在
  public <T> boolean hasMapper(Class<T> type) {
    return knownMappers.containsKey(type);
  }
  //添加到映射注册表
  public <T> void addMapper(Class<T> type) {
    if (type.isInterface()) {
      if (hasMapper(type)) {
        throw new BindingException("Type " + type + " is already known to the MapperRegistry.");
      }
      boolean loadCompleted = false;
      try {
        knownMappers.put(type, new MapperProxyFactory<T>(type));
        // It's important that the type is added before the parser is run
        // otherwise the binding may automatically be attempted by the
        // mapper parser. If the type is already known, it won't try.
        // 处理接口类(例如UserDao)中的注解
        MapperAnnotationBuilder parser = new MapperAnnotationBuilder(config, type);
        parser.parse();
        loadCompleted = true;
      } finally {
        if (!loadCompleted) {
          knownMappers.remove(type);
        }
      }
    }
  }

  /**
   * @since 3.2.2
   */
  //获取缓存中所有的Key,并且是不可修改的
  public Collection<Class<?>> getMappers() {
    return Collections.unmodifiableCollection(knownMappers.keySet());
  }

  /**
   * @since 3.2.2
   */
  //添加到映射注册表
  public void addMappers(String packageName, Class<?> superType) {
    ResolverUtil<Class<?>> resolverUtil = new ResolverUtil<Class<?>>();
    resolverUtil.find(new ResolverUtil.IsA(superType), packageName);
    Set<Class<? extends Class<?>>> mapperSet = resolverUtil.getClasses();
    for (Class<?> mapperClass : mapperSet) {
      addMapper(mapperClass);
    }
  }

  /**
   * @since 3.2.2
   */
  //添加到映射注册表
  public void addMappers(String packageName) {
    addMappers(packageName, Object.class);
  }
}

　　在方法getMappers中用到了Collections.unmodifiableCollection(knownMappers.keySet());，如果你不了解，可以查阅：Collections.unmodifiableMap,Collections.unmodifiableList,Collections.unmodifiableSet作用及源码解析

　　在了解了这两个类之后，就来解决第一个问题：1，为什么在以前的代码流程中从来没有addMapper，而这里却有getMapper？

二，addMapper和getMapper

　　1,关于addMapper，在文章Mybatis源码解析，一步一步从浅入深（五）：mapper节点的解析中，不知道大家有没有意识到，我少了一个部分没有解读：

　　

　　对了，就是第四部分的：绑定已经解析的命名空间

　　代码：bindMapperForNamespace();

　　是的，addMapper就是在这个方法中用到的。但是前提是，你需要了解java的动态代理。来看看源码：

private void bindMapperForNamespace() {
    //获取当前命名空间（String：com.zcz.learnmybatis.dao.UserDao）
    String namespace = builderAssistant.getCurrentNamespace();
    if (namespace != null) {
      Class<?> boundType = null;
      try {
        // 使用类加载器加载，加载类，获取类对象
        boundType = Resources.classForName(namespace);
      } catch (ClassNotFoundException e) {
        //ignore, bound type is not required
      }
      if (boundType != null) {
        //判断映射注册表中是否存在
        if (!configuration.hasMapper(boundType)) {
          // Spring may not know the real resource name so we set a flag
          // to prevent loading again this resource from the mapper interface
          // look at MapperAnnotationBuilder#loadXmlResource
          // 添加到已经解析的缓存
          configuration.addLoadedResource("namespace:" + namespace);
          // 添加到映射这测表
          configuration.addMapper(boundType);
        }
      }
    }
  }

　　看到了吧，就在最后一行代码。但是这里并不是简单的保存了一个类对象，而是在MapperRegistry中进行了进一步的处理：

 //添加到映射注册表
   public <T> void addMapper(Class<T> type) {
     if (type.isInterface()) {
       if (hasMapper(type)) {
         throw new BindingException("Type " + type + " is already known to the MapperRegistry.");
       }
       boolean loadCompleted = false;
       try {
 　　　　　// 在这里，保存的是new MapperProxyFactory实例对象。
         knownMappers.put(type, new MapperProxyFactory<T>(type));
         // It's important that the type is added before the parser is run
         // otherwise the binding may automatically be attempted by the
         // mapper parser. If the type is already known, it won't try.
         // 处理接口类(例如UserDao)中的注解
         MapperAnnotationBuilder parser = new MapperAnnotationBuilder(config, type);
         parser.parse();
         loadCompleted = true;
       } finally {
         if (!loadCompleted) {
           knownMappers.remove(type);
         }
       }
     }
   }

　　在第10行，保存的是映射代理工厂（MapperProxyFactory）的实例对象。到这里addMapper就解释清楚了。接下来看看getMapper方法。

　　2，getMapper

　　　　调用的地方：在文章Mybatis源码解析，一步一步从浅入深（二）：按步骤解析源码第三步中。

　　　　代码：configuration.<T>getMapper(type, this);

　　　　　　type：是UserDao.class

　　　　　　this：SqlSession的实例化对象

　　　　从第一部分Configuration中可以发现，Configuration又调用了MapperRegistry的getMapper方法

 //从映射注册表中获取
   @SuppressWarnings("unchecked")
   public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
     final MapperProxyFactory<T> mapperProxyFactory = (MapperProxyFactory<T>) knownMappers.get(type);
     if (mapperProxyFactory == null)
       throw new BindingException("Type " + type + " is not known to the MapperRegistry.");
     try {
       return mapperProxyFactory.newInstance(sqlSession);
     } catch (Exception e) {
       throw new BindingException("Error getting mapper instance. Cause: " + e, e);
     }
   }

　　从代码的第4行可以清晰的看到，或者根据类对象从缓存Map中，获取到了addMapper中保存的MapperProxyFactory对象实例。但是并没有将这个对象实例直接返回，而是通过调用的MapperProxyFactory的newInstance方法返回的一个UserDao实现类。接下来我们i就解释一下第二个问题：2，UserDao明明是我们定义的一个接口类，根本没有定义实现类，那这个userMapper是什么？是mybatis自动为我们生成的实现类吗？

三，映射代理类的实现

　　看过JAVA设计模式-动态代理(Proxy)示例及说明这篇文章的同学应该知道这个问题的答案了，userMapper是一个代理类对象实例。是通过映射代理工厂（MapperProxyFactory）的方法newInstance方法获取的。

　　但是在这里mybatis有一个很巧妙的构思，使得这个的动态代理的使用方法和文章JAVA设计模式-动态代理(Proxy)示例及说明中的使用方法有些许不同。

　　不妨在你的脑海中回顾一下JAVA设计模式-动态代理(Proxy)示例及说明中实现动态代理的关键因素：

　　　　1，一个接口

　　　　2，实现了接口的类

　　　　3，一个调用处理类（构造方法中要传入2中的类的实例对象）

　　　　4，调用Proxy.newProxyInstance方法获取代理类实例化对象

　　带着这个印象，我们来分析一下mybatis是怎么实现动态代理的。既然userMapper是通过映射代理工厂（MapperProxyFactory）生产出来的，那么我们就看看它的源码：　

 //映射代理工厂
 public class MapperProxyFactory<T> {
   // 接口类对象（UserDao.class）
   private final Class<T> mapperInterface;
   // 对象中的方法缓存
   private Map<Method, MapperMethod> methodCache = new ConcurrentHashMap<Method, MapperMethod>();

   //构造器
   public MapperProxyFactory(Class<T> mapperInterface) {
     //为接口类对象赋值
     this.mapperInterface = mapperInterface;
   }

   public Class<T> getMapperInterface() {
     return mapperInterface;
   }

   public Map<Method, MapperMethod> getMethodCache() {
     return methodCache;
   }

   // 实例化映射代理类
   @SuppressWarnings("unchecked")
   protected T newInstance(MapperProxy<T> mapperProxy) {
     return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);
   }

   // 实例化映射代理类
   public T newInstance(SqlSession sqlSession) {
     final MapperProxy<T> mapperProxy = new MapperProxy<T>(sqlSession, mapperInterface, methodCache);
     return newInstance(mapperProxy);
   }

 }

　　我们调用的newInstance方法就是第29行的方法。然后这个方法又调用了24行的方法。我们来看25行的代码，是不是很熟悉？Proxy.newProxyInstance(类加载器，接口类对象数组，实现了InvocationHandler的对象实例 mapperProxy)，到这里映射代理类的实例化已经解释清楚了，也就是解决了第二个问题，接下来我们扩展一下：

　　现在我们还没有看到MapperProxy类的源码，但是我们可以大胆的猜测，类MapperProxy一定是实现了InvocationHandler接口，并且也一定实现了Invoke方法：

 // 映射代理
 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;
   }
   //代理类调用的时候执行的方法
   public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
     // 检查Method方法所在的类是否是Object
     if (Object.class.equals(method.getDeclaringClass())) {
       try {
         return method.invoke(this, args);
       } catch (Throwable t) {
         throw ExceptionUtil.unwrapThrowable(t);
       }
     }
     // 应用缓存
     final MapperMethod mapperMethod = cachedMapperMethod(method);
     //执行查询
     return mapperMethod.execute(sqlSession, args);
   }

   //应用缓存
   private MapperMethod cachedMapperMethod(Method method) {
     MapperMethod mapperMethod = methodCache.get(method);
     if (mapperMethod == null) {
       mapperMethod = new MapperMethod(mapperInterface, method, sqlSession.getConfiguration());
       methodCache.put(method, mapperMethod);
     }
     return mapperMethod;
   }
 }

　　那么，mybatis使用动态代理的方式跟文章JAVA设计模式-动态代理(Proxy)示例及说明使用动态代理的方式，有哪些不同呢？

　　1，一个接口（UserDao）

　　2，实现了接口的类(没有)

　　3，一个调用处理类（构造方法中要传入2中的类的实例对象）（2中没有，自然这里也不会传入对象）

　　4，调用Proxy.newProxyInstance方法获取代理类实例化对象（有的）

　　

　　如果你实实在在的明白了JAVA设计模式-动态代理(Proxy)示例及说明这篇文中所叙述的内容，相信这篇文章不难理解。

　　好了，mybatis源码解析到这里，已经基本接近尾声了，继续探索吧：Mybatis源码解析，一步一步从浅入深（二）：按步骤解析源码

---

原创不易，转载请声明出处：https://www.cnblogs.com/zhangchengzi/p/9706395.html