mybatis $和#源代码分析

JDBC中，主要使用两种语句，一种是支持参数化和预编译的PreparedStatement,支持原生sql,支持设置占位符，参数化输入的参数，防止sql注入攻击，在mybatis的mapper配置文件中，我们通过使用#和$告诉mybatis我们需要对参数进行怎样的设置。sql注入指的是利用现有应用程序将恶意的sql命令提交存在安全漏洞。例如在提交表单时加入or拼接语句使其永远成立。对比JDBC执行流程connection->statement->result,在mybatis中由SqlSession提供给用户操作的API，Excutor具体执行数据库的操作。

SqlSession接口主要的实现类有：

先看DefaultSqlSession:

selectOne的方法最终转化成了selectList方法

最终执行的是箭头指的selectList方法，该方法中含有变量：MappedSatement和Excutor,MappedSatement是加载mapper.xml时匹配的namespace+id

StrictMap是Configuration类中的一个静态内部类，继承了HashMap,看一下selectList方法调用的wrapCollection方法

这就是为什么我们在mapper.xml中foreach用list或array遍历，再看Excutor执行的query方法，Excutor接口实现类

BaseExcutor的query方法：

由上可知若设置清除缓存，首先会清除缓存，首先会根据CacheKey查找缓存，查找结果为空，则从数据库查

 

doQuery:

可见Executor委托给StatementHandler执行查询，在此之前有一个预编译的过程（prepareStatement方法），StatementHandler接口的实现类：CallableStatementHandler,PreparedStatementHandler,SimpleStatementHandler对应JDBC中的CallableStatement,PreparedStatement和Statement,分别的执行方法：

看之前提到的prepareStatement方法

handler会对statement参数化设置，PreparedStatementHandler中：

由parameterhandler执行参数设置，上面是简单分析的查询流程，回头说$和#，我们知道，使用$时statement执行的是拼接操作，#的时候statement用的是占位符 ？，这是mybatis解析的时候造成的，根据测试例子：

SqlSessionFactoryBuilder的build方法：

XMLConfigBuilder负责解析总配置文件，其中方法有：

返回值都为XNode节点类型，看mapperElement:

package扫描包，resource和class扫描指定类和mapper.xml,XMLMapperBuilder:

主要看这两个方法，根据节点建立statement:

由上可以看出XMLStatementBuilder解析statement,也就是mapper.xml中的一个个statement,快成功了。。。

parseStatementNode解析一条记录中的各个属性，例如resultType,parameterType,useCache等等。。。该方法代码过长，其主要在：

两处，一个是SqlSource,一个是addMappedStatement,解析的属性值都对应到MappedStatement对象中

在MappedStatement对象中除这些外，还有个属性SqlSource,可见该对象决定sql语句的解析

只有一个抽象方法getBoundSql,SqlSource是如何获取的呢，这就用到了上面prepareStatement方法中的LanguageDriver的createSqlSource方法，继续跟进：

解析sql语句之前会先解析selectKey和include节点，LanguageDriver的实现类有XMLLanguageDriver和RawLanguageDriver

可以知道createSqlSource方法只在XMLLanguageDriver实现

委托给了XMLScriptBuilder的parseScriptNode方法：

public SqlSource parseScriptNode() {
    List<SqlNode> contents = parseDynamicTags(context);
    MixedSqlNode rootSqlNode = new MixedSqlNode(contents);
    SqlSource sqlSource = null;
    if (isDynamic) {
      sqlSource = new DynamicSqlSource(configuration, rootSqlNode);
    } else {
      sqlSource = new RawSqlSource(configuration, rootSqlNode, parameterType);
    }
    return sqlSource;
  }

根据isDynamic标志确定sqlSource类型，parseDynamicTags方法：

 private List<SqlNode> parseDynamicTags(XNode node) {
    List<SqlNode> contents = new ArrayList<SqlNode>();
    NodeList children = node.getNode().getChildNodes();
    for (int i = 0; i < children.getLength(); i++) {
      XNode child = node.newXNode(children.item(i));
      if (child.getNode().getNodeType() == Node.CDATA_SECTION_NODE || child.getNode().getNodeType() == Node.TEXT_NODE) {
        String data = child.getStringBody("");
        TextSqlNode textSqlNode = new TextSqlNode(data);
        if (textSqlNode.isDynamic()) {
          contents.add(textSqlNode);
          isDynamic = true;
        } else {
          contents.add(new StaticTextSqlNode(data));
        }
      } else if (child.getNode().getNodeType() == Node.ELEMENT_NODE) { // issue #628
        String nodeName = child.getNode().getNodeName();
        NodeHandler handler = nodeHandlers.get(nodeName);
        if (handler == null) {
          throw new BuilderException("Unknown element <" + nodeName + "> in SQL statement.");
        }
        handler.handleNode(child, contents);
        isDynamic = true;
      }
    }
    return contents;
  }

返回值是一个list,过程：

 遍历各个子节点

(1) 如果节点类型是文本或者CDATA，构造一个TextSqlNode或StaticTextSqlNode

(2) 如果节点类型是元素，说明该节点是个动态sql，然后会使用NodeHandler处理各个类型的子节点。这里的NodeHandler是XMLScriptBuilder的一个内部接口，其实现类包括TrimHandler、WhereHandler、SetHandler、IfHandler、ChooseHandler等。看类名也就明白了这个Handler的作用，比如我们分析的trim节点，对应的是TrimHandler；if节点，对应的是IfHandler...，TextSqlNode的isDynamic方法：

  public boolean isDynamic() {
    DynamicCheckerTokenParser checker = new DynamicCheckerTokenParser();
    GenericTokenParser parser = createParser(checker);//建立GenericTokenParser
    parser.parse(text);//GenericTokenParser解析text
    return checker.isDynamic();
  }

createParser方法：

 private GenericTokenParser createParser(TokenHandler handler) { 2 return new GenericTokenParser("${", "}", handler); 3 } 

构造方法：

  public GenericTokenParser(String openToken, String closeToken, TokenHandler handler) {
    this.openToken = openToken;
    this.closeToken = closeToken;
    this.handler = handler;
  }

根据是否Dynamic,TokenHandler的主要实现类有：DynamicCheckerTokenParser和ParameterMappingTokenHandler,VariableTokenHandler

 private GenericTokenParser createParser(TokenHandler handler) { 2 return new GenericTokenParser("${", "}", handler); 3 } //$的处理方式

DynamicCheckerTokenParser实现的handleToken方法

 public String handleToken(String content) {
      this.isDynamic = true;
      return null;
    }

ParameterMappingTokenHandler实现的handleToken方法：

 public String handleToken(String content) {//#的处理方式，返回占位符？
      parameterMappings.add(buildParameterMapping(content));
      return "?";
    }

#的方式大概就是这样，再看$,$使用的是DynamicCheckerTokenParser，这时候再看返回的DynamicSqlSource,其实现SqlSource接口的getBoundSql方法：

public BoundSql getBoundSql(Object parameterObject) {
    DynamicContext context = new DynamicContext(configuration, parameterObject);
    rootSqlNode.apply(context);//apply方法实际调用的是TextSqlNode的applay方法
    SqlSourceBuilder sqlSourceParser = new SqlSourceBuilder(configuration);
    Class<?> parameterType = parameterObject == null ? Object.class : parameterObject.getClass();
    SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType, context.getBindings());//SqlSourceBuilder的parse方法解析
    BoundSql boundSql = sqlSource.getBoundSql(parameterObject);
    for (Map.Entry<String, Object> entry : context.getBindings().entrySet()) {
      boundSql.setAdditionalParameter(entry.getKey(), entry.getValue());
    }
    return boundSql;
  }

TextSqlNode的apply方法：

public boolean apply(DynamicContext context) {
    GenericTokenParser parser = createParser(new BindingTokenParser(context));
    context.appendSql(parser.parse(text));
    return true;
  }

用的是BindingTokenParser的parse方法：

 public String parse(String text) {
    StringBuilder builder = new StringBuilder();
    if (text != null && text.length() > 0) {
      char[] src = text.toCharArray();
      int offset = 0;
      int start = text.indexOf(openToken, offset);
      while (start > -1) {
        if (start > 0 && src[start - 1] == '\\') {
          // the variable is escaped. remove the backslash.
          builder.append(src, offset, start - 1).append(openToken);
          offset = start + openToken.length();
        } else {
          int end = text.indexOf(closeToken, start);
          if (end == -1) {
            builder.append(src, offset, src.length - offset);
            offset = src.length;
          } else {
            builder.append(src, offset, start - offset);
            offset = start + openToken.length();
            String content = new String(src, offset, end - offset);
            builder.append(handler.handleToken(content));//又回到了handleToken方法，此时的handler为BindingTokenParser
            offset = end + closeToken.length();
          }
        }
        start = text.indexOf(openToken, offset);
      }
      if (offset < src.length) {
        builder.append(src, offset, src.length - offset);
      }
    }
    return builder.toString();
  }

BindingTokenParser的handleToken:

public String handleToken(String content) {
      Object parameter = context.getBindings().get("_parameter");
      if (parameter == null) {
        context.getBindings().put("value", null);
      } else if (SimpleTypeRegistry.isSimpleType(parameter.getClass())) {
        context.getBindings().put("value", parameter);//从此处可以看出mapper.xml中$或#中可以用value
      }
      Object value = OgnlCache.getValue(content, context.getBindings());//此处用了ognl处理
      return (value == null ? "" : String.valueOf(value)); // issue #274 return "" instead of "null"
    }

ognl不太清除，先分析到这