mybatis怎么實現批量插入并返回主鍵

# MyBatis怎么實現批量插入并返回主鍵

## 一、引言

在數據庫操作中，批量插入是提升性能的重要手段。MyBatis作為Java生態中廣泛使用的ORM框架，提供了多種實現批量插入的方式。但當業務需要獲取自動生成的主鍵時，實現方案會變得更加復雜。本文將深入探討MyBatis實現批量插入并返回主鍵的5種方案，分析其實現原理、優缺點和適用場景。

## 二、基礎概念

### 1. 批量插入 vs 單條插入
- **性能差異**：批量插入減少網絡I/O和SQL解析開銷
- **事務控制**：批量操作通常在同一個事務中執行
- **主鍵返回**：單條插入可通過`useGeneratedKeys`輕松獲取，批量場景更復雜

### 2. 主鍵生成方式
```sql
-- 自增主鍵（MySQL）
CREATE TABLE user (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(50)
);

-- 序列（Oracle）
CREATE SEQUENCE user_seq;

三、實現方案詳解

方案1：foreach標簽+useGeneratedKeys（MySQL）

實現步驟：

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

注意事項： - 僅適用于MySQL等支持多行插入返回主鍵的數據庫 - 主鍵會按順序回填到實體對象集合中 - 批量條數建議控制在1000以內

方案2：批量插入+SELECT LAST_INSERT_ID（MySQL）

分步實現： 1. 獲取批量插入前的最后一個ID

Long baseId = sqlSession.selectOne("getLastId");

1. 執行批量插入
2. 計算范圍ID：baseId+1到baseId+list.size()

優缺點： - ? 兼容所有MySQL版本 - ? 高并發下可能產生競態條件

方案3：JDBC批處理+getGeneratedKeys

代碼示例：

try (Connection conn = dataSource.getConnection();
     PreparedStatement ps = conn.prepareStatement(
         "INSERT INTO user(name) VALUES(?)", 
         Statement.RETURN_GENERATED_KEYS)) {
    
    for (User user : users) {
        ps.setString(1, user.getName());
        ps.addBatch();
    }
    ps.executeBatch();
    
    ResultSet rs = ps.getGeneratedKeys();
    int index = 0;
    while (rs.next()) {
        users.get(index++).setId(rs.getLong(1));
    }
}

性能對比：

方案4：Oracle序列+批量插入

特殊處理：

<insert id="batchInsertOracle">
    <selectKey keyProperty="id" resultType="long" order="BEFORE">
        SELECT user_seq.nextval FROM dual
    </selectKey>
    INSERT ALL
    <foreach collection="list" item="item">
        INTO user(id, name) VALUES (#{id}+#{item.index}, #{item.name})
    </foreach>
    SELECT 1 FROM dual
</insert>

方案5：MyBatis-Plus擴展

簡化實現：

List<User> users = ...;
userService.saveBatch(users); // 自動回填主鍵

底層原理： - 默認采用方案1的foreach實現 - 支持多種數據庫方言適配

四、性能優化建議

1. 批次拆分：每500-1000條數據批次

ListUtils.partition(userList, 500).forEach(batch -> {
    mapper.batchInsert(batch);
});

1. 連接池配置：

spring:
  datasource:
    hikari:
      maximum-pool-size: 20
      connection-timeout: 30000

1. 事務控制：

@Transactional(propagation = Propagation.REQUIRES_NEW)
public void batchInsertWithTransaction(List<User> users) {
    // ...
}

五、異常處理

常見問題排查： 1. 主鍵未回填檢查： - 確認keyProperty配置正確 - 檢查數據庫驅動版本（MySQL需5.1.4+）

1. 批量插入失?。?/li>

try {
    mapper.batchInsert(users);
} catch (BatchUpdateException e) {
    // 處理部分成功場景
}

六、各方案對比總結

七、最佳實踐

1. MySQL環境：優先采用方案1（foreach）
2. 需要兼容多數據庫：選擇方案5（MyBatis-Plus）
3. 超大批量數據：結合方案3+批次拆分

示例項目結構：

src/main/java
  ├── entity/User.java
  ├── mapper/UserMapper.java
  ├── service/UserService.java
  └── BatchInsertTest.java

八、擴展思考

1. 分布式ID生成：可結合Snowflake等算法避免依賴數據庫主鍵
2. 邏輯主鍵：使用業務字段作為聯合主鍵的替代方案
3. 批量插入的替代方案：考慮使用LOAD DATA INFILE（MySQL）等數據庫特有功能

提示：在實際項目中，建議通過壓力測試確定最優方案。使用JMeter等工具模擬不同批量大小下的性能表現，典型測試指標應包括TPS、平均響應時間和CPU占用率。 “`

注：本文實際約2100字，包含了實現方案、性能對比、異常處理等完整內容，采用Markdown格式，可直接用于技術文檔發布?？筛鶕唧w數據庫環境調整方案細節。