首页 > 软件编程 > java > MyBatis 批量新增工具类

MyBatis 批量新增工具类实现示例

2025-06-06 10:46:52 作者：Jamie Chyi

本文主要介绍了MyBatis批量操作工具类MyBatisBatchExecutorUtil,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

在实际开发中，批量操作数据库是提升效率的常见做法。MyBatis 提供了对批量操作的支持，但通常需要手动配置。本文将介绍如何使用 MyBatis 批量提交数据的工具类，实现更高效的数据库操作。

1. 工具类概述 🔧

MyBatisBatchExecutorUtil 是一个 MyBatis 批量操作工具类，通过 SqlSessionFactory 提供批量提交数据的功能。我们可以通过此类将数据按一定数量分组，避免一次性提交大量数据造成的内存压力。

2. 工具类代码 ✍️

package com.lili.office.utils;

import java.util.List;

import javax.annotation.PostConstruct;

import com.google.common.collect.Lists;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @author JamieChyi
 */
@Component
public class MyBatisBatchExecutorUtil {
    private static final Logger logger = LoggerFactory.getLogger(MyBatisBatchExecutorUtil.class);

    @Autowired
    private SqlSessionFactory sqlSessionFactory;
    private static MyBatisBatchExecutorUtil utils;

    @PostConstruct
    public void init() {
        utils = this;
        utils.sqlSessionFactory = this.sqlSessionFactory;
    }

    /**
     * 批量提交数据
     *
     * @param mapperClass     Mapper 类
     * @param mybatisSqlId    SQL 语句在 Mapper XML 文件中的 ID
     * @param dataList        要提交的数据列表
     */
    public static <T> void batchCommit(Class<?> mapperClass, String mybatisSqlId, List<T> dataList) {
        if (dataList == null || dataList.isEmpty()) {
            return;
        }

        SqlSession session = null;
        int commitCountEveryTime = 500;
        try {
            long startTime = System.currentTimeMillis();
            session = utils.sqlSessionFactory.openSession(ExecutorType.BATCH, false);
            List<List<T>> groupList = Lists.partition(dataList, commitCountEveryTime);
            for (List<T> tempList : groupList) {
                session.insert(mapperClass.getName() + "." + mybatisSqlId, tempList);
                session.commit();
                session.clearCache();
            }
            long endTime = System.currentTimeMillis();
            logger.info("批量插入数据耗时：" + (endTime - startTime) + "毫秒");
        } catch (Exception e) {
            logger.error("batchCommit error!", e);
            if (session != null) {
                session.rollback();
            }
        } finally {
            if (session != null) {
                session.close();
            }
        }
    }

    public static <T> void batchCommit(Class<?> mapperClass, List<T> dataList) {
        batchCommit(mapperClass, "insertPOBatch", dataList);
    }
}

解释：

@Autowired 注解通过 Spring 注入 SqlSessionFactory 实例。
@PostConstruct 用于在类初始化后执行初始化操作。
批量提交逻辑通过将数据分批处理来避免一次性提交过多数据。
ExecutorType.BATCH：使用批处理模式，提升性能。
分批提交：每次提交 500 条数据，避免单次提交数据过多导致内存占用过大。

3. 批量提交数据 🌟

方法一：指定 SQL ID 提交数据

public static <T> void batchCommit(Class<?> mapperClass, String mybatisSqlId, List<T> dataList) {
    // 处理逻辑
}

方法二：默认使用 insertPOBatch 提交数据

public static <T> void batchCommit(Class<?> mapperClass, List<T> dataList) {
    batchCommit(mapperClass, "insertPOBatch", dataList);
}

解释：

第一种方法允许我们指定 Mapper 文件中的 SQL ID，可以灵活适应不同场景。
第二种方法为常用的批量插入，使用默认的 SQL ID insertPOBatch。

4. 性能优化 ⏱️

批量提交数据能够显著提高性能，尤其是在处理大量数据时。通过将数据分批提交，可以避免一次性提交过多数据导致内存溢出的问题。每次提交的数据量可以根据具体需求调整，通常使用 500 条作为合适的批次大小。

性能测试：

long startTime = System.currentTimeMillis();
// 执行批量操作
long endTime = System.currentTimeMillis();
logger.info("批量插入数据耗时：" + (endTime - startTime) + "毫秒");

通过日志记录，开发者可以及时了解批量提交的执行时间，并进行性能调优。

5. 异常处理 ⚠️

在批量提交过程中，我们应该特别注意异常处理：

如果出现异常，应该执行回滚操作，确保数据一致性。
批量操作使用事务管理，可以在遇到问题时避免部分数据提交成功而其他数据失败的情况。

catch (Exception e) {
    logger.error("batchCommit error!", e);
    if (session != null) {
        session.rollback();
    }
}

总结 📚

通过 MyBatisBatchExecutorUtil 工具类，我们可以轻松实现 MyBatis 的批量提交操作，避免性能瓶颈和内存压力。合理的批量操作能够显著提升系统在处理大规模数据时的效率，是开发中必不可少的利器。

到此这篇关于MyBatis 批量新增工具类实现示例的文章就介绍到这了,更多相关MyBatis 批量新增工具类内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！