MyBatis源码之中心配置文件

首先分析源码,我们需要知道其使用在入手分析其源码。

MyBatis的使用:

String resource = "mybatis-config.xml";
InputStream inputStream = Resources.getResourceAsStream(resource);
SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
SqlSession session = sqlSessionFactory.openSession();
BlogMapper mapper = session.getMapper(BlogMapper.class);
Blog blog = mapper.selectBlogById(1);
session.close();

从原生的MyBatis使用上看,其使用分为了六部分,分别是:
  1. 文件读取成流
  2. 获取一个SqlSessionFactory工厂类,在这一步骤里,完成了解析配置文件,包括mybatis-config.xml和Mapper映射器文件。
  3. 通过SqlSessionFactory获得一个SqlSession
  4. 根据SqlSeesion的实例获取一个Mapper对象
  5. 使用mapper对象调用接口
  6. 关闭session

下面就按这些步骤去分析源码,分析这些步骤中隐藏的操作。
读取文件成流就不说了,直接进入第二步获取工厂类时解析配置文件分析。
首先,在分析其源码之前我们应该关心的内容:其解析过程做了什么,产生了哪些对象,这些对象时放到哪里的。

SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);

首先

我们需要了解SqlSessionFactory 是由SqlSessionFactoryBuilder()的build()方法得到,所以直接进入buil()方法中

  public SqlSessionFactory build(InputStream inputStream, String environment, Properties properties) {
    try {
      // 用于解析 mybatis-config.xml,同时创建了 Configuration 对象 
      XMLConfigBuilder parser = new XMLConfigBuilder(inputStream, environment, properties);
      // 解析XML,最终返回一个 DefaultSqlSessionFactory 
      return build(parser.parse());
    } catch (Exception e) {
      throw ExceptionFactory.wrapException("Error building SqlSession.", e);
    } finally {
      ErrorContext.instance().reset();
      try {
        inputStream.close();
      } catch (IOException e) {
        // Intentionally ignore. Prefer previous error.
      }
    }
  }

从方法里可以看出,这里创建了一个XMLConfigBuilder对象,这个对象时BaseBuilder的一个子类,是用来专门解析全局配置文件的,可以看到BaseBuilder的类图


image.png
  • XMLMapperBuilder:解析Mapper映射器文件的
  • XMLStatementBuilder:解析增删改查标签的
  • XMLScriptBuilder:解析动态SQL
  • SqlSourceBuilder:将SQL语句中的#{xxx}替换成占位符 ?和将占位符 ?对应的属性信息一起构建成到ParameterMapping中,以便在后续ParameterHandler真正地参数赋值
  • ParameterMappingTokenHandler 是SqlSourceBuilder的静态内部类,是为参数构建 ParameterMapping的。
    这些后面分析源码的时候都会看到

好了,言归正传

我们先来看看创建XMLConfigBuilder对象时做了些什么。

  public XMLConfigBuilder(InputStream inputStream, String environment, Properties props) {
    this(new XPathParser(inputStream, true, props, new XMLMapperEntityResolver()), environment, props);
  }

  private XMLConfigBuilder(XPathParser parser, String environment, Properties props) {
    super(new Configuration());
    ErrorContext.instance().resource("SQL Mapper Configuration");
    this.configuration.setVariables(props);
    this.parsed = false;
    this.environment = environment;
    this.parser = parser;
  }

XPathParser 类中持有一个Document对象,使我们的XML文件,还有一个XPath对象,是定位信息的方法,是java提供的,XPathParser解析出的元素用一个XNode对象存储。
XMLConfigBuilder的构造方法里可以看出new Configuration()创建了我们重要的对象Configuration,我们都知道它是全局唯一用来存放MyBatis行为信息的。

我们来看看创建Configuration对象时都做了些什么。

public class Configuration {

  protected Environment environment;

  protected boolean safeRowBoundsEnabled;
  protected boolean safeResultHandlerEnabled = true;
  protected boolean mapUnderscoreToCamelCase;
  protected boolean aggressiveLazyLoading;
  protected boolean multipleResultSetsEnabled = true;
  protected boolean useGeneratedKeys;
  protected boolean useColumnLabel = true;
  protected boolean cacheEnabled = true;
  protected boolean callSettersOnNulls;
  protected boolean useActualParamName = true;
  protected boolean returnInstanceForEmptyRow;
...
protected Class configurationFactory;

  protected final MapperRegistry mapperRegistry = new MapperRegistry(this);
  protected final InterceptorChain interceptorChain = new InterceptorChain();
  protected final TypeHandlerRegistry typeHandlerRegistry = new TypeHandlerRegistry(this);
  protected final TypeAliasRegistry typeAliasRegistry = new TypeAliasRegistry();
  protected final LanguageDriverRegistry languageRegistry = new LanguageDriverRegistry();

  protected final Map mappedStatements = new StrictMap("Mapped Statements collection")
      .conflictMessageProducer((savedValue, targetValue) ->
          ". please check " + savedValue.getResource() + " and " + targetValue.getResource());
  protected final Map caches = new StrictMap<>("Caches collection");
  protected final Map resultMaps = new StrictMap<>("Result Maps collection");
  protected final Map parameterMaps = new StrictMap<>("Parameter Maps collection");
  protected final Map keyGenerators = new StrictMap<>("Key Generators collection");

  protected final Set loadedResources = new HashSet<>();
  protected final Map sqlFragments = new StrictMap<>("XML fragments parsed from previous mappers");

  protected final Collection incompleteStatements = new LinkedList<>();
  protected final Collection incompleteCacheRefs = new LinkedList<>();
  protected final Collection incompleteResultMaps = new LinkedList<>();
  protected final Collection incompleteMethods = new LinkedList<>();
...

其中一部分源码就不全部贴上了,自己可以去仔细看看,在这个configuration类里可以看到它有很多的属性,并且有些属性是直接赋值了,所以我们就明白了在配置文件中的一些属性为什么不需要我们去设置了,因为在其内部已经帮我们设置了默认值,如果我们自己设置了值得话,在后面的配置文件中将会解析出来再进行替换。
还有里面的各种容器跟一些注册中心可以网上查找更多的资料去了解,因为太多了就不一一解读了。

我们再看其构造函数,里面都是为我们常用的类注册别名的。

 public Configuration() {
    // 为类型注册别名
    typeAliasRegistry.registerAlias("JDBC", JdbcTransactionFactory.class);
    typeAliasRegistry.registerAlias("MANAGED", ManagedTransactionFactory.class);

    typeAliasRegistry.registerAlias("JNDI", JndiDataSourceFactory.class);
    typeAliasRegistry.registerAlias("POOLED", PooledDataSourceFactory.class);
    typeAliasRegistry.registerAlias("UNPOOLED", UnpooledDataSourceFactory.class);

    typeAliasRegistry.registerAlias("PERPETUAL", PerpetualCache.class);
    typeAliasRegistry.registerAlias("FIFO", FifoCache.class);
    typeAliasRegistry.registerAlias("LRU", LruCache.class);
    typeAliasRegistry.registerAlias("SOFT", SoftCache.class);
    typeAliasRegistry.registerAlias("WEAK", WeakCache.class);

    typeAliasRegistry.registerAlias("DB_VENDOR", VendorDatabaseIdProvider.class);

    typeAliasRegistry.registerAlias("XML", XMLLanguageDriver.class);
    typeAliasRegistry.registerAlias("RAW", RawLanguageDriver.class);
···

XMLConfigBuilder的创建会新建一个Configuration对象,我们接着往下看

 XMLConfigBuilder parser = new XMLConfigBuilder(inputStream, environment, properties);
      // 解析XML,最终返回一个 DefaultSqlSessionFactory >>
      return build(parser.parse());

在得到XMLConfigBuilder 对象后,可以看到调用了其内部的parse()方法

  public Configuration parse() {
    if (parsed) {
      throw new BuilderException("Each XMLConfigBuilder can only be used once.");
    }
    parsed = true;
    // XPathParser,dom 和 SAX 都有用到 >>
    parseConfiguration(parser.evalNode("/configuration"));
    return configuration;
  }

  private void parseConfiguration(XNode root) {
    try {
      //issue #117 read properties first
      // 对于全局配置文件各种标签的解析
      propertiesElement(root.evalNode("properties"));
      // 解析 settings 标签
      Properties settings = settingsAsProperties(root.evalNode("settings"));
      loadCustomVfs(settings);
      loadCustomLogImpl(settings);
      // 类型别名
      typeAliasesElement(root.evalNode("typeAliases"));
      // 插件
      pluginElement(root.evalNode("plugins"));
      // 用于创建对象
      objectFactoryElement(root.evalNode("objectFactory"));
      // 用于对对象进行加工
      objectWrapperFactoryElement(root.evalNode("objectWrapperFactory"));
      // 反射工具箱
      reflectorFactoryElement(root.evalNode("reflectorFactory"));
      // settings 子标签赋值,默认值就是在这里提供的 >>
      settingsElement(settings);
      // read it after objectFactory and objectWrapperFactory issue #631
      // 创建了数据源 >>
      environmentsElement(root.evalNode("environments"));
      databaseIdProviderElement(root.evalNode("databaseIdProvider"));
      typeHandlerElement(root.evalNode("typeHandlers"));
      // 解析引用的Mapper映射器
      mapperElement(root.evalNode("mappers"));
    } catch (Exception e) {
      throw new BuilderException("Error parsing SQL Mapper Configuration. Cause: " + e, e);
    }
  }

parse()方法中首先会parser.evalNode("/configuration")获取到XNode对象,XNode是MyBatis对Node类的封装,它会自动解析出元素的元素名,元素的属性,反之起就是解析XML文件的,重点不在此。

parseConfiguration()方法中是对中心配置文件的解析

见名知意,可以看出它是一步一步的解析器配置中心的标签,如果设置有就将其解析并放入configuration中。

Properties settings = settingsAsProperties(root.evalNode("settings"));
      loadCustomVfs(settings);
      loadCustomLogImpl(settings);
      // 类型别名
      typeAliasesElement(root.evalNode("typeAliases"));
      // 插件
      pluginElement(root.evalNode("plugins"));
      // 用于创建对象
      objectFactoryElement(root.evalNode("objectFactory"));
      // 用于对对象进行加工
      objectWrapperFactoryElement(root.evalNode("objectWrapperFactory"));
      // 反射工具箱
      reflectorFactoryElement(root.evalNode("reflectorFactory"));
  • 第一是解析标签,读取我们引入的外部配置文件,例如db.properties,这里有两种类型,一种是放在resorce目录下的,是相对路径,一种是写绝对路径(url),解析的最终结果就是我们配置信息放到 XPathparser的variables中和Configuration的variables中
private void propertiesElement(XNode context) throws Exception {
    if (context != null) {
      // 创建了一个 Properties 对象,后面可以用到
      Properties defaults = context.getChildrenAsProperties();
      String resource = context.getStringAttribute("resource");
      String url = context.getStringAttribute("url");
      if (resource != null && url != null) {
        throw new BuilderException("The properties element cannot specify both a URL and a resource based property file reference.  Please specify one or the other.");
      }
      if (resource != null) {
        defaults.putAll(Resources.getResourceAsProperties(resource));
      } else if (url != null) {
        defaults.putAll(Resources.getUrlAsProperties(url));
      }
      Properties vars = configuration.getVariables();
      if (vars != null) {
        defaults.putAll(vars);
      }
      //为什么放进parser里面
      parser.setVariables(defaults);
      configuration.setVariables(defaults);
    }
  }
  • 第二是解析标签解析成一个Properties对象,将放到后面的setting方法中处理。
  private Properties settingsAsProperties(XNode context) {
    if (context == null) {
      return new Properties();
    }
    Properties props = context.getChildrenAsProperties();
    // Check that all settings are known to the configuration class
    MetaClass metaConfig = MetaClass.forClass(Configuration.class, localReflectorFactory);
    for (Object key : props.keySet()) {
      if (!metaConfig.hasSetter(String.valueOf(key))) {
        throw new BuilderException("The setting " + key + " is not known.  Make sure you spelled it correctly (case sensitive).");
      }
    }
    return props;
  }
  • 第三步loadCustomVfs(settings) 是获取Vitual File System的自定义实现类,比如要读取本地文件,或者FTP远程文件的时候,就可以用到自定义的VFS类,是根据在标签里的标签设置生成一个抽象类VFS的子类,最后赋值到Configuration中
private void loadCustomVfs(Properties props) throws ClassNotFoundException {
    String value = props.getProperty("vfsImpl");
    if (value != null) {
      String[] clazzes = value.split(",");
      for (String clazz : clazzes) {
        if (!clazz.isEmpty()) {
          @SuppressWarnings("unchecked")
          Class vfsImpl = (Class)Resources.classForName(clazz);
          configuration.setVfsImpl(vfsImpl);
        }
      }
    }
  }
  • 第四步 loadCustomLogImpl(settings)根据标签获取日志的实现类,包括可以有LOG4J,LOG4J2,SLF4J等待,在这里生成一个Log接口的实现类并赋值到Configuration中
private void loadCustomLogImpl(Properties props) {
    Class logImpl = resolveClass(props.getProperty("logImpl"));
    configuration.setLogImpl(logImpl);
  }
  • 第五步typeAliasesElement(root.evalNode("typeAliases"))这一步是解析类型别名,类的别名和类的关系,放在TypeAliasRegistry对象里面。
 private void typeAliasesElement(XNode parent) {
    // 放入 TypeAliasRegistry
    if (parent != null) {
      for (XNode child : parent.getChildren()) {
        if ("package".equals(child.getName())) {
          String typeAliasPackage = child.getStringAttribute("name");
          configuration.getTypeAliasRegistry().registerAliases(typeAliasPackage);
        } else {
          String alias = child.getStringAttribute("alias");
          String type = child.getStringAttribute("type");
          try {
            Class clazz = Resources.classForName(type);
            if (alias == null) {
              typeAliasRegistry.registerAlias(clazz);
            } else {
              typeAliasRegistry.registerAlias(alias, clazz);
            }
          } catch (ClassNotFoundException e) {
            throw new BuilderException("Error registering typeAlias for '" + alias + "'. Cause: " + e, e);
          }
        }
      }
    }
  }
  • 第六步pluginElement(root.evalNode("plugins"))插件的解析,这一步就是把插件解析换成Interceptor类,设置属性,首先它会先判断是否为空,就是判断配置中心是否配置有这一标签。有则动态代理实现的插件,放入configuration对象的interceptorChain里。
  private void pluginElement(XNode parent) throws Exception {
    if (parent != null) {
      for (XNode child : parent.getChildren()) {
        String interceptor = child.getStringAttribute("interceptor");
        Properties properties = child.getChildrenAsProperties();
        //动态代理 我们实现的插件
        Interceptor interceptorInstance = (Interceptor) resolveClass(interceptor).getDeclaredConstructor().newInstance();
        interceptorInstance.setProperties(properties);
        configuration.addInterceptor(interceptorInstance);
      }
    }
  }
  • 对于objectFactoryElement、objectWrapperFactoryElement、 reflectorFactoryElement的解析跟插件类似;
  • ObjectFactory 是用来创建返回的对象
  • ObjectWrapperFactory 是用来对对象做特殊处理的,
  • ReflectorFactory 是反射工具箱,对反射进行了封装
    这些对象都是用resolveClass创建的
 settingsElement(settings);

看看这个方法做了些什么

 private void settingsElement(Properties props) {
    configuration.setAutoMappingBehavior(AutoMappingBehavior.valueOf(props.getProperty("autoMappingBehavior", "PARTIAL")));
    configuration.setAutoMappingUnknownColumnBehavior(AutoMappingUnknownColumnBehavior.valueOf(props.getProperty("autoMappingUnknownColumnBehavior", "NONE")));
    configuration.setCacheEnabled(booleanValueOf(props.getProperty("cacheEnabled"), true));
    configuration.setProxyFactory((ProxyFactory) createInstance(props.getProperty("proxyFactory")));
    configuration.setLazyLoadingEnabled(booleanValueOf(props.getProperty("lazyLoadingEnabled"), false));
    configuration.setAggressiveLazyLoading(booleanValueOf(props.getProperty("aggressiveLazyLoading"), false));
    configuration.setMultipleResultSetsEnabled(booleanValueOf(props.getProperty("multipleResultSetsEnabled"), true));
    configuration.setUseColumnLabel(booleanValueOf(props.getProperty("useColumnLabel"), true));
    configuration.setUseGeneratedKeys(booleanValueOf(props.getProperty("useGeneratedKeys"), false));
    configuration.setDefaultExecutorType(ExecutorType.valueOf(props.getProperty("defaultExecutorType", "SIMPLE")));
    configuration.setDefaultStatementTimeout(integerValueOf(props.getProperty("defaultStatementTimeout"), null));
    configuration.setDefaultFetchSize(integerValueOf(props.getProperty("defaultFetchSize"), null));
    configuration.setDefaultResultSetType(resolveResultSetType(props.getProperty("defaultResultSetType")));
    configuration.setMapUnderscoreToCamelCase(booleanValueOf(props.getProperty("mapUnderscoreToCamelCase"), false));
    configuration.setSafeRowBoundsEnabled(booleanValueOf(props.getProperty("safeRowBoundsEnabled"), false));
    configuration.setLocalCacheScope(LocalCacheScope.valueOf(props.getProperty("localCacheScope", "SESSION")));
    configuration.setJdbcTypeForNull(JdbcType.valueOf(props.getProperty("jdbcTypeForNull", "OTHER")));
···

可以看到里面就是将中心配置文件的标签里我们设置的属性进行重新赋值,settings二级标签中一共有26个配置,比如二级缓存、延迟加载、默认执行器类型等。一些默认值也是在这里赋值的。

我们再来看看中心配置文件我们设置的数据库相关的配置它是怎么解析的

private void environmentsElement(XNode context) throws Exception {
    if (context != null) {
      if (environment == null) {
        environment = context.getStringAttribute("default");
      }
      for (XNode child : context.getChildren()) {
        String id = child.getStringAttribute("id");
        if (isSpecifiedEnvironment(id)) {
          // 事务工厂
          TransactionFactory txFactory = transactionManagerElement(child.evalNode("transactionManager"));
          // 数据源工厂(例如 DruidDataSourceFactory )
          DataSourceFactory dsFactory = dataSourceElement(child.evalNode("dataSource"));
          // 数据源
          DataSource dataSource = dsFactory.getDataSource();
          // 包含了 事务工厂和数据源的 Environment
          Environment.Builder environmentBuilder = new Environment.Builder(id)
              .transactionFactory(txFactory)
              .dataSource(dataSource);
          // 放入 Configuration
          configuration.setEnvironment(environmentBuilder.build());
        }
      }
    }
  }
  • typeHandlerElement 跟TypeAlias一样,TypeHandler有两种配置方式,一种是单独配置一个类,一种是指定一个package,最后我们得到的是javaType和JdbcType,以及用来做相互映射的TypeHandler之间的映射关系,存放在TypeHandlerRegistry对象里。
private void typeHandlerElement(XNode parent) {
    if (parent != null) {
      for (XNode child : parent.getChildren()) {
        if ("package".equals(child.getName())) {
          String typeHandlerPackage = child.getStringAttribute("name");
          typeHandlerRegistry.register(typeHandlerPackage);
        } else {
          String javaTypeName = child.getStringAttribute("javaType");
          String jdbcTypeName = child.getStringAttribute("jdbcType");
          String handlerTypeName = child.getStringAttribute("handler");
          Class javaTypeClass = resolveClass(javaTypeName);
          JdbcType jdbcType = resolveJdbcType(jdbcTypeName);
          Class typeHandlerClass = resolveClass(handlerTypeName);
          if (javaTypeClass != null) {
            if (jdbcType == null) {
              typeHandlerRegistry.register(javaTypeClass, typeHandlerClass);
            } else {
              typeHandlerRegistry.register(javaTypeClass, jdbcType, typeHandlerClass);
            }
          } else {
            typeHandlerRegistry.register(typeHandlerClass);
          }
        }
      }
    }
  }

映射关系

private final Map>> typeHandlerMap = new ConcurrentHashMap<>();

第二大步骤,就是解析我们在配置文件中配置的Mapper映射器文件了

      // 解析引用的Mapper映射器
      mapperElement(root.evalNode("mappers"));

这一步是解析Mapper映射器文件了,到了这里对于配置中心文件的解析就剩下映射器了,下一篇会详细的介绍。

你可能感兴趣的:(MyBatis源码之中心配置文件)