MyBatis中selectKey的使用

2020-12-05

数据库主键包括自增和非自增，有时候新增一条数据后可能还需要这个新增的主键，可以采用selectKey来帮助我们获取新增的主键。

这里针对自增主键进行测试。

1. 实体类

public class User {
    private long userId;
    private String username;
    private String password;

    // getter&setter
}

2. Dao层

@Repository
public interface UserDao {
    long save(@Param("user") User user);
}

3. mapper.xml

<insert id="save" parameterType="demo.srammy.selectkey.model.User">
    <selectKey resultType="java.lang.Long" order="AFTER" keyProperty="user.userId">
        SELECT LAST_INSERT_ID()
    </selectKey>
    insert into t_user (user_name,password)
    values (#{user.username},#{user.password})
</insert>

selectKey会将SELECT LAST_INSERT_ID()的结果放入到传入的model的主键里面。

keyProperty对应model中的主键的属性名，这里是实体类中的userId，因为它跟数据库的主键对应。注意这里要写出参数名，这里参数名是user（在UserDao类中@Param("user")指定参数名为user）。

order=AFTER表示SELECT LAST_INSERT_ID()在insert执行之后执行，多用与自增主键。order=BEFORE表示SELECT LAST_INSERT_ID()在insert执行之前执行，这种适合那种主键不是自增的类型resultType。

4. 测试

@Autowired
private UserDao userDao;

@Test
public void contextLoads() {
    User user = new User();
    user.setUsername("bbb");
    user.setPassword("iuiuiui");
    System.out.println(userDao.save(user)); // 1
    System.out.println(user.getUserId()); // 13 返回了新增user对应的主键
}

1. 实体类

2. Dao层

3. mapper.xml

4. 测试

参考