Mybatis实现批量新增的工具类

前言

  • 最近接了一个需求要把项目里的批量新增改掉,以前批量新增都是用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的新特性函数式接口,函数式接口可以让我们把一个方法或者说是一段代码逻辑当做参数传递。巧用这些特性可以提高我们代码的可复用性。

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值