mybatis之foreach用法
在做 mybatis 的 mapper.xml 文件的时候,我们时常用到这样的情况:动态生成 sql 语句的查询条件,这个时候我们就可以用 mybatis 的 foreach 了
foreach 元素的属性主要有 item,index,collection,open,separator,close。
- item:集合中元素迭代时的别名,该参数为必选。
- index:在 list 和数组中,index 是元素的序号,在 map 中,index 是元素的 key,该参数可选
- open:foreach 代码的开始符号,一般是 (和 close=")" 合用。常用在 in(),values() 时。该参数可选
- separator:元素之间的分隔符,例如在 in()的时候,separator="," 会自动在元素中间用“,“隔开,避免手动输入逗号导致 sql 错误,如 in(1,2,) 这样。该参数可选。
- close: foreach 代码的关闭符号,一般是 )和 open="(" 合用。常用在 in(),values() 时。该参数可选。
- collection: 要做 foreach 的对象,作为入参时,List 对象默认用 "list" 代替作为键,数组对象有 "array" 代替作为键,Map 对象没有默认的键。当然在作为入参时可以使用 @Param("keyName") 来设置键,设置 keyName 后,list,array 将会失效。 除了入参这种情况外,还有一种作为参数对象的某个字段的时候。举个例子:如果 User 有属性 List ids。入参是 User 对象,那么这个 collection = "ids".如果 User 有属性 Ids ids; 其中 Ids 是个对象,Ids 有个属性 List id; 入参是 User 对象,那么 collection = "ids.id"
在使用 foreach 的时候最关键的也是最容易出错的就是 collection 属性,该属性是必须指定的,但是在不同情况下,该属性的值是不一样的,主要有一下 3 种情况:
- 如果传入的是单参数且参数类型是一个 List 的时候,collection 属性值为 list .
- 如果传入的是单参数且参数类型是一个 array 数组的时候,collection 的属性值为 array .
- 如果传入的参数是多个的时候,我们就需要把它们封装成一个 Map 了,当然单参数也可以封装成 map,实际上如果你在传入参数的时候,在 MyBatis 里面也是会把它封装成一个 Map 的,map 的 key 就是参数名,所以这个时候 collection 属性值就是传入的 List 或 array 对象在自己封装的 map 里面的 key.
针对最后一条,我们来看一下官方说法:
注意 你可以将一个 List 实例或者数组作为参数对象传给 MyBatis,当你这么做的时候,MyBatis 会自动将它包装在一个 Map 中并以名称为键。List 实例将会以“list”作为键,而数组实例的键将是“array”。
所以,不管是多参数还是单参数的 list,array 类型,都可以封装为 map 进行传递。如果传递的是一个 List,则 mybatis 会封装为一个 list 为 key,list 值为 object 的 map,如果是 array,则封装成一个 array 为 key,array 的值为 object 的 map,如果自己封装呢,则 colloection 里放的是自己封装的 map 里的 key 值。
源码分析
由于官方文档对这块的使用,描述的比较简短,细节上也被忽略掉了 (可能是开源项目文档一贯的问题吧),也使用不少同学在使用中遇到了问题。特别是 foreach 这个函数中,collection 属性做什么用,有什么注意事项。由于文档不全,这块只能通过源代码剖析的方式来分析一下各个属性的相关要求。
collection 属性的用途是接收输入的数组或是 List 接口实现。但对于其名称的要求,Mybatis 在实现中还是有点不好理解的,所以需要特别注意这一点。
下面开始分析源代码 (笔记使用的是 Mybatis 3.0.5 版本)
先找到 Mybatis 执行 SQL 配置解析的入口
MapperMethod.java 类中 public Object execute(Object[] args) 该方法是执行的入口.
针对 in 集合查询,对应用就是 selectForList 或 SelctForMap 方法。
但不管调用哪个方法,都会对原来 JDK 传入的参数 Object[] 类型,通过 getParam 方法转换成一个 Object, 那这个方法是做什么的呢?分析源码如下:
上图中标红的两处,很惊讶的发现,一个参数与多个参数的处理方式是不同的 (后续很多同学遇到的问题,就有一大部分出自这个地方)。如果参数个数大于一个,则会被封装成 Map, key 值如果使用了 Mybatis 的 Param 注解,则会使用该 key 值,否则默认统一使用数据序号, 从 1 开始。这个问题先记下,继续分析代码,接下来如果是 selectForList 操作 (其它操作就对应用相应方法), 会调用 DefaultSqlSession 的 public List selectList(String statement, Object parameter, RowBounds rowBounds) 方法
又一个发现,见源代码如下:
上图标红部分,对参数又做了一次封装,我们看一下代码
现在有点清楚了,如果参数类型是 List, 则必须在 collecion 中指定为 list, 如果是数据组,则必须在 collection 属性中指定为 array.
现在就问题就比较清楚了,如果是一个参数的话,collection 的值取决于你的参数类型。
如果是多个值的话,除非使用注解 Param 指定,否则都是数字开头,所以在 collection 中指定什么值都是无用的。下图是 debug 显示结果。
使用方法
1. 单参数 List 类型
1 <select id="countByUserList" resultType="_int" parameterType="list"> 2 select count(*) from users 3 <where> 4 id in 5 <foreach item="item" collection="list" separator="," open="(" close=")" index=""> 6 #{item.id, jdbcType=NUMERIC} 7 </foreach> 8 </where> 9 </select>
测试代码:
1 @Test 2 public void shouldHandleComplexNullItem() { 3 SqlSession sqlSession = sqlSessionFactory.openSession(); 4 try { 5 Mapper mapper = sqlSession.getMapper(Mapper.class); 6 User user1 = new User(); 7 user1.setId(2); 8 user1.setName("User2"); 9 List<User> users = new ArrayList<User>(); 10 users.add(user1); 11 users.add(null); 12 int count = mapper.countByUserList(users); 13 Assert.assertEquals(1, count); 14 } finally { 15 sqlSession.close(); 16 } 17 }
2. 单参数 array 数组的类型:
1 <select id="dynamicForeach2Test" resultType="Blog"> 2 select * from t_blog where id in 3 <foreach collection="array" index="index" item="item" open="(" separator="," close=")"> 4 #{item} 5 </foreach> 6 </select>
上述 collection 为 array,对应的 Mapper 代码:
public List dynamicForeach2Test(int[] ids);
对应的测试代码:
1 @Test 2 public void dynamicForeach2Test() { 3 SqlSession session = Util.getSqlSessionFactory().openSession(); 4 BlogMapper blogMapper = session.getMapper(BlogMapper.class); 5 int[] ids = new int[] {1,3,6,9}; 6 List blogs = blogMapper.dynamicForeach2Test(ids); 7 for (Blog blog : blogs) 8 System.out.println(blog); 9 session.close(); 10 }
3. 自己把参数封装成 Map 的类型
1 <select id="dynamicForeach3Test" resultType="Blog"> 2 select * from t_blog where title like "%"#{title}"%" and id in 3 <foreach collection="ids" index="index" item="item" open="(" separator="," close=")"> 4 #{item} 5 </foreach> 6 </select>
上述 collection 的值为 ids,是传入的参数 Map 的 key,对应的 Mapper 代码:
public List dynamicForeach3Test(Map params);
对应测试代码:
1 @Test 2 public void dynamicForeach3Test() { 3 SqlSession session = Util.getSqlSessionFactory().openSession(); 4 BlogMapper blogMapper = session.getMapper(BlogMapper.class); 5 final List ids = new ArrayList(); 6 ids.add(1); 7 ids.add(2); 8 ids.add(3); 9 ids.add(6); 10 ids.add(7); 11 ids.add(9); 12 Map params = new HashMap(); 13 params.put("ids", ids); 14 params.put("title", "中国"); 15 List blogs = blogMapper.dynamicForeach3Test(params); 16 for (Blog blog : blogs) 17 System.out.println(blog); 18 session.close(); 19 }
注意注意 sql 语句 SELECT * FROM ny_jobs WHERE id IN () 这个会报错,所以最后判断一下 Ids 是有元素的