購物下單后，訂單號是怎么自動生成的？插入數(shù)據(jù)后如何直接獲取自增主鍵？本文將帶你深入理解 MyBatis 中的主鍵回填機制，并通過流程圖和代碼示例徹底掌握它。

一、業(yè)務(wù)場景：下單后如何返回訂單編號？

想象一個典型的購物交易流程：

1. 用戶點擊“立即購買”，填寫收貨信息，提交訂單。
2. 后端服務(wù)將訂單數(shù)據(jù)（用戶ID、商品ID、金額、狀態(tài)等）插入到數(shù)據(jù)庫的 orders 表中。
3. 插入成功后，系統(tǒng)需要立即返回一個訂單編號給前端，用于后續(xù)支付、查詢等操作。

這個訂單編號通常是數(shù)據(jù)庫表的主鍵，比如自增的 id 字段。但在執(zhí)行插入 SQL 時，我們并不知道這個 id 是多少 —— 它是數(shù)據(jù)庫自動生成的。

如果采用“先插入，再查詢”的方式：

-- 第一步：插入訂單（id 自動生成）
INSERT INTO orders(user_id, amount, status) VALUES(1001, 299.00, 0);
-- 第二步：根據(jù)唯一業(yè)務(wù)字段查詢剛插入的訂單
SELECT id FROM orders WHERE user_id = 1001 AND create_time = ...;

這種做法不僅繁瑣，而且在高并發(fā)下很容易出錯（多個用戶同時下單，條件可能不唯一）。于是，主鍵回填技術(shù)應(yīng)運而生。

二、什么是主鍵回填？

主鍵回填（Key Fillback / Key Return）是指在執(zhí)行數(shù)據(jù)庫插入操作后，自動將生成的主鍵值填充到傳入的 Java 對象中。這樣，你無需再次查詢，就能直接通過對象的 getId() 方法拿到主鍵。

簡單來說：插入時主鍵為 null，插入后 MyBatis 幫你把主鍵值“填回去”。

效果演示

Order order = new Order();
order.setUserId(1001);
order.setAmount(299.00);
order.setStatus(0);
// 此時 order.getId() == null
orderMapper.insertOrder(order);
// 插入后，MyBatis 自動將生成的主鍵賦值給 order 對象
System.out.println(order.getId()); // 輸出 12345

三、主鍵回填的兩種實現(xiàn)方式

MyBatis 主要提供兩種方式實現(xiàn)主鍵回填，分別適用于不同的數(shù)據(jù)庫和場景。

方式一：useGeneratedKeys+keyProperty（推薦，簡單高效）

適用數(shù)據(jù)庫：支持自動生成主鍵的數(shù)據(jù)庫，如 MySQL、SQL Server（自增列）、PostgreSQL（serial 類型）等。

原理：MyBatis 利用 JDBC 的 Statement.getGeneratedKeys() 方法獲取數(shù)據(jù)庫自動生成的主鍵。

配置示例：

<insert id="insertOrder" useGeneratedKeys="true" keyProperty="id">
    INSERT INTO orders(user_id, amount, status)
    VALUES (#{userId}, #{amount}, #{status})
</insert>

• useGeneratedKeys="true"：開啟主鍵回填功能。
• keyProperty="id"：指定將生成的主鍵值賦給 Java 對象的哪個屬性（對應(yīng) Order 類中的 id 字段）。

Java 接口方法：

public interface OrderMapper {
    int insertOrder(Order order);
}

使用注意：

• 只能用于單條插入（批量插入時需特殊處理，不同數(shù)據(jù)庫支持情況不一）。
• 數(shù)據(jù)庫表必須定義自增列或類似機制。

方式二：<selectKey>標(biāo)簽（靈活，支持多種數(shù)據(jù)庫）

適用場景：

• 數(shù)據(jù)庫沒有自增列（如 Oracle 使用序列）。
• 需要在插入前生成主鍵（比如使用 UUID）。
• 需要執(zhí)行自定義查詢來獲取主鍵值（如 MySQL 的 LAST_INSERT_ID()）。

原理：通過子查詢執(zhí)行一個 SQL 語句來獲取主鍵，可以指定在插入語句之前（order="BEFORE"）或之后（order="AFTER"）執(zhí)行。

示例 1：MySQL 插入后獲取自增 ID

<insert id="insertOrder">
    <!-- 插入后執(zhí)行，獲取最后插入的 ID -->
    <selectKey keyProperty="id" resultType="int" order="AFTER">
        SELECT LAST_INSERT_ID()
    </selectKey>
    INSERT INTO orders(user_id, amount, status)
    VALUES (#{userId}, #{amount}, #{status})
</insert>

示例 2：Oracle 插入前從序列獲取 ID

<insert id="insertOrder">
    <selectKey keyProperty="id" resultType="long" order="BEFORE">
        SELECT ORDERS_SEQ.NEXTVAL FROM DUAL
    </selectKey>
    INSERT INTO orders(id, user_id, amount, status)
    VALUES (#{id}, #{userId}, #{amount}, #{status})
</insert>

示例 3：使用 UUID 作為主鍵（插入前生成）

<insert id="insertUser">
    <selectKey keyProperty="id" resultType="String" order="BEFORE">
        SELECT REPLACE(UUID(), '-', '')
    </selectKey>
    INSERT INTO users(id, name) VALUES (#{id}, #{name})
</insert>

屬性說明：

• keyProperty：目標(biāo)對象的屬性名。
• resultType：主鍵的 Java 類型。
• order：BEFORE（在插入前執(zhí)行）或 AFTER（在插入后執(zhí)行）。

四、流程圖：主鍵回填的執(zhí)行過程

為了更直觀地理解兩種方式的執(zhí)行流程，下面用流程圖表示。

1.useGeneratedKeys方式流程

2.<selectKey order="AFTER">方式流程（以 MySQL 為例）

3.<selectKey order="BEFORE">方式流程（以 Oracle 為例）

五、兩種方式的對比與選擇

選擇建議：

• MySQL + 自增主鍵 → 優(yōu)先使用 useGeneratedKeys。
• Oracle 序列 / UUID / 復(fù)合主鍵 → 使用 <selectKey>。
• 需要插入前生成主鍵（如分布式 ID） → 使用 <selectKey order="BEFORE">。

六、常見問題與注意事項

1. 為什么插入后對象的 id 還是 null？

• 檢查 useGeneratedKeys="true" 和 keyProperty 是否正確配置。
• 檢查 keyProperty 對應(yīng)的屬性在 Java 對象中是否有 setter 方法（MyBatis 通過反射賦值）。
• 確認(rèn)數(shù)據(jù)庫表確實支持自增列（MySQL 需 AUTO_INCREMENT）。

2. 批量插入時能否使用主鍵回填？

• MySQL 驅(qū)動支持批量插入時返回生成的主鍵，但 MyBatis 的 useGeneratedKeys 對批量支持不統(tǒng)一。建議使用循環(huán)單條插入，或者使用 <foreach> 配合特殊寫法（需要數(shù)據(jù)庫驅(qū)動支持）。
• 簡單場景下，可以放棄批量回填，插入后單獨查詢。

3.LAST_INSERT_ID()與并發(fā)

MySQL 的 LAST_INSERT_ID() 是連接級別的，每個客戶端連接返回自己最后插入的 ID，不會受其他并發(fā)連接影響，因此是安全的。

4. 主鍵回填后，對象會被修改嗎？

會！MyBatis 會直接修改傳入的 Java 對象的屬性值。因此你無需重新接收返回值，直接使用原對象即可。

七、完整示例代碼

數(shù)據(jù)庫表（MySQL）

CREATE TABLE `orders` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `user_id` int(11) NOT NULL,
  `amount` decimal(10,2) NOT NULL,
  `status` tinyint(4) DEFAULT '0',
  PRIMARY KEY (`id`)
);

Java 實體類

public class Order {
    private Integer id;
    private Integer userId;
    private BigDecimal amount;
    private Integer status;
    // 省略 getter / setter / toString
}

Mapper 接口

@Mapper
public interface OrderMapper {
    // 方式一：useGeneratedKeys
    int insertOrder(Order order);
    // 方式二：selectKey (MySQL AFTER)
    int insertOrderWithSelectKey(Order order);
}

XML 映射文件

<!-- 方式一 -->
<insert id="insertOrder" useGeneratedKeys="true" keyProperty="id">
    INSERT INTO orders(user_id, amount, status)
    VALUES (#{userId}, #{amount}, #{status})
</insert>
<!-- 方式二 -->
<insert id="insertOrderWithSelectKey">
    <selectKey keyProperty="id" resultType="int" order="AFTER">
        SELECT LAST_INSERT_ID()
    </selectKey>
    INSERT INTO orders(user_id, amount, status)
    VALUES (#{userId}, #{amount}, #{status})
</insert>

測試代碼

@SpringBootTest
class OrderMapperTest {
    @Autowired
    private OrderMapper orderMapper;
    @Test
    void testInsert() {
        Order order = new Order();
        order.setUserId(1001);
        order.setAmount(new BigDecimal("199.00"));
        order.setStatus(0);
        int rows = orderMapper.insertOrder(order);
        System.out.println("影響行數(shù)：" + rows);
        System.out.println("回填后的主鍵：" + order.getId()); // 輸出自增 ID
    }
}

八、總結(jié)

主鍵回填是 MyBatis 中非常實用且高頻使用的特性，它解決了插入數(shù)據(jù)后立即獲取主鍵的痛點。掌握兩種實現(xiàn)方式：

• useGeneratedKeys：簡潔高效，MySQL 首選。
• <selectKey>：靈活強大，適配各種數(shù)據(jù)庫及非自增主鍵場景。

在實際開發(fā)中，根據(jù)數(shù)據(jù)庫類型和業(yè)務(wù)需求選擇合適的方式，可以大大簡化代碼邏輯，提升開發(fā)效率。

以上就是MyBatis主鍵回填的兩種實現(xiàn)方式的詳細(xì)內(nèi)容，更多關(guān)于MyBatis實現(xiàn)主鍵回填的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論