文档章节

Mybatis源码分析之参数处理

Lucare
 Lucare
发布于 2018/07/23 20:29
字数 2413
阅读 30
收藏 1

Mybatis对参数的处理是值得推敲的,不然在使用的过程中对发生的一系列错误直接懵了。

以前遇到参数绑定相关的错误我就是直接给加@param注解,也稀里糊涂地解决了,但是后来遇到了一些问题推翻了我的假设:单个参数不需要使用 @param 。由此产生了一个疑问,Mybatis到底是怎么处理参数的?

几种常见的情景:

单个参数

  • 不使用注解,基于${}和#{}的引用,基本类型和自定义对象都可以
  • 不使用注解,基于foreach标签的使用,list和array不可以
  • 不使用注解,基于if标签的判断,基本类型 boolean 也报错

初步封装

第一次处理是在MapperMethod中:

private Object getParam(Object[] args) {
    final int paramCount = paramPositions.size();
    if (args == null || paramCount == 0) {
      return null;
    } else if (!hasNamedParameters && paramCount == 1) {
      return args[paramPositions.get(0)];
    } else {
      Map<String, Object> param = new MapperParamMap<Object>();
      for (int i = 0; i < paramCount; i++) {
        param.put(paramNames.get(i), args[paramPositions.get(i)]);
      }
      // issue #71, add param names as param1, param2...but ensure backward compatibility
      for (int i = 0; i < paramCount; i++) {
        String genericParamName = "param" + String.valueOf(i + 1);
        if (!param.containsKey(genericParamName)) {
          param.put(genericParamName, args[paramPositions.get(i)]);
        }
      }
      return param;
    }
}

这里会有三种可能:null,object[],MapperParamMap<Object>,第三种可以构造出我们常见的param1、parm2......

AuthAdminUser findAuthAdminUserByUserId(@Param("userId") String userId);

当我们在Mapper接口中如此定义时,就会走上面的else代码块,MapperParamMap将包含两个元素,一个key为userId,另一个为param1。

第二次处理是在DefaultSqlSession中,调用executor的query方法时,将参数包装成集合:

  private Object wrapCollection(final Object object) {
    if (object instanceof List) {
      StrictMap<Object> map = new StrictMap<Object>();
      map.put("list", object);
      return map;
    } else if (object != null && object.getClass().isArray()) {
      StrictMap<Object> map = new StrictMap<Object>();
      map.put("array", object);
      return map;
    }
    return object;
  }

这个时候会将其他两种类型(list或array)也转换为map集合,MapperParamMap和StrictMap都继承了HashMap,只是将super.containsKey(key)为false的时候抛出了一个异常。

实例呈现

当我们写Mapper接口时,一个参数通常也不使用@param注解。

如果这个参数是 List 类型呢?

List<String> selectFeeItemTypeNameByIds(List<Integer> itemIds);

对应的mapper配置文件:

<select id="selectFeeItemTypeNameByIds" parameterType="java.util.List" resultType="java.lang.String">
	SELECT fee_item_type_name
	FROM tb_uhome_fee_item_type
	WHERE fee_item_type_id IN
	<foreach collection="itemIds" item="itemId" open="(" close=")" separator="," >
		#{itemId}
	</foreach>
</select>

测试一下,直接报错:

nested exception is org.apache.ibatis.binding.BindingException: Parameter 'itemIds' not found. Available parameters are [list]

然后把itemIds替换为list就好了:

<foreach collection="list" item="itemId" open="(" close=")" separator="," >
	#{itemId}
</foreach>

这个正是验证了上述源码中的操作,在DefaultSqlSession的wrapCollection方法中:

if (object instanceof List) {
  StrictMap<Object> map = new StrictMap<Object>();
  map.put("list", object);
  return map;
}

如果这个参数用在 if 标签中呢?

List<Map<String, Object>> selectPayMethodListByPlatform(boolean excludeInner);

xml中这样使用:

<select id="selectPayMethodListByPlatform" resultType="java.util.HashMap" parameterType="boolean">
	select a.`NAME`as payMethodName, a.`VALUE` as payMethod
	from tb_fcs_dictionary a
	where a.`CODE` = 'PAY_METHOD'
	and a.`STATUS` = 1
	and a.TYPE = 'PLATFORM'
	<if test="excludeInner">
		and a.value not in (14,98)
	</if>
</select>

直接报如下错误:

There is no getter for property named 'excludeInner' in 'class java.lang.Boolean'

跟踪下DynamicContext的内部类ContextAccessor的getProperty方法: image

那我们加上注解@Param("excludeInner") 再看看: image

没有使用注解,存储的就是一个Boolean类型的值,返回null。使用了注解,这个值有名称且存放在MapperParamMap中,直接可以根据名称取到。

查看调用栈

在ForEachSqlNode中会调用ExpressionEvaluator的evaluateIterable方法来获取迭代器对象:

public Iterable<?> evaluateIterable(String expression, Object parameterObject) {
    try {
      Object value = OgnlCache.getValue(expression, parameterObject);
      if (value == null) throw new SqlMapperException("The expression '" + expression + "' evaluated to a null value.");
      if (value instanceof Iterable) return (Iterable<?>) value;
      if (value.getClass().isArray()) {
          // the array may be primitive, so Arrays.asList() may throw 
          // a ClassCastException (issue 209).  Do the work manually
          // Curse primitives! :) (JGB)
          int size = Array.getLength(value);
          List<Object> answer = new ArrayList<Object>();
          for (int i = 0; i < size; i++) {
              Object o = Array.get(value, i);
              answer.add(o);
          }
          
          return answer;
      }
      throw new BuilderException("Error evaluating expression '" + expression + "'.  Return value (" + value + ") was not iterable.");
    } catch (OgnlException e) {
      throw new BuilderException("Error evaluating expression '" + expression + "'. Cause: " + e, e);
    }
}

IfSqlNode中也会调用ExpressionEvaluator的evaluateBoolean方法来检测表达式正确与否:

public boolean evaluateBoolean(String expression, Object parameterObject) {
    try {
      Object value = OgnlCache.getValue(expression, parameterObject);
      if (value instanceof Boolean) return (Boolean) value;
      if (value instanceof Number) return !new BigDecimal(String.valueOf(value)).equals(BigDecimal.ZERO);
      return value != null;
    } catch (OgnlException e) {
      throw new BuilderException("Error evaluating expression '" + expression + "'. Cause: " + e, e);
    }
}

两者都会使用Ognl来获取表达式的值:

Object value = OgnlCache.getValue(expression, parameterObject);

实际处理

在DynamicSqlSource的getBoundSql方法中:

  • 参数绑定

DynamicContext context = new DynamicContext(configuration, parameterObject);

  public DynamicContext(Configuration configuration, Object parameterObject) {
    if (parameterObject != null && !(parameterObject instanceof Map)) {
      MetaObject metaObject = configuration.newMetaObject(parameterObject);
      bindings = new ContextMap(metaObject);
    } else {
      bindings = new ContextMap(null);
    }
    bindings.put(PARAMETER_OBJECT_KEY, parameterObject);
    bindings.put(DATABASE_ID_KEY, configuration.getDatabaseId());
  }
  • Node逐级处理(各种标签和${}的处理)

rootSqlNode.apply(context);

这个就是处理动态sql的关键,将if、choose和foreach等剥离出来,使用ognl的表达式来获取相关属性的值,例如上面提到的foreach和if标签。

然后将其转换成简单的text,在TextSqlNode中最终处理${param},将其替换为实际参数值。

替换方式如下:

public String handleToken(String content) {
  try {
    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);
    }
    Object value = OgnlCache.getValue(content, context.getBindings());
    return (value == null ? "" : String.valueOf(value)); // issue #274 return "" instead of "null"
  } catch (OgnlException e) {
    throw new BuilderException("Error evaluating expression '" + content + "'. Cause: " + e, e);
  }
}
  • 参数解析(#{}的处理)

SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType);

SqlSourceBuilder#parse:

public SqlSource parse(String originalSql, Class<?> parameterType) {
    ParameterMappingTokenHandler handler = new ParameterMappingTokenHandler(configuration, parameterType);
    GenericTokenParser parser = new GenericTokenParser("#{", "}", handler);
    String sql = parser.parse(originalSql);
    return new StaticSqlSource(configuration, sql, handler.getParameterMappings());
}

GenericTokenParser的parse方法将#{xx}替换为 ? ,如下面的sql语句:

SELECT DISTINCT
	A.ORGAN_ID as organId,
	CONCAT(A. NAME, ' [', IFNULL(A.PY_NAME, ''), ']') as organName
FROM
	ORGAN A,
	ORGAN_REL B,
	V_USER_ORGAN C
WHERE
	A.ORGAN_ID = B.ORGAN_ID
AND B.ORGAN_CODE LIKE CONCAT(LEFT(C.ORGAN_CODE, 8), '%')
AND B.PAR_ID = 1
AND A.STATUS = 1
AND C.USER_ID = #{userId}

替换后为:

SELECT DISTINCT
	A.ORGAN_ID as organId,
	CONCAT(A. NAME, ' [', IFNULL(A.PY_NAME, ''), ']') as organName
FROM
	ORGAN A,
	ORGAN_REL B,
	V_USER_ORGAN C
WHERE
	A.ORGAN_ID = B.ORGAN_ID
AND B.ORGAN_CODE LIKE CONCAT(LEFT(C.ORGAN_CODE, 8), '%')
AND B.PAR_ID = 1
AND A.STATUS = 1
AND C.USER_ID = ?

然后构造一个StaticSqlSource:

new StaticSqlSource(configuration, sql, handler.getParameterMappings());

这个就跟我们直接使用JDBC一样,使用?作为占位符。

最终在DefaultParameterHandler中给设置进参数:

public void setParameters(PreparedStatement ps)
      throws SQLException {
    ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());
    List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
    if (parameterMappings != null) {
      MetaObject metaObject = parameterObject == null ? null : configuration.newMetaObject(parameterObject);
      for (int i = 0; i < parameterMappings.size(); i++) {
        ParameterMapping parameterMapping = parameterMappings.get(i);
        if (parameterMapping.getMode() != ParameterMode.OUT) {
          Object value;
          String propertyName = parameterMapping.getProperty();
          PropertyTokenizer prop = new PropertyTokenizer(propertyName);
          if (parameterObject == null) {
            value = null;
          } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
            value = parameterObject;
          } else if (boundSql.hasAdditionalParameter(propertyName)) {
            value = boundSql.getAdditionalParameter(propertyName);
          } else if (propertyName.startsWith(ForEachSqlNode.ITEM_PREFIX)
              && boundSql.hasAdditionalParameter(prop.getName())) {
            value = boundSql.getAdditionalParameter(prop.getName());
            if (value != null) {
              value = configuration.newMetaObject(value).getValue(propertyName.substring(prop.getName().length()));
            }
          } else {
            value = metaObject == null ? null : metaObject.getValue(propertyName);
          }
          TypeHandler typeHandler = parameterMapping.getTypeHandler();
          if (typeHandler == null) {
            throw new ExecutorException("There was no TypeHandler found for parameter " + propertyName + " of statement " + mappedStatement.getId());
          }
          JdbcType jdbcType = parameterMapping.getJdbcType();
          if (value == null && jdbcType == null) jdbcType = configuration.getJdbcTypeForNull();
          typeHandler.setParameter(ps, i + 1, value, jdbcType);
        }
      }
    }
  }

这里分为五种情况(高版本合并了第三和第四种):

  • parameterObject为null,value直接为null
  • parameterObject类型为typeHandlerRegistry中匹配类型value直接赋值为parameterObject
  • 参数是动态参数,通过动态参数取值
  • 参数是动态参数而且是foreach中的(前缀为_frch_),也是通过动态参数取值
  • 复杂对象或者map类型,通过反射取值

总结

像 if 和 foreach 这种标签都是直接通过Ognl来取值。

"${}" 的处理在TextSqlNode中,使用OGNL方式取值,当场替换为实际参数值。

"#{}" 的处理在SqlSourceBuilder的parse中,使用占位符(?)替换,最后在设置参数的时候使用Mybatis的MetaObject取值。

当我们使用单个参数未用注解时:

  • 用在形如foreach和if的标签中(针对上面两个实例)
List<String> selectFeeItemTypeNameByIds(List<Integer> itemIds);

List<Map<String, Object>> selectPayMethodListByPlatform(boolean excludeInner);

MapperMethod的getParam方法将返回这两个参数本身。

DefaultSqlSession的wrapCollection方法将把list放到一个key为 "list"的map中,boolean类型的还是返回本身。

这样在DynamicSqlSource的getBoundSql方法中构造DynamicContext时:

public DynamicContext(Configuration configuration, Object parameterObject) {
    if (parameterObject != null && !(parameterObject instanceof Map)) {
      MetaObject metaObject = configuration.newMetaObject(parameterObject);
      bindings = new ContextMap(metaObject);
    } else {
      bindings = new ContextMap(null);
    }
    bindings.put(PARAMETER_OBJECT_KEY, parameterObject);
    bindings.put(DATABASE_ID_KEY, configuration.getDatabaseId());
}

list类型的由于被包装了一下,将走else。而boolean类型直接创建一个包含metaObject的ContextMap。

不管怎样,“itemIds”走到这里已经丢了,后面解析表达式的时候根据这个名字是肯定拿不到的。

而boolean类型的 "excludeInner" 将在ContextMap中如此出现(仅仅有个值key却为“_parameter”):

key: "_parameter"  value: true
key: "_databaseId"  value: "MySQL"

不过它持有的MetaObject类型的parameterMetaObject对象却不为null。

看下ContextMap中的重写的get方法:

public Object get(Object key) {
      String strKey = (String) key;
      if (super.containsKey(strKey)) {
        return super.get(strKey);
      }

      if (parameterMetaObject != null) {
        Object object = parameterMetaObject.getValue(strKey);
        if (object != null) {
          super.put(strKey, object);
        }
            
        return object;
      }
        
      return null;
}

当父类中没有时(这个肯定没有),它将去parameterMetaObject中拿,这一拿就拿出问题来了:

There is no getter for property named 'excludeInner' in 'class java.lang.Boolean'

一路跟到MetaObject的getValue方法,又到BeanWrapper的get方法,然后就把它当做一个普通的对象,用反射去调它的get方法:

private Object getBeanProperty(PropertyTokenizer prop, Object object) {
    try {
      Invoker method = metaClass.getGetInvoker(prop.getName());
      try {
        return method.invoke(object, NO_ARGUMENTS);
      } catch (Throwable t) {
        throw ExceptionUtil.unwrapThrowable(t);
      }
    } catch (RuntimeException e) {
      // 进了这个运行时异常:说它没的get方法,哈哈
      throw e;
    } catch (Throwable t) {
      throw new ReflectionException("Could not get property '" + prop.getName() + "' from " + object.getClass() + ".  Cause: " + t.toString(), t);
    }
}

这个excludeInner本来就是一个boolean类型的参数,哪有什么get方法,能调到才怪!

针对上面两个实例的分析就结束了,从这里也大致知道了Mybatis是如何处理参数的。总的来说,不管一个参数还是几个参数,加@param注解是没错的!加了就会给你统统放map里,然后到ContextMap中取整个map,由于是map类型,将继续到map里取具体的对象。

从这里可以看出来,如果我们在接口中声明时就只用一个map来装所有参数,key为参数名,value为参数值,然后不使用注解,效果也是一样的。


有问题欢迎讨论。

© 著作权归作者所有

Lucare

Lucare

粉丝 6
博文 11
码字总数 11404
作品 0
深圳
程序员
私信 提问
mybatis源码分析之TypeHandler

上一篇: mybatis源码分析之MapperMethod https://my.oschina.net/u/657390/blog/755787 分析了MapperMethod从创建到执行的过程,MapperMethod的执行包括执行sql返回结果. 在执行sql和返回结果...

udbwcso
2016/11/18
601
0
源码分析 Mybatis 的 foreach 为什么会出现性能问题

背景 最近在做一个类似于综合报表之类的东西,需要查询所有的记录(数据库记录有限制),大概有1W条记录,该报表需要三个表的数据,也就是根据这 1W 个 ID 去执行查询三次数据库,其中,有一...

TSMYK
2018/12/16
1K
13
mybatis源码探究(-)MapperProxyFactory&MapperProxy

在MyBatis中MapperProxyFactory,MapperProxy,MapperMethod是三个很重要的类。 弄懂了这3个类你就大概清楚Mapper接口与SQL的映射, 为什么是接口,没有实例类也可以完成注入或者调用。 其中M...

贝克田庄
2018/08/22
0
0
Mybatis执行SQL的4大基础组件详解

温馨提示:本篇是源码分析Mybatis ShardingJdbc SQL语句执行的前置篇。 源码分析Mybatis系列目录: 1、源码分析Mybatis MapperProxy初始化之Mapper对象的扫描与构建 2、源码分析Mybatis Map...

丁威
09/17
0
0
mybatis(五)TypeHandler简介及配置(mybatis源码篇)

上篇文章为大家介绍了mybatis中别名的使用,以及其源码。本篇将为大家介绍TypeHandler, 并简单分析其源码。 Mybatis中的TypeHandler是什么?   无论是 MyBatis 在预处理语句(PreparedSta...

LCZ777
2014/12/19
90
0

没有更多内容

加载失败,请刷新页面

加载更多

springboot 403 问题

添加WebAppConfigurer 配置 @Configuration@EnableAutoConfigurationpublic class WebAppConfigurer extends WebMvcConfigurerAdapter { public WebAppConfigurer() { } ......

布袋和尚_爱吃鱼
24分钟前
3
0
Python自动更换壁纸爬虫与tkinter结合

直接上代码 import ctypesimport timeimport requestsimport osfrom threading import Threadfrom tkinter import Tk, Label, Button,Entry,StringVar,messagebox# '放到AppData\Roami......

物种起源-达尔文
24分钟前
3
0
Postgresql Study 笔记

Postgresql 安装 Windows, MAC Install Postgresql 下载地址: https://www.enterprisedb.com/downloads/postgres-postgresql-downloads Linux Install sudo apt-get update sudo apt-get in......

slagga
26分钟前
3
0
layer.open 打开新页面传参问题

如图所示,点击出售,把A页面的数据传到弹框上面,因为弹框比较复杂,所以使用引入一个新页面。 A.html a.js B.html b.js 1、第一种方案 sellInte: function (){ var obj = document.g...

木九天
29分钟前
4
0
沙龙报名 | 区块链数据服务技术应用实践

京东云是国内首家提供区块链数据在线分析服务产品的公司,也是行业内首家对区块链数据服务进行开源的公司。 本次沙龙是京东云BDS开源后,首次在深圳举办线下沙龙,我们将邀请京东云BDS团队核...

京东云技术新知
29分钟前
4
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部