利用數據庫 LOAD DATA 特性實現數據批量寫入
引言
在開發應用程序時，經常會遇到需要向數據庫批量寫入大量數據的場景，如數據初始化、數據遷移或者從外部系統導入數據等。傳統的逐條插入方式在數據量較大時，性能會變得非常低下，因為每一次插入操作都需要與數據庫建立連接、進行 SQL 解析和執行等一系列開銷較大的操作。
MyBatis-Plus 作為一款優秀的 MyBatis 增強工具，為我們提供了多種數據操作方式，同時結合數據庫自身的特性，如LOAD DATA語句，可以極大地提升批量寫入數據的效率。
本文將詳細介紹如何使用 MyBatis-Plus 結合LOAD DATA特性來實現高效的數據批量寫入。
數據庫 LOAD DATA 特性介紹
LOAD DATA是 MySQL 等數據庫提供的一種高效的數據導入語句。它允許從一個文本文件中快速地將數據加載到數據庫表中。與傳統的 INSERT 語句相比，LOAD DATA具有以下優勢：
- 速度快：LOAD DATA是基于文件系統的操作，直接將文件中的數據批量加載到數據庫表中，避免了多次網絡傳輸和 SQL 解析的開銷。在處理大量數據時，速度比逐條 INSERT 語句快很多倍。
- 減少事務開銷：傳統的 INSERT 操作如果放在一個事務中，隨著數據量的增加，事務的持續時間會變長，占用數據庫資源。而LOAD DATA可以在短時間內完成大量數據的插入，減少了事務的持續時間，降低了鎖爭用的可能性。
例如，在 MySQL 中，LOAD DATA的基本語法如下：
LOAD DATA [LOCAL] INFILE 'file_name'
INTO TABLE table_name
[FIELDS TERMINATED BY 'field_separator' [OPTIONALLY] ENCLOSED BY 'enclosure_character']
[LINES TERMINATED BY 'line_separator']
[IGNORE number LINES]
[(column_list)]其中，file_name是包含要導入數據的文本文件路徑；table_name是目標數據庫表；FIELDS TERMINATED BY指定字段之間的分隔符；LINES TERMINATED BY指定行之間的分隔符；IGNORE number LINES表示忽略文件開頭的指定行數；(column_list)指定要導入數據對應的表列。
實現
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper
PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.example.demo.mapper.UserMapper">
<insert id="batchInsertByLoadData">
LOAD DATA LOCAL INFILE '/path/to/your/data/file.txt'
INTO TABLE user
FIELDS TERMINATED BY ','
LINES TERMINATED BY '\n'
(id, name, age)
</insert>
</mapper>補充
MyBatis-Plus 的saveBatch方法是其提供的一種便捷的批量插入方式，內部會根據配置的批量插入大小（默認為 1000）將數據拆分成多個批次進行插入操作。
性能對比
- LOAD DATA：如前所述，它基于文件系統操作，直接將文件數據批量加載到數據庫表，避免了多次網絡傳輸和 SQL 解析，在處理大量數據（如數十萬條甚至更多）時，性能優勢極其明顯，速度比逐條插入快數倍甚至數十倍。
- saveBatch：雖然它也實現了批量插入，通過將數據分批次發送到數據庫減少了連接次數，但每一批次的數據仍需進行 SQL 解析和執行等操作。在數據量較大時，相比LOAD DATA，性能會有較大差距。例如，當插入 10 萬條數據時，LOAD DATA可能只需要幾秒鐘，而saveBatch可能需要幾十秒甚至更長時間，具體時間取決于數據庫配置、服務器性能等因素。
適用場景
- LOAD DATA：適用于大規模數據的一次性導入，比如數據初始化、從外部系統進行全量數據遷移等場景。由于其對數據格式要求較為嚴格，且操作相對復雜，不太適合少量數據的頻繁插入。
- saveBatch：適用于數據量相對較小（如幾千條以內）或者對數據格式和操作靈活性要求較高的場景。例如，在日常業務中，當需要批量插入一些用戶操作記錄等少量數據時，saveBatch使用起來更加方便快捷。
最后
通過 MyBatis-Plus 結合數據庫的LOAD DATA特性以及使用其自帶的saveBatch方法，都可以實現數據的批量寫入，但兩者各有優劣。在實際應用中，開發者需要根據具體的業務場景、數據量大小、代碼復雜度要求以及事務控制等因素，綜合考慮選擇合適的方法。如果是處理大規模數據的一次性導入，LOAD DATA無疑是更好的選擇，能顯著提升性能；而對于少量數據的頻繁插入或者對開發效率要求較高的場景，saveBatch則更為合適。希望本文能為讀者在解決數據批量寫入問題時提供全面的參考和幫助，以便根據實際需求做出最優決策。
