MyBatis 获取自增主键

实体类属性如下

public class User {
    private Integer id;
    private String username;
    private String userEmail;
    private String userCity;
    private Integer age;
    // 省略get、set方法
}

如果数据库支持自动生成主键的字段（比如 MySQL 和 SQL Server），那么可以设置 useGeneratedKeys=”true”，然后再把 keyProperty 设置到目标属性上就 OK 了，keyProperty设置的是需要映射到实体类中的属性。

插入单条记录

接口定义如下

public int addUser(User user);

接口对应的 Mapper.xml 定义如下所示

<insert id="addUser"
        parameterType="com.example.mybatis.entity.User"
        useGeneratedKeys="true"
        keyProperty="id">
    insert into user
    (username, user_email, user_city, age)
    values (#{username}, #{userEmail}, #{userCity}, #{age})
</insert>

Mybatis会将获取到的自增主键设置到User类中的id属性

插入多条记录

接口定义如下

public int addUserList(List<User> userList); 

接口对应的 Mapper.xml 定义如下所示

<insert id="addUserList"
        parameterType="com.example.mybatis.entity.User"
        useGeneratedKeys="true"
        keyProperty="id">
    insert into user
    (username, user_email, user_city, age)
    values
    <foreach item="item" collection="list" separator=",">
        (#{item.username}, #{item.userEmail}, #{item.userCity}, #{item.age})
    </foreach>
</insert>

特殊情况

对于不支持自动生成类型的数据库或可能不支持自动生成主键的 JDBC 驱动，MyBatis 有另外一种方法来生成主键。这里有一个简单的示例，通过select unix_timestamp(now())可以获取当前时间戳

<insert id="addUser">
    <selectKey keyProperty="id" resultType="int" order="BEFORE">
        select unix_timestamp(now())
    </selectKey>
    insert into user
    (id, username, user_email, user_city, age)
    values (#{id}, #{username}, #{userEmail}, #{userCity}, #{age})
</insert>

在上面的示例中，selectKey元素中的语句将会首先运行，User的 id 会被设置，然后插入语句会被调用。这可以提供给你一个与数据库中自动生成主键类似的行为，同时保持了 Java 代码的简洁。

selectKey 元素描述如下：

<selectKey
  keyProperty="id"
  resultType="int"
  order="BEFORE"
  statementType="PREPARED">

本作品采用《CC 协议》，转载必须注明作者和本文链接