Mybatis多表插入

很多时候我们会碰到这样一个需求，当前我有（用户）表和（角色）表，他们之间是多对多的关系，所以我们通常会创建一个关联表（用户_角色）表，现在我们有固定的角色，只需要添加用户即可，给用户指定角色，那我们怎么做呢，废话不多说了，下面给出思路:

        表 :user   /   role   /   user_role 

对应JavaBean如下:

用户表:

[java]  view plain  copy

1. public class User implements Serializable {  
2.   
3.     // Fields  
4.   
5.     private Integer id;  
6.     private String login;  
7.     private String name;  
8.     private String pass;  
9.     private String phone;  
10.     private String email;  
11.     private Integer isadmin;  
12.   
13.        // get...set...  
14. }  

角色表:

[java]  view plain  copy

1. public class Role implements Serializable {  
2.   
3.     // Fields  
4.   
5.     private Integer roleId;  
6.     private String roleName;  
7.     private Date createdTime;  
8.     private Date updateTime;  
9.   
10.     //get...set...  
11. }  

用户_角色关联表(当然，它不创建也可以):

[java]  view plain  copy

1. public class UserRole implements Serializable {  
2.   
3.     // Fields  
4.   
5.     private Integer id;  
6.     private Integer userId;  
7.     private Integer roleId;  
8.   
9.     //get...set...  
10. }  

//**以下所有操作中，所有JavaBean使用别名(typeAlias)方式，不会的话自行百度

需求，添加用户并为当前用户指定多个角色,主要Sql:

1: 首先插入用户（实现插入用户并且返回插入用户的主键）:

[sql]  view plain  copy

1. <insert id="save" parameterType="User" useGeneratedKeys="true"  
2.         keyProperty="id">  
3.         insert into user(login,name,pass,phone,email,isadmin)  
4.         values(#{login},#{name},#{pass},#{phone},#{email},${isadmin})  
5. </insert>  

//useGeneratedKeys="true"  设置是否使用JDBC的getGenereatedKeys方法获取主键并赋值到keyProperty设置的主键字段中

//keyProperty="id"  指定JavaBean的主键字段

2:为用户匹配角色:

方式一（适用于除了插入被关联表的ID，还要插入被关联表中的其他信息，例如:角色类型（ 当前设计中此字段不存在，仅仅用于举例子））：

[sql]  view plain  copy

1. <insert id="saveUserOrRole" parameterType="map">  
2.         insert into user_role(  
3.         user_id,role_id,role_type  
4.         )  
5.         select u.id,r.role_id,r.role_type from user u,role r where  
6.         u.id = ${userId} and  
7.         r.role_id in  
8.         <foreach collection="roleIds" item="rId" open="(" separator=","  
9.             close=")">  
10.             #{rId}  
11.         </foreach>  
12. </insert>  

方式二（适用于只需要插入被关联表的ID）

[sql]  view plain  copy

1. insert into user_role(  
2.         user_id,role_id  
3.         ) values  
4.         <foreach collection="roleIds" item="rId" separator=",">  
5.             (${userId},#{rId})  
6.         </foreach>  

//注意在执行完第一条插入用户sql后，在外部访问对象User的主键是已经存在的，类似hibernate持久化后对象已经拥有ID信息

这里把测试代码给出（Junit4）:

[java]  view plain  copy

1. @RunWith(SpringJUnit4ClassRunner.class)  
2. @ContextConfiguration("classpath*:spring-*.xml")  
3. public class TestSSM {  
4.     @Resource  
5.     protected UserMapper userMapper;  
6.       
7.       
8.     @Test  
9.     public void testInsert() {  
10.         User user = new User();  
11.         user.setEmail("yhq1913@sina.com");  
12.         user.setLogin("ytzl");  
13.         user.setName("云图智联");  
14.         //密码加密  
15.         user.setPass(DigestUtil.hmacSign("ytzl", "yuntuzhilian"));  
16.         user.setPhone("15910721339");  
17.         user.setIsadmin(1);  
18.         userMapper.save(user);  
19.         Map<String, Object> saveParams = new HashMap<>();  
20.         saveParams.put("userId", user.getId());  
21.         //这里指定两个角色的ID进行测试  
22.         saveParams.put("roleIds", Arrays.asList("1", "2"));  
23.         userMapper.saveUserOrRole(saveParams);  
24.     }  
25. }