MyBatis批量插入優(yōu)化的方法步驟

更新時(shí)間：2026年02月11日 08:21:29 作者：花寶寶hua

本文主要介紹了MyBatis批量插入優(yōu)化的方法步驟,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

最初的代碼（5分鐘）

最開始寫的很簡單，foreach循環(huán)插入：

// 方式1：循環(huán)單條插入（最慢）
for (User user : userList) {
    userMapper.insert(user);
}

10萬條數(shù)據(jù)，每條都要走一次網(wǎng)絡(luò)請(qǐng)求、一次SQL解析、一次事務(wù)提交。

算一下：假設(shè)每條插入需要3ms，10萬條就是300秒 = 5分鐘。

這是最蠢的寫法，但我見過很多項(xiàng)目都這么寫。

第一次優(yōu)化：批量SQL（30秒）

把循環(huán)插入改成批量SQL：

<!-- Mapper.xml -->
<insert id="batchInsert">
    INSERT INTO user (name, age, email) VALUES
    <foreach collection="list" item="item" separator=",">
        (#{item.name}, #{item.age}, #{item.email})
    </foreach>
</insert>

// 分批插入，每批1000條
int batchSize = 1000;
for (int i = 0; i < userList.size(); i += batchSize) {
    int end = Math.min(i + batchSize, userList.size());
    List<User> batch = userList.subList(i, end);
    userMapper.batchInsert(batch);
}

從5分鐘降到30秒，提升10倍。

原理：一條SQL插入多條數(shù)據(jù)，減少網(wǎng)絡(luò)往返次數(shù)。

但還有問題：30秒還是太慢。

第二次優(yōu)化：JDBC批處理（8秒）

MySQL有個(gè)參數(shù)叫rewriteBatchedStatements，開啟后可以把多條INSERT合并成一條。

第一步：修改數(shù)據(jù)庫連接URL

jdbc:mysql://localhost:3306/test?rewriteBatchedStatements=true

第二步：使用MyBatis的批處理模式

@Autowired
private SqlSessionFactory sqlSessionFactory;

public void batchInsertWithExecutor(List<User> userList) {
    try (SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH)) {
        UserMapper mapper = sqlSession.getMapper(UserMapper.class);
        
        int batchSize = 1000;
        for (int i = 0; i < userList.size(); i++) {
            mapper.insert(userList.get(i));
            
            if ((i + 1) % batchSize == 0) {
                sqlSession.flushStatements();
                sqlSession.clearCache();
            }
        }
        sqlSession.flushStatements();
        sqlSession.commit();
    }
}

從30秒降到8秒。

原理：ExecutorType.BATCH模式下，MyBatis會(huì)緩存SQL，最后一次性發(fā)送給數(shù)據(jù)庫執(zhí)行。配合rewriteBatchedStatements=true，MySQL驅(qū)動(dòng)會(huì)把多條INSERT合并。

第三次優(yōu)化：多線程并行（3秒）

8秒還是不夠快，上多線程：

public void parallelBatchInsert(List<User> userList) {
    int threadCount = 4;  // 根據(jù)數(shù)據(jù)庫連接池大小調(diào)整
    int batchSize = userList.size() / threadCount;
    
    ExecutorService executor = Executors.newFixedThreadPool(threadCount);
    List<Future<?>> futures = new ArrayList<>();
    
    for (int i = 0; i < threadCount; i++) {
        int start = i * batchSize;
        int end = (i == threadCount - 1) ? userList.size() : (i + 1) * batchSize;
        List<User> subList = userList.subList(start, end);
        
        futures.add(executor.submit(() -> {
            batchInsertWithExecutor(subList);
        }));
    }
    
    // 等待所有任務(wù)完成
    for (Future<?> future : futures) {
        try {
            future.get();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
    
    executor.shutdown();
}

從8秒降到3秒。

注意事項(xiàng)：

線程數(shù)不要超過數(shù)據(jù)庫連接池大小
如果需要事務(wù)一致性，這個(gè)方案不適用
要考慮主鍵沖突的問題

優(yōu)化效果對(duì)比

方案	耗時(shí)	提升倍數(shù)
循環(huán)單條插入	300秒	基準(zhǔn)
批量SQL	30秒	10倍
JDBC批處理	8秒	37倍
多線程并行	3秒	100倍

踩過的坑

坑1：foreach拼接SQL過長

<foreach collection="list" item="item" separator=",">

如果一次插入太多條，SQL會(huì)非常長，可能超過max_allowed_packet限制。

解決：分批插入，每批500-1000條。

坑2：rewriteBatchedStatements不生效

檢查幾個(gè)點(diǎn)：

URL參數(shù)是否正確：rewriteBatchedStatements=true
是否使用了ExecutorType.BATCH
MySQL驅(qū)動(dòng)版本是否太舊

坑3：自增主鍵返回問題

批量插入時(shí)想獲取自增主鍵：

<insert id="batchInsert" useGeneratedKeys="true" keyProperty="id">

注意：rewriteBatchedStatements=true時(shí)，自增主鍵返回可能有問題，需要升級(jí)MySQL驅(qū)動(dòng)到8.0.17+。

坑4：內(nèi)存溢出

10萬條數(shù)據(jù)一次性加載到內(nèi)存，可能OOM。

解決：分頁讀取 + 分批插入。

int pageSize = 10000;
int total = countTotal();
for (int i = 0; i < total; i += pageSize) {
    List<User> page = selectByPage(i, pageSize);
    batchInsertWithExecutor(page);
}

最終方案代碼

@Service
public class BatchInsertService {
    
    @Autowired
    private SqlSessionFactory sqlSessionFactory;
    
    /**
     * 高性能批量插入
     * 10萬條數(shù)據(jù)約3秒
     */
    public void highPerformanceBatchInsert(List<User> userList) {
        if (userList == null || userList.isEmpty()) {
            return;
        }
        
        int threadCount = Math.min(4, Runtime.getRuntime().availableProcessors());
        int batchSize = (int) Math.ceil((double) userList.size() / threadCount);
        
        ExecutorService executor = Executors.newFixedThreadPool(threadCount);
        CountDownLatch latch = new CountDownLatch(threadCount);
        
        for (int i = 0; i < threadCount; i++) {
            int start = i * batchSize;
            int end = Math.min((i + 1) * batchSize, userList.size());
            
            if (start >= userList.size()) {
                latch.countDown();
                continue;
            }
            
            List<User> subList = new ArrayList<>(userList.subList(start, end));
            
            executor.submit(() -> {
                try {
                    doBatchInsert(subList);
                } finally {
                    latch.countDown();
                }
            });
        }
        
        try {
            latch.await();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        
        executor.shutdown();
    }
    
    private void doBatchInsert(List<User> userList) {
        try (SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false)) {
            UserMapper mapper = sqlSession.getMapper(UserMapper.class);
            
            for (int i = 0; i < userList.size(); i++) {
                mapper.insert(userList.get(i));
                
                if ((i + 1) % 1000 == 0) {
                    sqlSession.flushStatements();
                    sqlSession.clearCache();
                }
            }
            
            sqlSession.flushStatements();
            sqlSession.commit();
        }
    }
}