32. MyBatis中的selectKey标签是什么？如何实现主键回填？

<selectKey>标签是MyBatis中用于在插入数据之前或之后生成主键并将其回填到对象中的一个功能标签。这个标签通常用于处理那些不支持自动生成主键的数据库，或者在需要使用复杂的逻辑来生成主键的场景。

1. <selectKey>标签的作用

• 主键生成：通过<selectKey>标签，可以在插入操作前或插入操作后执行一条SQL语句，用于生成主键值。

• 主键回填：生成的主键值会自动回填到插入数据对应的对象属性中，以便在后续的业务逻辑中使用。

2. <selectKey>标签的属性

• keyProperty：指定要回填主键的对象属性名。

• resultType：指定主键的返回类型，如int、long、string等。

• order：指定主键生成的时机，BEFORE表示在执行插入SQL语句之前生成主键，AFTER表示在执行插入SQL语句之后生成主键。

• statementType：指定SQL语句的类型，默认为STATEMENT，也可以设置为PREPARED。

3. <selectKey>标签的使用场景

3.1 在插入前生成主键

假设我们使用一个数据库序列（如Oracle的sequence）来生成主键，可以通过在插入操作之前生成主键并回填。

示例：使用Oracle的sequence在插入前生成主键

• 数据库表结构：

  • users表：包含id、username、email字段。

• Java实体类：

  public class User {
      private Integer id;
      private String username;
      private String email;
      // Getters and Setters
  }

• 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>

• 解释：

  • keyProperty="id"：表示生成的主键值将被设置到User对象的id属性中。

  • resultType="int"：表示生成的主键值的类型为int。

  • order="BEFORE"：表示在执行插入操作之前生成主键。

  • SELECT seq_users.NEXTVAL FROM DUAL：这是Oracle的序列生成语句，用于获取下一个主键值。

3.2 在插入后生成主键

在某些数据库（如MySQL）中，主键可以在插入数据后由数据库自动生成，例如通过自增字段。这时可以通过<selectKey>在插入操作后获取生成的主键值并回填。

示例：使用MySQL的自增主键在插入后获取主键

• 数据库表结构：

  • users表：包含id（自增）、username、email字段。

• Java实体类：

  public class User {
      private Integer id;
      private String username;
      private String email;
      // Getters and Setters
  }

• MyBatis映射文件：

  <insert id="insertUser" useGeneratedKeys="true" keyProperty="id">
      INSERT INTO users(username, email)
      VALUES(#{username}, #{email})
  </insert>

• 解释：

  • useGeneratedKeys="true"：表示使用JDBC的getGeneratedKeys方法来获取数据库自动生成的主键。

  • keyProperty="id"：表示将生成的主键值回填到User对象的id属性中。

使用<selectKey>获取自增主键的示例

有时可能希望使用<selectKey>来手动获取自增主键：

<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"：表示在插入操作完成后获取主键。

  • SELECT LAST_INSERT_ID()：用于获取MySQL数据库中最后插入记录的自增主键值。

4. 主键回填的过程

当<selectKey>标签被使用时，MyBatis会在插入操作前后执行指定的SQL语句，并将生成的主键值自动回填到对应的对象属性中。这一过程的典型步骤如下：

1. 执行<selectKey>标签内的SQL：如果order为BEFORE，在执行插入操作之前执行<selectKey>中的SQL语句，并将生成的主键值设置到对象中。如果order为AFTER，则在插入操作完成后执行SQL。

2. 回填主键：将生成的主键值回填到指定的Java对象属性中。

3. 执行插入操作：将回填主键的对象插入到数据库中。

5. 使用<selectKey>的优缺点

优点：

• 灵活性高：可以使用复杂的逻辑生成主键，支持各种数据库特性，如Oracle的sequence。

• 跨数据库支持：适用于不支持自动主键生成的数据库。

• 控制力强：可以精确控制主键生成的时机和方式，满足各种业务需求。

缺点：

• 额外的SQL查询：在order="BEFORE"的情况下，会在插入操作前额外执行一条SQL语句，可能会稍微影响性能。

• 复杂度增加：使用<selectKey>需要手动配置SQL语句和主键映射，增加了配置的复杂度。

总结

<selectKey>标签在MyBatis中提供了一种灵活的方式来生成和回填主键，适用于各种场景，包括使用数据库序列、自增主键、UUID等。根据具体需求，可以选择在插入操作之前或之后生成主键，并自动将主键值回填到对象中。合理使用<selectKey>可以帮助我们更好地管理数据库中的主键，确保数据的一致性和完整性。