Mybatis实现批量新增的工具类(batch模式)
前言:
Mybatis内置的ExecutorType有3种,默认的是simple,该模式下它为每个语句的执行创建一个新的预处理语句,单条提交sql;而batch模式重复使用已经预处理的语句,并且批量执行所有更新语句,显然batch性能将更优;
batch模式存在的问题:
在Insert操作时,在事务没有提交之前,是没有办法获取到自增的id,这在某型情形下是不符合业务要求的;
一、单数据源
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");
}