前言
- 最近接了一个需求要把项目里的批量新增改掉,以前批量新增都是用Mybatis的foreach拼接的,生产环境由于list数据太多导致拼接的sql过长导致生产事故。所以要把项目里的foreach拼接的批量新增都改掉。
- 由于项目批量新增的地方很多,对应不同的Mapper、不同的表和不同的实体类。一个一个的改太麻烦,最后写了一个通用的工具类,每次新增的时候只用准备好要新增的实体类集合,mapper和单条新增的方法,然后调用这个工具类就可以了。
代码实现
一、单数据源
1.批量新增的工具类(BatchInsertUtil)
package com.lh.utils;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import java.util.List;
import java.util.function.BiConsumer;
@Slf4j
public class BatchInsertUtil {
public static <M, T> void batchInsert(List<T> list, Class<M> clazz, BiConsumer<M, T> biConsumer) {
if (list == null || list.size() == 0) {
log.info("BatchInsertUtil batchInsert list data is null!");
return;
}
SqlSessionFactory sqlSessionFactory = SpringUtil.getBean(SqlSessionFactory.class);
SqlSession session = sqlSessionFactory.openSession(ExecutorType.BATCH);
try {
M mapper = session.getMapper(clazz);
list.forEach(a -> {
biConsumer.accept(mapper, a);
});
session.commit();
session.clearCache();
} catch (Exception e) {
e.printStackTrace();
log.error("BatchInsertUtil batchInsert is exception!clazz={}", clazz.getName(), e);
session.rollback();
} finally {
session.close();
}
}
}
2.调用演示
public void testInsert() {
AtomicInteger atomic = new AtomicInteger(10);
List<CustDetail> list = Arrays.asList(
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(