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 是有元素的