10. MyBatis如何处理数据库主键生成？有哪些常见的主键生成策略？

PangHu大约 4 分钟

在MyBatis中，处理数据库主键生成通常涉及到如何获取插入数据后生成的主键值，并将其返回给调用者。MyBatis提供了多种方式来处理主键生成，具体策略取决于数据库和业务需求。以下是MyBatis中常见的主键生成策略及其使用方式。

1. 数据库自动生成主键

许多数据库支持自动生成主键（如MySQL的自增主键）。MyBatis可以直接利用数据库的自动生成机制，并在插入记录后获取生成的主键。

1.1 使用`useGeneratedKeys`和`keyProperty`

适用场景：使用数据库的自增主键或序列等自动生成主键。
配置方式：
```
<insert id="insertUser" useGeneratedKeys="true" keyProperty="id">
    INSERT INTO users(username, email) VALUES(#{username}, #{email})
</insert>
```
- useGeneratedKeys="true"：表示启用数据库的自动生成主键功能。
- keyProperty="id"：表示将生成的主键值注入到实体类的id属性中。
说明：这种方式依赖于JDBC的getGeneratedKeys()方法，MyBatis会自动获取生成的主键并赋值给实体类。

1.2 使用`@Options`注解

适用场景：同样适用于数据库的自增主键或序列。

配置方式：

@Insert("INSERT INTO users(username, email) VALUES(#{username}, #{email})")
@Options(useGeneratedKeys = true, keyProperty = "id")
void insertUser(User user);

@Options注解与useGeneratedKeys和keyProperty属性具有相同的功能，主要用于基于注解的Mapper接口。

2. 使用`<selectKey>`标签

<selectKey>标签用于在插入数据之前或之后生成主键，并将其赋值给实体类。适用于一些需要手动生成主键或使用数据库序列的场景。

2.1 插入前生成主键

适用场景：需要在插入数据之前生成主键，例如使用UUID、时间戳或数据库序列。

配置方式：

<insert id="insertUser">
    <selectKey keyProperty="id" resultType="int" order="BEFORE">
        SELECT seq_users.NEXTVAL FROM DUAL
    </selectKey>
    INSERT INTO users(id, username, email) VALUES(#{id}, #{username}, #{email})
</insert>

order="BEFORE"：表示在执行主SQL语句（INSERT INTO）之前生成主键。
SELECT seq_users.NEXTVAL FROM DUAL：使用数据库序列生成主键。

说明：这种方式适合在插入之前就需要生成主键的场景，比如需要在应用程序中显示生成的主键。

2.2 插入后生成主键

适用场景：在某些情况下，可以在插入数据后从数据库获取生成的主键。

配置方式：

<insert id="insertUser">
    INSERT INTO users(username, email) VALUES(#{username}, #{email})
    <selectKey keyProperty="id" resultType="int" order="AFTER">
        SELECT LAST_INSERT_ID()
    </selectKey>
</insert>

order="AFTER"：表示在执行主SQL语句之后生成或获取主键。
SELECT LAST_INSERT_ID()：获取MySQL数据库生成的最后一个自增主键。

说明：这种方式通常用于那些不支持自动生成主键的数据库或需要在插入之后获取主键的场景。

3. UUID生成主键

如果数据库不提供自动生成主键的功能，或者你希望主键具有更高的唯一性，可以在应用程序中生成UUID作为主键。

配置方式：

使用Java生成UUID：

public interface UserMapper {
    @Insert("INSERT INTO users(id, username, email) VALUES(#{id}, #{username}, #{email})")
    void insertUser(User user);
}

// 在业务层生成UUID
User user = new User();
user.setId(UUID.randomUUID().toString());
userMapper.insertUser(user);

使用<selectKey>标签生成UUID：

<insert id="insertUser">
    <selectKey keyProperty="id" resultType="string" order="BEFORE">
        SELECT UUID()
    </selectKey>
    INSERT INTO users(id, username, email) VALUES(#{id}, #{username}, #{email})
</insert>

说明：UUID生成的主键具有全局唯一性，适合分布式系统和跨库的数据集成，但它的长度较长，不适合作为外键使用。

4. 数据库序列生成主键

某些数据库（如Oracle）支持通过序列生成主键，MyBatis可以在插入前获取序列的下一个值作为主键。

配置方式：

<insert id="insertUser">
    <selectKey keyProperty="id" resultType="int" order="BEFORE">
        SELECT seq_users.NEXTVAL FROM DUAL
    </selectKey>
    INSERT INTO users(id, username, email) VALUES(#{id}, #{username}, #{email})
</insert>

说明：序列生成的主键通常是递增的数字，适合大多数情况下的主键生成需求。

常见的主键生成策略总结

数据库自增主键：通过useGeneratedKeys和keyProperty自动获取数据库自增的主键，适用于MySQL等支持自增主键的数据库。
UUID：通过Java或数据库函数生成UUID，适合需要全局唯一标识的场景。
数据库序列：通过数据库序列生成主键，适用于Oracle等支持序列的数据库。
手动生成主键：通过业务逻辑手动生成主键（如时间戳、组合键），适用于需要自定义主键的场景。

总结

MyBatis提供了多种方式来处理主键生成，主要包括利用数据库的自增主键、UUID生成、序列生成以及通过<selectKey>标签生成主键。选择哪种策略主要取决于数据库的特性、应用的需求以及业务场景。通过合理配置这些策略，可以有效地管理数据库中的主键生成，确保数据的一致性和系统的稳定性。