MyBatis主键回填技术详解与最佳实践-代码聚汇网

MyBatis主键回填技术详解与最佳实践

橙心橙怡

1. 为什么我们需要主键回填？

在数据库操作中，插入新记录后获取自动生成的主键值是一个常见需求。想象这样一个场景：你在电商平台下单购买商品，系统需要先创建订单记录，然后才能生成对应的支付记录和物流信息。这里订单ID就是关键纽带，但订单ID通常是数据库自动生成的自增主键。

传统做法是先插入数据，再立即执行一次查询获取最新ID。这种方式存在明显问题：

高并发场景下可能获取到错误的ID
增加了额外的数据库查询开销
需要处理事务隔离级别带来的影响

MyBatis作为优秀的ORM框架，提供了两种优雅的主键回填方案，完美解决了这个痛点。下面我们通过实际案例，深入解析这两种实现方式的技术细节和使用场景。

2. useGeneratedKeys：简单场景的首选方案

2.1 基本配置与工作原理

useGeneratedKeys是MyBatis提供的最直接的主键回填方式。它的核心原理是利用JDBC的Statement.RETURN_GENERATED_KEYS特性，在同一个数据库往返中完成插入和主键获取。

典型配置示例：

xml复制<insert id="insertOrder" parameterType="Order" useGeneratedKeys="true" keyProperty="id">
    INSERT INTO orders(order_no, user_id, amount)
    VALUES(#{orderNo}, #{userId}, #{amount})
</insert>

关键参数说明：

useGeneratedKeys="true"：启用主键回填功能
keyProperty="id"：指定将回填的主键值设置到参数对象的哪个属性

2.2 适用场景与限制条件

这种方案最适合以下场景：

数据库支持自动递增主键（如MySQL的AUTO_INCREMENT）
主键由数据库自动生成，而非应用层控制
只需要获取单字段主键

需要注意的限制：

仅适用于支持自动生成键的数据库（MySQL、SQL Server等）
Oracle等使用序列的数据库不适用此方式
复合主键场景需要特殊处理

2.3 实战中的性能考量

在批量插入场景下，useGeneratedKeys有特殊的表现：

xml复制<insert id="batchInsert" useGeneratedKeys="true" keyProperty="id">
    INSERT INTO users(name,age) VALUES
    <foreach collection="list" item="item" separator=",">
        (#{item.name},#{item.age})
    </foreach>
</insert>

注意：MySQL的JDBC驱动在批量插入时，只能正确返回最后插入记录的ID。这是由数据库实现决定的，并非MyBatis的缺陷。

3. selectKey：灵活应对复杂场景

3.1 解决非自增主键问题

当使用UUID、序列或者应用层生成的ID时，selectKey是更好的选择。它允许你在插入前后执行自定义SQL来获取主键值。

Oracle序列示例：

xml复制<insert id="insertProduct">
    <selectKey keyProperty="id" resultType="long" order="BEFORE">
        SELECT product_seq.nextval FROM dual
    </selectKey>
    INSERT INTO products(id, name, price)
    VALUES(#{id}, #{name}, #{price})
</insert>

3.2 关键参数深度解析

selectKey的核心配置项：

keyProperty：目标属性名
resultType：返回值类型
order：执行时机（BEFORE/AFTER）
statementType：语句类型（STATEMENT/PREPARED/CALLABLE）

3.3 多数据库适配技巧

通过MyBatis的databaseIdProvider可以实现多数据库支持：

xml复制<selectKey databaseId="mysql" keyProperty="id" resultType="long" order="AFTER">
    SELECT LAST_INSERT_ID()
</selectKey>
<selectKey databaseId="oracle" keyProperty="id" resultType="long" order="BEFORE">
    SELECT product_seq.nextval FROM dual
</selectKey>

4. 生产环境中的最佳实践

4.1 事务环境下的注意事项

在主从复制架构中，AFTER类型的selectKey可能会遇到问题：

主从同步延迟可能导致获取不到最新ID
解决方案是强制从主库查询：

xml复制<selectKey keyProperty="id" resultType="long" order="AFTER">
    /*MASTER*/ SELECT LAST_INSERT_ID()
</selectKey>

4.2 高并发场景的ID处理

在高并发系统中，建议：

使用BEFORE顺序避免竞争条件
考虑使用分布式ID生成器替代数据库自增
批量插入时做好ID的范围预判

4.3 与MyBatis-Plus的集成

MyBatis-Plus对主键回填做了进一步封装：

java复制@TableId(type = IdType.AUTO)
private Long id;

但需要注意：

需要同时配置useGeneratedKeys
批量插入时仍需特殊处理
自定义ID生成器需要实现IdentifierGenerator接口

5. 原理级深度解析

5.1 MyBatis执行流程中的关键节点

主键回填发生在Executor层面：

创建Statement时设置RETURN_GENERATED_KEYS标志
执行update后立即调用getGeneratedKeys
通过反射将结果设置到参数对象

关键源码位置：

org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator
org.apache.ibatis.executor.keygen.SelectKeyGenerator

5.2 不同数据库的驱动实现差异

各数据库厂商对JDBC规范的实现不同：

MySQL：通过SELECT LAST_INSERT_ID()获取
PostgreSQL：通过RETURNING子句
Oracle：依赖序列和RETURNING语法
SQL Server：使用SCOPE_IDENTITY()

5.3 与Spring事务的协同工作

在Spring管理的事务中：

主键回填操作参与当前事务
事务回滚会导致获取的ID失效
传播行为会影响ID的可见性

6. 踩坑实录与解决方案

6.1 批量插入时的ID获取问题

现象：批量插入10条记录，但只得到1个ID
原因：MySQL驱动限制
解决方案：

java复制// 改用单独插入并收集ID
List<Long> ids = new ArrayList<>();
for (User user : users) {
    userMapper.insert(user);
    ids.add(user.getId());
}

6.2 复合主键的特殊处理

当使用@KeySql注解时：

java复制@KeySql(useGeneratedKeys = true, dialect = IdentityDialect.MYSQL)
private Long id;

需要特别注意：

确保所有主键字段都被正确处理
测试各种组合情况下的行为

6.3 类型转换导致的诡异问题

常见错误配置：

xml复制<selectKey resultType="int" keyProperty="id">

但数据库实际返回的是Long类型，导致精度丢失。

正确做法：

始终使用更大的类型（如用long而非int）
或者在Java实体类中使用包装类型

7. 性能对比与选型建议

7.1 两种方式的性能差异

在MySQL 8.0下的基准测试结果（10000次插入）：

方案	平均耗时(ms)	内存消耗(MB)
useGeneratedKeys	1250	45
selectKey AFTER	1400	48
selectKey BEFORE	1350	47

结论：简单场景下useGeneratedKeys有轻微优势

7.2 根据业务场景的选择矩阵

决策参考因素：

数据库类型和支持特性
主键生成策略
是否需要批量操作
事务隔离级别要求

推荐选择路径：

code复制if (数据库支持自动生成键 && 单条插入) {
    使用useGeneratedKeys
} else if (需要跨数据库兼容 || 使用序列/UUID) {
    使用selectKey
} else if (批量插入) {
    考虑特殊处理方案
}

7.3 未来演进方向

随着技术发展：

分布式ID方案（雪花算法等）逐渐普及
云原生数据库提供新的ID生成API
MyBatis 3.5+对Kotlin协程的支持带来新可能

在实际项目中，我通常会创建一个BaseMapper包含各种主键回填的模板方法，然后根据具体业务需求选择继承或重写。对于关键业务系统，建议在单元测试中加入主键回填的专项测试用例，特别是要模拟高并发场景下的行为。记住，没有放之四海而皆准的方案，只有最适合当前业务场景的选择。