Mybatis 源碼(二)Mybatis 初始化

Mybatis 初始化是由SqlSessionFactoryBuilder來完成的,主要的工作解析XML文件,并將解析的類容封裝到Configuration類中,最后將Configuration類封裝到SqlSessionFactory中并返回,自此初始化完成。

完成對XML文件解析的是XMLConfigBuilder、XMLMapperBuilder、XMLStatementBuilder三個類來完成:

  • XMLConfigBuilder:負責全局配置文件(mybatis-config.xml)中除了mappers節(jié)點的解析。
  • XMLMapperBuilder:負責解析xxxMapper.xml映射文件中cache-ref、cache、parameterMapresultMap、sql節(jié)點;根據(jù) namespace 將Mapper接口的動態(tài)代理工廠注冊到 MapperRegistry 中。
  • XMLStatementBuilder:負責解析xxxMapper.xml映射文件中SQL語句節(jié)點,如:select、insert、updatedelete。
  • XMLScriptBuilder:負責解析SQL腳本,然后封裝成SqlSource。

Mybatis 初始化流程

SqlSessionFactoryBuilder

SqlSessionFactoryBuilder會將解析任務托給XMLConfigBuilder類,源碼如下:

public SqlSessionFactory build(InputStream inputStream, String environment, Properties properties) {
  try {
    // 委托給XMLConfigBuilder去解析配置文件
    XMLConfigBuilder parser = new XMLConfigBuilder(inputStream, environment, properties);
    // 開始解析
    return build(parser.parse());
  } 
  ...
}

public SqlSessionFactory build(Configuration config) {
  return new DefaultSqlSessionFactory(config);
}

XMLConfigBuilder

負責全局配置文件(mybatis-config.xml)中除了mappers節(jié)點的解析。解析<mappers>節(jié)點委托給XMLMapperBuilder解析器。

private void parseConfiguration(XNode root) {
  try {
    //issue #117 read properties first
    // 解析<properties>節(jié)點
    propertiesElement(root.evalNode("properties"));
    // 解析<settings>節(jié)點
    Properties settings = settingsAsProperties(root.evalNode("settings"));
    loadCustomVfs(settings);
    loadCustomLogImpl(settings);
    // 解析<typeAliases>節(jié)點
    typeAliasesElement(root.evalNode("typeAliases"));
    pluginElement(root.evalNode("plugins"));
    objectFactoryElement(root.evalNode("objectFactory"));
    objectWrapperFactoryElement(root.evalNode("objectWrapperFactory"));
    reflectorFactoryElement(root.evalNode("reflectorFactory"));
    settingsElement(settings);
    // read it after objectFactory and objectWrapperFactory issue #631
    environmentsElement(root.evalNode("environments"));
    databaseIdProviderElement(root.evalNode("databaseIdProvider"));
    typeHandlerElement(root.evalNode("typeHandlers"));
    // 解析<mappers>節(jié)點
    mapperElement(root.evalNode("mappers"));
  } catch (Exception e) {
    throw new BuilderException("Error parsing SQL Mapper Configuration. Cause: " + e, e);
  }
}

private void mapperElement(XNode parent) throws Exception {
  if (parent != null) {

    for (XNode child : parent.getChildren()) {
      if ("package".equals(child.getName())) {
        String mapperPackage = child.getStringAttribute("name");
        configuration.addMappers(mapperPackage);
      } else {
        String resource = child.getStringAttribute("resource");
        String url = child.getStringAttribute("url");
        String mapperClass = child.getStringAttribute("class");
        if (resource != null && url == null && mapperClass == null) {
          ErrorContext.instance().resource(resource);
          InputStream inputStream = Resources.getResourceAsStream(resource);
          // 委托XMLMapperBuilder來解析映射文件
          XMLMapperBuilder mapperParser = new XMLMapperBuilder(inputStream, configuration, resource, configuration.getSqlFragments());
          mapperParser.parse();
        } else if (resource == null && url != null && mapperClass == null) {
          ErrorContext.instance().resource(url);
          InputStream inputStream = Resources.getUrlAsStream(url);
          // 委托XMLMapperBuilder來解析映射文件
          XMLMapperBuilder mapperParser = new XMLMapperBuilder(inputStream, configuration, url, configuration.getSqlFragments());
          mapperParser.parse();
        } else if (resource == null && url == null && mapperClass != null) {
          Class<?> mapperInterface = Resources.classForName(mapperClass);
          configuration.addMapper(mapperInterface);
        } else {
          throw new BuilderException("A mapper element may only specify a url, resource or class, but not more than one.");
        }
      }
    }
  }
}

XMLMapperBuilder

負責解析xxxMapper.xml映射文件中cache-ref、cache、parameterMap、resultMap、sql節(jié)點;根據(jù) namespace 將Mapper接口的動態(tài)代理工廠(MapperProxyFactory)注冊到 MapperRegistry 中。

public void parse() {
  // 防止重復解析
  if (!configuration.isResourceLoaded(resource)) {
    // 解析xxxMapper.xml 映射文件中的 mapper節(jié)點
    configurationElement(parser.evalNode("/mapper"));
    // 標記為已經(jīng)加載過
    configuration.addLoadedResource(resource);
    // 根據(jù) namespace 將Mapper接口的動態(tài)代理工廠注冊到 MapperRegistry 中
    bindMapperForNamespace();
  }
  // 兜底
  parsePendingResultMaps();
  parsePendingCacheRefs();
  parsePendingStatements();
}

private void configurationElement(XNode context) {
  try {
    String namespace = context.getStringAttribute("namespace");
    if (namespace == null || namespace.equals("")) {
      throw new BuilderException("Mapper's namespace cannot be empty");
    }
    builderAssistant.setCurrentNamespace(namespace);
    // 解析<cache-ref>節(jié)點
    cacheRefElement(context.evalNode("cache-ref"));
    // 解析<cache>節(jié)點
    cacheElement(context.evalNode("cache"));
    // 解析<parameterMap>節(jié)點
    parameterMapElement(context.evalNodes("/mapper/parameterMap"));
    // 解析<resultMap>節(jié)點
    resultMapElements(context.evalNodes("/mapper/resultMap"));
    sqlElement(context.evalNodes("/mapper/sql"));
    // 解析<select|insert|update|delete>節(jié)點
    buildStatementFromContext(context.evalNodes("select|insert|update|delete"));
  } catch (Exception e) {
    throw new BuilderException("Error parsing Mapper XML. The XML location is '" + resource + "'. Cause: " + e, e);
  }
}

private void buildStatementFromContext(List<XNode> list) {
  if (configuration.getDatabaseId() != null) {
    buildStatementFromContext(list, configuration.getDatabaseId());
  }
  buildStatementFromContext(list, null);
}

private void buildStatementFromContext(List<XNode> list, String requiredDatabaseId) {
  for (XNode context : list) {
    // 解析<select|insert|update|delete>節(jié)點委托給 XMLStatementBuilder
    final XMLStatementBuilder statementParser = new XMLStatementBuilder(configuration, builderAssistant, context, requiredDatabaseId);
    try {
      statementParser.parseStatementNode();
    } catch (IncompleteElementException e) {
      configuration.addIncompleteStatement(statementParser);
    }
  }
}

XMLStatementBuilder

負責解析xxxMapper.xml映射文件中SQL語句節(jié)點,如:select、insert、update、delete。解析<select|insert|update|delete>節(jié)點委托給XMLStatementBuilder解析器。

public void parseStatementNode() {
  String id = context.getStringAttribute("id");
  String databaseId = context.getStringAttribute("databaseId");

  if (!databaseIdMatchesCurrent(id, databaseId, this.requiredDatabaseId)) {
    return;
  }

  // 根據(jù)節(jié)點的名稱來判斷SQL語句的類型(select|insert|update|delete)
  String nodeName = context.getNode().getNodeName();
  SqlCommandType sqlCommandType = SqlCommandType.valueOf(nodeName.toUpperCase(Locale.ENGLISH));
  boolean isSelect = sqlCommandType == SqlCommandType.SELECT;
  boolean flushCache = context.getBooleanAttribute("flushCache", !isSelect);
  boolean useCache = context.getBooleanAttribute("useCache", isSelect);
  boolean resultOrdered = context.getBooleanAttribute("resultOrdered", false);

  // Include Fragments before parsing
  // 解析<include>節(jié)點
  XMLIncludeTransformer includeParser = new XMLIncludeTransformer(configuration, builderAssistant);
  includeParser.applyIncludes(context.getNode());

  String parameterType = context.getStringAttribute("parameterType");
  Class<?> parameterTypeClass = resolveClass(parameterType);

  String lang = context.getStringAttribute("lang");
  LanguageDriver langDriver = getLanguageDriver(lang);

  // Parse selectKey after includes and remove them.
  // 解析<selectKey>節(jié)點,并在XML中刪除<selectKey>節(jié)點
  processSelectKeyNodes(id, parameterTypeClass, langDriver);

  // Parse the SQL (pre: <selectKey> and <include> were parsed and removed)
  KeyGenerator keyGenerator;
  String keyStatementId = id + SelectKeyGenerator.SELECT_KEY_SUFFIX;
  keyStatementId = builderAssistant.applyCurrentNamespace(keyStatementId, true);
  if (configuration.hasKeyGenerator(keyStatementId)) {
    keyGenerator = configuration.getKeyGenerator(keyStatementId);
  } else {
    keyGenerator = context.getBooleanAttribute("useGeneratedKeys",
        configuration.isUseGeneratedKeys() && SqlCommandType.INSERT.equals(sqlCommandType))
        ? Jdbc3KeyGenerator.INSTANCE : NoKeyGenerator.INSTANCE;
  }

  // 解析SQL語句,然后封裝成SqlSource
  SqlSource sqlSource = langDriver.createSqlSource(configuration, context, parameterTypeClass);
  StatementType statementType = StatementType.valueOf(context.getStringAttribute("statementType", StatementType.PREPARED.toString()));
  Integer fetchSize = context.getIntAttribute("fetchSize");
  Integer timeout = context.getIntAttribute("timeout");
  String parameterMap = context.getStringAttribute("parameterMap");
  String resultType = context.getStringAttribute("resultType");
  Class<?> resultTypeClass = resolveClass(resultType);
  String resultMap = context.getStringAttribute("resultMap");
  String resultSetType = context.getStringAttribute("resultSetType");
  ResultSetType resultSetTypeEnum = resolveResultSetType(resultSetType);
  if (resultSetTypeEnum == null) {
    resultSetTypeEnum = configuration.getDefaultResultSetType();
  }
  String keyProperty = context.getStringAttribute("keyProperty");
  String keyColumn = context.getStringAttribute("keyColumn");
  String resultSets = context.getStringAttribute("resultSets");

  // 使用建造者模式構建MappedStatement對象
  builderAssistant.addMappedStatement(id, sqlSource, statementType, sqlCommandType,
      fetchSize, timeout, parameterMap, parameterTypeClass, resultMap, resultTypeClass,
      resultSetTypeEnum, flushCache, useCache, resultOrdered,
      keyGenerator, keyProperty, keyColumn, databaseId, langDriver, resultSets);
}

Mybatis 初始化流程圖

mybatis初始化過程.png

核心數(shù)據(jù)結構類

Configuration

Configuration其實就是XML配置文件的Java形態(tài),Configuration是單例的,生命周期是應用級的;

configuration.png

ResultMap

對應xxxMapper.xml映射文件中的resultMap節(jié)點。<resultMap>節(jié)點中的子節(jié)點使用ResultMapping來封裝,如:<id>、<result>等節(jié)點。

ResultMap.png

MappedStatement

對應xxxMapper.xml映射文件中的<select>、<insert>、<update><delete>節(jié)點。

MappedStatment.png

SqlSource

對應xxxMapper.xml映射文件中的sql語句,經(jīng)過解析SqlSource包含的語句最終僅僅包含?占位符,可以直接提交給數(shù)據(jù)庫執(zhí)行;

MapperRegistry

它是Mapper接口動態(tài)代理工廠類的注冊中心。在MyBatis中,通過MapperProxy實現(xiàn)InvocationHandler接口,通過MapperProxyFactory生成動態(tài)代理的實例對象;

MyBatis建造者類圖

Mybatis的整個初始化過程使用了建造者模式。建造者模式比較適合生成復雜對象,主要關注對象的實例化具體細節(jié)。


MyBatis建造者類圖.png

Mybatis 源碼中文注釋

https://github.com/xiaolyuh/mybatis

最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容