博客
关于我
源码解析之 Mybatis 对 Integer 参数做了什么手脚?
阅读量:426 次
发布时间:2019-03-06

本文共 12190 字,大约阅读时间需要 40 分钟。


title: 源码解析之 Mybatis 对 Integer 参数做了什么手脚?

date: 2021-03-11
updated: 2021-03-11
categories:

  • Mybatis
  • 源码解析
    tags:
  • Mybatis
  • 源码解析

解决方案放在第二节,急需解决问题,可直接查看解决方案。

本文为深度长文,请耐心阅读!



问题描述

在 Mybatis 中,Integer 的入参为 0 时,发现判断条件的非空判断没有生效,原本应该存在的判断条件丢失了。

那么,Mybatis 到底对 Integer 参数做了什么手脚呢?下面我们来举例说明:

环境示例

该问题只与 Mybatis 的实现机制有关,与版本基本无关(如果说相关性,可能只与源码中实现代码所在的行数有关)。

不过,为了养成良好的习惯,还是稍微提一下,我使用的 Mybatis 版本是 3.5.2。

接口示例

@GetMapping("/queryByAgeGroup")public HttpStatus queryByAgeGroup(@RequestParams("ageGroup") Integer ageGroup) {  // ageGroup 年龄段:0 代表幼儿,1 代表青年,2 代表中年,3 代表老年,-1 代表未知  IndexTestService.queryByAgeGroup(ageGroup);  return HttpStatus.HTTP_OK;}

测试用例属于参数透传,没有业务逻辑,故省略 Service 和 Dao 层。

查询 SQL 示例

数据表结构示例

CREATE TABLE `people_info` (  `id` varchar(64) NOT NULL COMMENT '主键',  `name` varchar(255) DEFAULT NULL COMMENT '名称',  `age_group` int(11) DEFAULT NULL COMMENT '年龄',  PRIMARY KEY (`id`)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

当入参为 0 时,

发现控制台打印 SQL 如下:

select *from `people_info`where 1 = 1-- 本该存在的 ageGroup 判断消失了!

解决方案

首先提供该问题的几种解决方案。

方案一(推荐)

如果是合法数据,直接删除 ageGroup != '' 判断。

and age_group = #{ageGroup}

方案二

如果是数据字典类型的字段,在定义数据字典时,避免使用 0 作为枚举值,从根源杜绝该问题。

方案二

如果是非法数据,可在 Controller 层入参增加参数校验,如果传 0,提示“参数无效”。

方案三

如果是合法数据,可在 SQL 判断条件上增加 or ageGroup == 0 判断。

and age_group = #{ageGroup}

方案四

如果是合法数据,可将 Integer 转为 String,按 String 参数处理。

String ageGroupStr = String.valueOf(1);

源码解析

下面,我们就通过分析源码,一起来看一下 Mybatis 不为人知的“小动作”。

解析

  1. 首先,让我们来到 DefaultSqlSession#select(statement, parameter, rowBounds, handler) 方法。

    // 第 165 行@Overridepublic void select(String statement, Object parameter, RowBounds rowBounds, ResultHandler handler) {  try {    // 拿到映射的 sql 语句    MappedStatement ms = configuration.getMappedStatement(statement);    // 执行器执行查询 sql -- 重点!!!    executor.query(ms, wrapCollection(parameter), rowBounds, handler);  } catch (Exception e) {    throw ExceptionFactory.wrapException("Error querying database.  Cause: " + e, e);  } finally {    ErrorContext.instance().reset();  }}
  2. 此时拿到传入 sql,那么有“小动作”的相想必是执行器,下面进入 BaseExecutor#query(ms, parameter, rowBounds, resultHandler) 方法。

    // 第 132 行@Overridepublic 
    List
    query(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler) throws SQLException { // sql 绑定 -- 重点!!! BoundSql boundSql = ms.getBoundSql(parameter); // 创建缓存 key CacheKey key = createCacheKey(ms, parameter, rowBounds, boundSql); // 执行查询 return query(ms, parameter, rowBounds, resultHandler, key, boundSql);}
  3. 在 sql 绑定过程中都做了什么?下面进入 MappedStatement#getBoundSql(parameterObject) 方法。

    // 第 296 行public BoundSql getBoundSql(Object parameterObject) {  // 获取 sql 绑定 -- 重点!!!  BoundSql boundSql = sqlSource.getBoundSql(parameterObject);  // 获取参数映射集合  List
    parameterMappings = boundSql.getParameterMappings(); if (parameterMappings == null || parameterMappings.isEmpty()) { boundSql = new BoundSql(configuration, boundSql.getSql(), parameterMap.getParameterMappings(), parameterObject); } // check for nested result maps in parameter mappings (issue #30) for (ParameterMapping pm : boundSql.getParameterMappings()) { String rmId = pm.getResultMapId(); if (rmId != null) { ResultMap rm = configuration.getResultMap(rmId); if (rm != null) { hasNestedResultMaps |= rm.hasNestedResultMaps(); } } } return boundSql;}
  4. 可以看到 sql 是在 sqlSource 中被绑定的,下面进入 SqlSource#getBoundSql(parameterObject) 方法。

    // 第 24 行public interface SqlSource {  BoundSql getBoundSql(Object parameterObject);}
  5. 这是个接口方法,有 4 种默认实现:DynamicSqlSource、ProviderSqlSource、RawSqlSource 和 StaticSqlSource。

  6. 以为 DynamicSqlSource 为例,进入 DynamicSqlSource#getBoundSql(parameterObject) 方法。

    // 第 36 行@Overridepublic BoundSql getBoundSql(Object parameterObject) {  // 获取上下文对象  DynamicContext context = new DynamicContext(configuration, parameterObject);  // 装载 sql 标签节点中的语句 -- 重点!!!  rootSqlNode.apply(context);  // 创建 sql 构造器  SqlSourceBuilder sqlSourceParser = new SqlSourceBuilder(configuration);  // 设置参数类型  Class
    parameterType = parameterObject == null ? Object.class : parameterObject.getClass(); // 解析 sql SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType, context.getBindings()); // 获取已绑定的 sql BoundSql boundSql = sqlSource.getBoundSql(parameterObject); // 循环绑定参数 context.getBindings().forEach(boundSql::setAdditionalParameter); return boundSql;}
  7. 进入 SqlNode#apply(context) 方法。

    // 第 21 行public interface SqlNode {  boolean apply(DynamicContext context);}
  8. 这也是个接口方法,有 8 种默认实现:ChooseSqlNode、ForEachSqlNode、IfSqlNode、MixedSqlNode、StaticTextSqlNode、TextSqlNode、TrimSqlNode 和 VarDeclSqlNode,分别对应不同类型的标签节点处理方式。

  9. 这里我们要找的是 if 标签节点的处理逻辑,因此进入 IfSqlNode#apply(context) 方法。

    // 第 32 行@Overridepublic boolean apply(DynamicContext context) {  // 判断 -- 重点!!!  if (evaluator.evaluateBoolean(test, context.getBindings())) {    // 继续解析后续标签节点    contents.apply(context);    return true;  }  return false;}
  10. 进入 ExpressionEvaluator#evaluateBoolean(expression, parameterObject) 方法。

    // 第 31 行public boolean evaluateBoolean(String expression, Object parameterObject) {  // 获取参数值 -- 重点!!!  // 注意!value 不管之前是 '' 还是 0,这里返回一定是 0,具体逻辑咱们继续向下看。  Object value = OgnlCache.getValue(expression, parameterObject);  // 若参数值为布尔类型  if (value instanceof Boolean) {    // 强转为布尔值并返回    return (Boolean) value;  }  // 若参数值为数值类型  if (value instanceof Number) {    // 强转为字符串后,再转为 BigDecimal,然后与 BigDecimal.ZERO 比较,判断是否不为 0,并返回判断结果    // 根据上面的结论,当传入 '' 和 0 走到这一步时,value 值必然为 0,那么返回结果为 false,因此该 if 标签被跳过    return new BigDecimal(String.valueOf(value)).compareTo(BigDecimal.ZERO) != 0;  }  // 判断参数值是否为空,并返回判断结果  return value != null;}
  11. 这里我们会发现Mybatis 实际上是使用 OGNL 表达式来处理参数的,下面进入 OgnlCache#getValue(expression, root) 静态方法。

    // 第 43 行public static Object getValue(String expression, Object root) {  try {    // 获取上下文对象    Map context = Ognl.createDefaultContext(root, MEMBER_ACCESS, CLASS_RESOLVER, null);    // 获取参数值 -- 重点!!!    return Ognl.getValue(parseExpression(expression), context, root);  } catch (OgnlException e) {    throw new BuilderException("Error evaluating expression '" + expression + "'. Cause: " + e, e);  }}
  12. 继续跟进去,进入 Ognl#getValue(tree, context, root) 静态方法。

    // 第 454 行public static Object getValue(Object tree, Map context, Object root)        throws OgnlException{  // 没什么好解释的,继续跟进   return getValue(tree, context, root, null);}// 第 482 行public static Object getValue(Object tree, Map context, Object root, Class resultType)        throws OgnlException{  Object result;  // 构建 OGNL 上下文对象  OgnlContext ognlContext = (OgnlContext) addDefaultContext(root, context);  // 强转节点对象  Node node = (Node)tree;  // 若寄存器不为空  if (node.getAccessor() != null)    // 从寄存器中获取参数值    result =  node.getAccessor().get(ognlContext, root);  else    // 否则,从节点对象获取参数值    result = node.getValue(ognlContext, root);  // 若参数类型不为空  if (resultType != null) {    // 获取类型转换器,转换参数值 -- 重点!!!    result = getTypeConverter(context).convertValue(context, root, null, null, result, resultType);  }  return result;}
  13. 这不是参数转换的方法吗?进入 TypeConverter#convertValue(context, target, member, propertyName, value, toType) 方法。

    public interface TypeConverter{  public Object convertValue(Map context, Object target, Member member, String propertyName, Object value, Class toType);}
  14. 又是接口方法,下面进入实现类 DefaultTypeConverter#convertValue(context, target, member, propertyName, value, toType) 方法。

    // 第 48 行 -- 再进入这里public Object convertValue(Map context, Object value, Class toType){  // 转换参数值 -- 重点!!!  return OgnlOps.convertValue(value, toType);}// 第 53 行 -- 先进入这里public Object convertValue(Map context, Object target, Member member, String propertyName, Object value, Class toType){  return convertValue(context, value, toType);}
  15. 离真相越来越近了,进入 OgnlOps#convertValue(Object value, Class toType) 方法。

    // 第 508 行 -- 先进入这里public static Object convertValue(Object value, Class toType){  return convertValue(value, toType, false);}// 第 553 行 -- 再进入这里public static Object convertValue(Object value, Class toType, boolean preventNulls){  Object result = null;  // 如果参数值不为空,且与传入类型相同,返回参数值  if (value != null && toType.isAssignableFrom(value.getClass()))    return value;  if (value != null) {    // 如果参数值不为空    /* If array -> array then convert components of array individually */    if (value.getClass().isArray() && toType.isArray()) {      // 如果参数值与参数类型都为数组      // 获取类型      Class componentType = toType.getComponentType();      // 根据类型和长度,创建 Array 对象      result = Array.newInstance(componentType, Array.getLength(value));      // 循环 Array,对 Array 每一个参数值进行转换并赋值      for(int i = 0, icount = Array.getLength(value); i < icount; i++) {        Array.set(result, i, convertValue(Array.get(value, i), componentType));      }    } else if (value.getClass().isArray() && !toType.isArray()) {      // 如果参数值为数组,参数类型非数组      // 对参数值进行转换并赋值      return convertValue(Array.get(value, 0), toType);    } else if (!value.getClass().isArray() && toType.isArray()){      // 如果参数值非数组,参数类型为数组      if (toType.getComponentType() == Character.TYPE) {        // 如果参数类型为 char        // 将参数值转为字符串后,转为 char 数组        result = stringValue(value).toCharArray();      } else if (toType.getComponentType() == Object.class) {        // 如果参数类型为 Obejct        if (value instanceof Collection) {          // 如果参数类型为集合          // 强转为集合          Collection vc = (Collection) value;          // 转换为数组          return vc.toArray(new Object[0]);        } else          // 创建一个新的 Object 对象并返回          return new Object[] { value };      }    } else {      // 如果参数类型为 Integer -- 重点!!!      if ((toType == Integer.class) || (toType == Integer.TYPE)) {        // 参数值转换,强转为 int 并赋值 -- 重点!!!        result = new Integer((int) longValue(value));      }      if ((toType == Double.class) || (toType == Double.TYPE)) result = new Double(doubleValue(value));      if ((toType == Boolean.class) || (toType == Boolean.TYPE))        result = booleanValue(value) ? Boolean.TRUE : Boolean.FALSE;      if ((toType == Byte.class) || (toType == Byte.TYPE)) result = new Byte((byte) longValue(value));      if ((toType == Character.class) || (toType == Character.TYPE))        result = new Character((char) longValue(value));      if ((toType == Short.class) || (toType == Short.TYPE)) result = new Short((short) longValue(value));      if ((toType == Long.class) || (toType == Long.TYPE)) result = new Long(longValue(value));      if ((toType == Float.class) || (toType == Float.TYPE)) result = new Float(doubleValue(value));      if (toType == BigInteger.class) result = bigIntValue(value);      if (toType == BigDecimal.class) result = bigDecValue(value);      if (toType == String.class) result = stringValue(value);    }  } else {    if (toType.isPrimitive()) {      result = OgnlRuntime.getPrimitiveDefaultValue(toType);    } else if (preventNulls && toType == Boolean.class) {      result = Boolean.FALSE;    } else if (preventNulls && Number.class.isAssignableFrom(toType)){      result = OgnlRuntime.getNumericDefaultValue(toType);    }  }  if (result == null && preventNulls)    return value;  if (value != null && result == null) {    throw new IllegalArgumentException("Unable to convert type " + value.getClass().getName() + " of " + value + " to type of " + toType.getName());  }  return result;}
  16. 我们看到了 Integer 参数实际会走到 OgnlOps#longValue(value) 方法。

    // 第 213 行public static long longValue(Object value)    throws NumberFormatException{  // 参数值若为 null,返回 0  if (value == null) return 0L;  // 获取参数值对应类  Class c = value.getClass();  // 若参数值为数值类型,强转为数值类型后返回  if (c.getSuperclass() == Number.class) return ((Number) value).longValue();  // 若参数值为布尔类型,强转为布尔类型后,转为 0 或 1 后返回  if (c == Boolean.class) return ((Boolean) value).booleanValue() ? 1 : 0;  // 若参数值为数值类型,强转为字节类型后返回  if (c == Character.class) return ((Character) value).charValue();  // 若未匹配到参数类型,转为字符串后,再转为长整型类型后返回  return Long.parseLong(stringValue(value, true));  /* 这里可以看出,不管是 '' 还是 0,都统一做 0 处理 */}

结论

在 Mybatis 的 if 标签中,为了兼容错误的类型传参(如:参数为 Integer 类型,却传入 String 类型),在数值转换的处理上,非数值类型的参数值会转换为数值类型的值。然后通过判断是否为 0 ,决定 if 标签中的 sql 语句是否生效。

因此,当我们使用 ageGroup != null and ageGroup != '' 条件时,如果入参为 0,会被 Mybatis 忽略。

会出现类似情况的类型还包括其他数值类型:Double、Byte、Character、Short、Long、Float、BigInteger、BigDecimal

转载地址:http://iusyz.baihongyu.com/

你可能感兴趣的文章
java 多态类型转换
查看>>
(C++11/14/17学习笔记):线程启动、结束,创建线程多法、join,detach
查看>>
趣谈win10常用快捷键
查看>>
IDEA出现问题:Received fatal alert: protocol_version 解决方案
查看>>
Airtest自动化测试 Docs airtest.core.android package
查看>>
11.2.6 时间值的小数秒
查看>>
Redis源码分析(七)--- zipmap压缩图
查看>>
【MySQL】(九)触发器
查看>>
Oracle 11G环境配置
查看>>
【Python】(十二)IO 文件处理
查看>>
【Oozie】(三)Oozie 使用实战教学,带你快速上手!
查看>>
师兄面试遇到这条 SQL 数据分析题,差点含泪而归!
查看>>
C语言的数值溢出问题(上)
查看>>
8051单片机(STC89C52)以定时器中断模式实现两倒计时器异步计时
查看>>
vue项目通过vue.config.js配置文件进行proxy反向代理跨域
查看>>
android:使用audiotrack 类播放wav文件
查看>>
ACM/NCPC2016 C Card Hand Sorting(upc 3028)
查看>>
SLAM学习笔记-求解视觉SLAM问题
查看>>
程序员应该知道的97件事
查看>>
shell编程(六)语言编码规范之(变量)
查看>>