Mybatis实现批量新增的工具类(batch模式)

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");
}