Mybatis批量插入Oracle數據的筆記

基本環境

語言：Java 8 數據庫：Oracle ORM 框架：MyBatis 3.4.5

需求

批量插入數據，數據需要有自增 id。每次插入有一個唯一的 sessionId 來標記這些記錄，插入完成之後返回這個 sessionId。

方案

循環插入單條記錄，僞代碼：

int sessionId = dao.querySessionId();
for (Record record : recordList) {
    dao.insertRecord(record, sessionId);
}
return sessionId;

上述解決方案很簡明易懂，但是循環中每次 insert 操作都會與 DB 交互一次，當數據量很大時，會花費很多時間在網絡傳輸上，導致性能問題。

改進

問題可以通過批量插入來改善。

帶自增 id 的批量插入

Oracle 中比較常見的批量插入模版是：

INSERT ALL
   INTO target_table (col1, col2, col3) VALUES ('id1_1', 'val1_2', 'val1_3')
   INTO target_table (col1, col2, col3) VALUES ('id2_1', 'val2_2', 'val2_3')
   ...
Subquery;

但是每條記錄都需要有一個自增 id，Oracle 中一般用 Sequence 來實現，於是比較容易想到的是用下面這種方式：

INSERT ALL
   INTO table (col1, col2, col3) VALUES (SEQUENCE.NEXTVAL, 'val1_2', 'val1_3')
   INTO table (col1, col2, col3) VALUES (SEQUENCE.NEXTVAL, 'val2_2', 'val2_3')
   ...
SELECT 1 FROM DUAL;

不過遺憾的是，上述方案行不通，因爲所有的 SEQUENCE.NEXTVAL 都會是同一個值，會產生主鍵衝突。

接着嘗試其他的方案。Oracle 支持從一個表裏批量選取數據插入另一個表中：

INSERT INTO target_table (col1, col2, col3)
SELECT col1,
       col2,
       col3
FROM source_table
WHERE condition;

用上述方式，我們可以把被插入的數據用 UNION ALL 構造一個子表，也就是上面的 source_table 來實現批量插入。跟 INSERT ALL INTO 相比的好處是，可以使用 Sequence 的自增值：

INSERT INTO target_table (col1, col2, col3)
SELECT SEQUENCE.NEXTVAL,
       col2,
       col3
FROM (
    SELECT 'val1_2' col2, 'val1_3' col3 FROM dual UNION ALL
    SELECT 'val2_2' col2, 'val2_3' col3 FROM dual UNION ALL
    ...
);

用 MyBatis 的 dynamic sql 來實現大致如下：

<insert id="sampleInsert" parameterType="java.util.List">
    INSERT INTO target_table (col1, col2, col3)
    SELECT SEQUENCE.NEXTVAL,, col2, col3 FROM
    <foreach collection="list" item="item" index="index" open="(" close=")" separator=" UNION ALL ">
        SELECT #{item.val2} col2, #{item.val2} col3 FROM dual
    </foreach>
</insert>

插入完成之後返回 sessionId

在 Mybatis 中，返回某個 property 可以用 SelectKey。SelectKey 是 Insert 的子標籤，實現原理是在執行插入語句之前先做一次 SelectKey 的子查詢，此處，可以將子查詢的結果賦值到查詢的參數當中，例如

public class Foo {
    private int id;
    private String col2;
    private String col3;
}
public interface FooDao {
    void sampleInsert(Foo foo);
}

<insert id="sampleInsert" useGeneratedKeys="true" parameterType="Foo" keyProperty="id">
    <selectKey keyProperty="id" order="BEFORE" resultType="int">
        SELECT SESSION_SEQUENCE.NEXTVAL FROM DUAL
    </selectKey>
    INSERT INTO target_table (col1, col2, col3)
    VALUES (#{id}, #{col2}, #{col3})
</insert>

當插入結束之後，參數 foo.id 就會是 SESSION_SEQUENCE 的自增值（注意在 Dao 中不能用 @Param() 標註參數）。

然而這種方式只支持單條記錄的插入，Oracle 中批量插入的情況就無法完成賦值了。所以此處只能分成兩步來做：

1. 獲取 sessionId（SESSION_SEQUENCE.NEXTVAL）
2. 批量插入帶 sessionId 的記錄 或者
3. 用 SelectKey 選出 sessionId（SESSION_SEQUENCE.NEXTVAL）並批量插入記錄
4. 獲取 sessionId（SESSION_SEQUENCE.CURRVAL）

注意上述第二種方式，需要保證兩個方法在同一個 Transaction 裏面，否則 Sequence 的值會不一致。

總結

本文簡單總結了在 MyBatis 中往 Oracle 批量插入數據的方法，作一個快速筆記。

Reference

The Ultimate Guide to Oracle INSERT ALL Statement Oracle INSERT INTO SELECT Inserting Multiple Rows Using a Single Statement
Inserting multiple rows with sequence in Oracle