Mybatis系列番外篇之多参数

Posted 2022-06-04 禹爸爸

Mybatis系列番外篇之多参数

引言

使用过Mybatis的小伙伴们都知道，在映射文件中只能使用parameterType属性指定一个传入参数，可是在实际的项目中，往往需要用到多个传入参数，那么应该如何实现呢？本文就以Mybatis接口式编程方式来分享一下我的实现方式。

分层设计

在实际的工作项目中，需要使用用户和角色的概念对系统权限进行管理，那么就引出了为用户分配角色的问题。

Service层

通常我们会在Service层定义一个方法用来为用户添加角色。这个方法需要传入两个参数，一个是用户id，另一个是角色id列表，方法签名如下。

 

public void grantRoles(int userId, List<Integer> roleIds) throws NotFoundException;

Dao层

但是在Dao层却不可以使用这样的方法签名，因为Mybatis的映射文件中，只能使用parameterType指定一个传入参数，所以，方法签名就变成这个样子了。

 

public void grantRoles(Map<String, Object> parameter);

这里使用了Map类型对Service层传入的两个参数进行了包装，将其转为一个参数。

 

当然，在编写映射文件时，也可以不使用parameterType属性来指定传入参数类型，这样就可以使用类似#0,#1...方式来顺序获取传入的参数。

只是这种方式不适用于此场景中，因为在为用户分配角色时，角色的数量是不确定的，这样就没法确定传入参数的数量。所以使用Map类型对传入参数进行封装。

Mybatis映射文件

 

	<insert id="grantRoles" parameterType="java.util.Map">
	    INSERT IGNORE sec_user_role(user_id, role_id) VALUES
	    <foreach collection="roleIds" item="item" index="index" separator=",">
            (#userId, #item)
        </foreach>
	</insert>

在配置文件中出现的userId和roleIds是则是Map参数中的两个Key，#userId则是取出对应的value，由于roleIds对应的值是一个List类型，所以，这里使用了foreach标签遍历列表中的所有值。

 

Service层实现代码

 

	@Override
	public void grantRoles(int userId, List<Integer> roleIds) throws NotFoundException 
		User user = this.getById(userId);
		if (null != user) 
			Map<String, Object> parameter = new HashMap<String, Object>();
			parameter.put("userId", userId);
			parameter.put("roleIds", roleIds);
			dao.grantRoles(parameter);
		 else 
			String msg = "数据库中查找不到Id为[" + userId + "]的用户！";
			logger.error(msg);
			throw new NotFoundException(msg);