文章目录
前言
- 最近接了一个需求要把项目里的批量新增改掉,以前批量新增都是用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;
/**
* Mybatis 批量新增工具类
*/
@Slf4j
public class BatchInsertUtil {
/**
* 批量新增方法
*
* @param list 要新增的集合
* @param clazz Mapper类
* @param biConsumer 对应的单条新增方法
* @param <M> mapper类型
* @param <T> 结合元素类型
*/
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(), "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(), "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(), "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(), "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(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get())
);
BatchInsertUtil.batchInsert(list, CustDetailMapper.class, CustDetailMapper::insert);
}
3.获取SpringBean的工具类(SpringUtil )
package com.lh.utils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
@Slf4j
@Component
public class SpringUtil implements ApplicationContextAware {
private static ApplicationContext applicationContext;
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
if (SpringUtil.applicationContext == null) {
SpringUtil.applicationContext = applicationContext;
}
}
//获取applicationContext
public static ApplicationContext getApplicationContext() {
return applicationContext;
}
//通过name获取 Bean.
public static Object getBean(String name) {
return getApplicationContext().getBean(name);
}
//通过class获取Bean.
public static <T> T getBean(Class<T> clazz) {
return getApplicationContext().getBean(clazz);
}
//通过name,以及Clazz返回指定的Bean
public static <T> T getBean(String name, Class<T> clazz) {
return getApplicationContext().getBean(name, clazz);
}
}
4.Mybatis的Mapper(CustDetailMapper )
public interface CustDetailMapper {
int insert(CustDetail record);
}
5.Mapper的xml文件
<?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.lh.dao.sys.CustDetailMapper">
<insert id="insert" parameterType="com.lh.dawn.entity.ring.CustDetail">
insert into cust_detail (id, cust_id, city,
name, tel, age, input_time,
update_time)
values (#{id,jdbcType=INTEGER}, #{custId,jdbcType=VARCHAR}, #{city,jdbcType=VARCHAR},
#{name,jdbcType=VARCHAR}, #{tel,jdbcType=VARCHAR}, #{age,jdbcType=INTEGER}, #{inputTime,jdbcType=TIMESTAMP},
#{updateTime,jdbcType=TIMESTAMP})
</insert>
</mapper>
二、多数据源
在平时的项目中很多时候我们会接多个数据库,这个时候会有多个数据源的情况。此时我们就要注意对应的mapper属于哪个数据源。把方法稍作改造就可以了。
1.找到对应的SqlSessionFactory
如下:CustDetailMapper 所对应的SqlSessionFactory名称为ringSqlSessionFactory
2.新增一个批量新增的方法
/**
* 多数据源批量新增方法
*
* @param list 要新增的集合
* @param clazz Mapper类
* @param biConsumer 对应的单条新增方法
* @param sqlSessionFactoryName sqlSessionFactory名字
* @param <M> mapper类型
* @param <T> 结合元素类型
*/
public static <M, T> void batchInsert(List<T> list, Class<M> clazz, BiConsumer<M, T> biConsumer, String sqlSessionFactoryName) {
if (list == null || list.size() == 0) {
log.info("BatchInsertUtil batchInsert list data is null!");
return;
}
SqlSessionFactory sqlSessionFactory = SpringUtil.getBean(sqlSessionFactoryName, 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();
}
}
3.调用示例
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(), "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(), "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(), "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(), "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(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get()),
new CustDetail("202007130000" + atomic.incrementAndGet(), "小麻花" + atomic.get(), "18729872910" + atomic.get())
);
BatchInsertUtil.batchInsert(list, CustDetailMapper.class, CustDetailMapper::insert,"ringSqlSessionFactory");
}
三、总结
其实这个工具类是利用了java8的新特性函数式接口,函数式接口可以让我们把一个方法或者说是一段代码逻辑当做参数传递。巧用这些特性可以提高我们代码的可复用性。