java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > mybatisPlus批量插入

mybatisPlus批量插入优化加快性能

作者:大飞哥~BigFei

这篇文章主要介绍了mybatisPlus批量插入优化加快性能,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下

1.背景

​ 由于之前一个同事问我mybatisPlus的IService层的saveBatch(Collection entityList)方法批量插入速度实在是太慢了,能否优化下?我过去跟他看了下具体的需求是这样的,一张业务表中有30多万的业务数据,表里的一个字段是一个json的数据字段,要把30多万的数据查出来针对这个json字段解析之后存入另外一张表中,后面他使用了mybatisPlus的IService层的saveBatch发现好慢好慢,,去重是把分页查询出来插入数据的最后一条数据的id存入redis中,每次分页查询数据排序之后id比上一次处理的最后一条数据的id大的数据,如果比该条数据的id小则忽略,后面我给他看了下使用saveBatch插入一批数据1000条大概要好几十分钟的,之前他使用的for循环里面insert每一条处理好的数据,这种是真的非常的慢,慢到你怀疑人生,这个insert方法我在下面的demo中准备数据,往一个表里插入30万条数据的时候使用了下实在是太慢了,后面我使用了一个mybatisPlus的sql注入器的一个批量的方法:

Integer insertBatchSomeColumn(Collection<UserEntity> entityList);

​ 使用了该方法之后,在一个测试方法中使用主线程插入模拟的30万数据到一个user的表中只用了:23.209105秒,然后使用多线程异步插入用了:12.1030808秒,这个时间跟机器的性能和网路有关,所以每一次执行这个时间会有所不同的,由于我们使用的数据库是云数据库,所以插入需要走网络,没有使用本地的安装的mysql数据库,使用这个sql批量插入的注入器给那个同事优化了一波之后,他原来使用insertr处理要10个小时后面使用saveBatch也要在2个小时以上,后面优化之后大概估计只要20分钟不到就完了,但是我后面使用这个demo用30万数据,解析一个json字段入库到一张表中是非常快的,就花了10多秒~30多秒的时间就干完了,所以说他那个需求还有优化的空间,后面数据全部跑完之后就不用了,不是我搞所以就能用就行,比原来的效率也是提升了一大截。

2.方案

​ 查询可以使用mybatis的原生sql查询,需要写一个批量插入的方法insert标签中使用foreach遍历一个list然后插入逐条插入数据即可,但是这种需要写大量的代码,还要处理每个插入的字段为空的判断处理,不然遇到插入字段为null就会报错,让人很头疼,所以还是使用mybatisPlus的sql批量注入器基本不需要写啥代码即可实现批量插入操作,而且性能和效率还是杠杆的。

​ mybatisPlus的IService层的saveBatch为啥慢?看了下它的源码发现底层封装还是比较深,本质上底层还是一条一条的去插入数据,所以才会慢。

​ mybatisPlus的sql注入器有一下几个可以给我们扩展使用的类:

图片

​ 项目中批量sql注入器的使用如下:

​ 新增一个 MySqlInjector类:

package com.zlf.sb.demo.sqlinjector;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import java.util.List;
public class MySqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new InsertBatchSomeColumn());//添加批量插入方法
        return methodList;
    }
}

​ 将MySqlInjector交给Spring容器管理:新增一个MybatisPlusConfig类如下:

package com.zlf.sb.demo.config;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.extension.plugins.pagination.optimize.JsqlParserCountOptimize;
import com.zlf.sb.demo.sqlinjector.MySqlInjector;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
@Configuration
public class MybatisPlusConfig {
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
        // 设置请求的页面大于最大页后操作, true调回到首页,false 继续请求  默认false
        // paginationInterceptor.setOverflow(false);
        // 设置最大单页限制数量,默认 500 条,-1 不受限制
        paginationInterceptor.setLimit(-1); // 这里也是一个大坑,需要注意的
        // 开启 count 的 join 优化,只针对部分 left join
        paginationInterceptor.setCountSqlParser(new JsqlParserCountOptimize(true));
        return paginationInterceptor;
    }
    @Bean
    @Primary//批量插入配置
    public MySqlInjector mySqlInjector() {
        return new MySqlInjector();
    }
}

​ UserMapper接口中加入insertBatchSomeColumn方法:

package com.zlf.sb.demo.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.zlf.sb.demo.entity.UserEntity;
import org.apache.ibatis.annotations.Options;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.cursor.Cursor;
import org.apache.ibatis.mapping.ResultSetType;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Repository;
import java.util.Collection;
@Scope("prototype")
@Repository
public interface UserMapper extends BaseMapper<UserEntity> {
    /**
     * 批量插入 仅适用于mysql
     *
     * @param entityList 实体列表
     * @return 影响行数
     */
    Integer insertBatchSomeColumn(Collection<UserEntity> entityList);
    /**
     * 统计总数
     *
     * @return
     */
    @Select("SELECT count(*) FROM user")
    Integer countTotal();
    /**
     * 分页游标查询
     *
     * @return
     */
    @Options(resultSetType = ResultSetType.FORWARD_ONLY)
    @Select("SELECT * FROM user ORDER BY create_time DESC LIMIT #{start}, #{pageSize} ")
    Cursor<UserEntity> getCursorPageData(@Param("start") Integer start, @Param("pageSize") Integer pageSize);
}

​ 准备了两张表:user,user_json_data,user表中有个json_data的字段是一个存json格式数据的字段,然后准备30万的数据,向user表中插入30万条构造的数据,SpringBootDemoApplicationTests中代码如下:

//@Test
    public void addUserDataTest() {
        StopWatch stopWatch = new StopWatch();
        // 开始时间
        stopWatch.start();
        int totalData = 300000;
        int splitSize = 10000;
        int splitPages = (totalData + splitSize - 1) / splitSize;
        /*for (int i = 1; i <= splitPages; i++) {
            //起始条数
            int firstIndex = (i - 1) * splitSize + 1;
            //截止条数
            int lastIndex = i * splitSize;
            List<UserEntity> userEntityList = new ArrayList<>();
            for (int j = firstIndex; j <= lastIndex; j++) {
                UserEntity user = new UserEntity();
                user.setName("zlf" + j);
                user.setAge(j);
                JSONObject jo = new JSONObject();
                try {
                    jo.put("name", "zlf" + j);
                    jo.put("age", j);
                } catch (JSONException e) {
                    throw new RuntimeException(e);
                }
                //log.info("json:{}",j.toString());
                user.setJsonData(jo.toString());
                //userMapper.insert(user);
                userEntityList.add(user);
            }
            //userService.saveBatch(userEntityList);
            userMapper.insertBatchSomeColumn(userEntityList);
        }
        // 结束时间
        stopWatch.stop();
        log.info("执行时长:{}秒", stopWatch.getTotalTimeSeconds());*/
        //执行时长:23.209105秒
        // 异步
        ThreadPoolExecutor executor = ThreadPoolService.getInstance();
        AtomicInteger count = new AtomicInteger(0);
        CountDownLatch countDownLatch = new CountDownLatch(splitPages);
        for (int i = 1; i <= splitPages; i++) {
            int finalI = i;
            CompletableFuture.runAsync(() -> {
                //起始条数
                int firstIndex = (finalI - 1) * splitSize + 1;
                //截止条数
                int lastIndex = finalI * splitSize;
                List<UserEntity> userEntityList = new ArrayList<>();
                for (int j = firstIndex; j <= lastIndex; j++) {
                    UserEntity user = new UserEntity();
                    user.setName("zlf" + j);
                    user.setAge(j);
                    JSONObject jo = new JSONObject();
                    try {
                        jo.put("name", "zlf" + j);
                        jo.put("age", j);
                    } catch (JSONException e) {
                        throw new RuntimeException(e);
                    }
                    //log.info("json:{}",j.toString());
                    user.setJsonData(jo.toString());
                    //userMapper.insert(user);
                    userEntityList.add(user);
                }
                count.getAndAdd(userEntityList.size());
                userMapper.insertBatchSomeColumn(userEntityList);
                //userService.saveBatch(userEntityList);
                countDownLatch.countDown();
            }, executor);
        }
        try {
            countDownLatch.await(); //保证之前的所有的线程都执行完成,才会走下面的;
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        // 结束时间
        stopWatch.stop();
        log.info("total:{},执行时长:{}秒", count.get(), stopWatch.getTotalTimeSeconds());
        // total:300000,执行时长:12.1030808秒
    }

2.1 多线程分页查询 、 生产者消费者模型、多线程sql注入器批量插入

​ 测试用例的方法:

    @Test
    public void queryInsertDataTest() {
        userService.queryInsertData();
    }

​ UserServiceImpl类中的方法:

/**
     * 多线程分页查询
     * 生产者+消费者模型
     * 多线程插入
     */
    @Override
    public void queryInsertData() {
        StopWatch stopWatch = new StopWatch();
        // 开始时间
        stopWatch.start();
        Page<UserEntity> page = new Page<>(1, 10);
        QueryWrapper<UserEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.lambda().orderByDesc(UserEntity::getCreateTime);
        Page<UserEntity> pageResult = this.getBaseMapper().selectPage(page, queryWrapper);
        int pageSize = 10000;
        if (Objects.nonNull(pageResult)) {
            long total = pageResult.getTotal();
            int pages = (int) ((total + pageSize - 1) / pageSize);
            log.info("pages:{}", pages);
            ThreadPoolExecutor executor = ThreadPoolService.getInstance();
            CountDownLatch countDownLatch1 = new CountDownLatch(pages);
            CountDownLatch countDownLatch2 = new CountDownLatch(pages);
            LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue = userJsonDataService.getLinkedBlockingQueue();
            for (int i = 1; i <= pages; i++) {
                try {
                    QueryDoBizProducer queryDoBizProducer = new QueryDoBizProducer(linkedBlockingQueue, i, queryCount, countDownLatch1, pageSize);
                    CompletableFuture.runAsync(queryDoBizProducer, executor);
                } catch (Exception e) {
                    log.info("异常1:{}", e.getMessage());
                }
            }
            try {
                countDownLatch1.await();
            } catch (InterruptedException e) {
                log.info("异常2:{}", e.getMessage());
            }
            for (int i = 1; i <= pages; i++) {
                try {
                    InsertDataConsumer insertDataConsumer = new InsertDataConsumer(linkedBlockingQueue, insertCount, countDownLatch2);
                    CompletableFuture.runAsync(insertDataConsumer, executor);
                } catch (Exception e) {
                    log.info("异常3:{}", e.getMessage());
                }
            }
            try {
                countDownLatch2.await();
            } catch (InterruptedException e) {
                log.info("异常4:{}", e.getMessage());
            }
            log.info("queryCount:{}", queryCount);
            log.info("insertCount:{}", insertCount);
        }
        // 结束时间
        stopWatch.stop();
        log.info("执行时长:{}秒", stopWatch.getTotalTimeSeconds());
        // 执行时长:18.903922秒
    }

​ 自定义单例线程池ThreadPoolService:

package com.zlf.sb.demo.service;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
/**
 * @author zlf
 * @description:
 * @time: 2022/7/13 10:18
 */
public class ThreadPoolService {
    /**
     * CPU 密集型:核心线程数 = CPU核数 + 1
     * <p>
     * IO 密集型:核心线程数 = CPU核数 * 2
     * <p>
     * 注意:IO密集型 (某大厂实战经验)
     * 核心线程数 = CPU核数 / (1 - 阻塞系数)
     * 例如阻塞系数为0.8 ,CPU核数为 4 ,则核心线程数为 20
     */
    private static final int DEFAULT_CORE_SIZE = Runtime.getRuntime().availableProcessors() * 2;
    private static final int MAX_QUEUE_SIZE = 100;
    private static final int QUEUE_INIT_MAX_SIZE = 200;
    private volatile static ThreadPoolExecutor executor;
    private ThreadPoolService() {
    }
    // 获取单例的线程池对象
    public static ThreadPoolExecutor getInstance() {
        if (executor == null) {
            synchronized (ThreadPoolService.class) {
                if (executor == null) {
                    executor = new ThreadPoolExecutor(DEFAULT_CORE_SIZE,// 核心线程数
                            MAX_QUEUE_SIZE, // 最大线程数
                            Integer.MAX_VALUE, // 闲置线程存活时间
                            TimeUnit.MILLISECONDS,// 时间单位
                            new LinkedBlockingDeque<Runnable>(QUEUE_INIT_MAX_SIZE),// 线程队列
                            Executors.defaultThreadFactory()// 线程工厂
                    );
                }
            }
        }
        return executor;
    }
    public void execute(Runnable runnable) {
        if (runnable == null) {
            return;
        }
        executor.execute(runnable);
    }
    // 从线程队列中移除对象
    public void cancel(Runnable runnable) {
        if (executor != null) {
            executor.getQueue().remove(runnable);
        }
    }
}

​ 生产者代码:

package com.zlf.sb.demo.producer;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.TypeReference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zlf.sb.demo.entity.UserEntity;
import com.zlf.sb.demo.entity.UserJsonDataEntity;
import com.zlf.sb.demo.mapper.UserMapper;
import com.zlf.sb.demo.util.SpringUtils;
import lombok.extern.slf4j.Slf4j;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.atomic.AtomicInteger;
@Slf4j
public class QueryDoBizProducer implements Runnable {
    private LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue;
    private AtomicInteger count;
    private UserMapper mapper;
    private long current;
    private int pageSize;
    private CountDownLatch countDownLatch;
    public QueryDoBizProducer(LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue, long current, AtomicInteger count, CountDownLatch countDownLatch, int pageSize) throws Exception {
        this.linkedBlockingQueue = linkedBlockingQueue;
        this.mapper = SpringUtils.getBean(UserMapper.class);
        this.current = current;
        this.count = count;
        this.countDownLatch = countDownLatch;
        this.pageSize = pageSize;
    }
    @Override
    public void run() {
        try {
            Page<UserEntity> page2 = new Page<>(current, pageSize);
            QueryWrapper<UserEntity> queryWrapper2 = new QueryWrapper<>();
            queryWrapper2.lambda().orderByDesc(UserEntity::getCreateTime);
            Page<UserEntity> pageData = mapper.selectPage(page2, queryWrapper2);
            if (Objects.nonNull(pageData)) {
                List<UserEntity> records = pageData.getRecords();
                List<UserJsonDataEntity> list = new ArrayList<>();
                for (UserEntity rs : records) {
                    UserJsonDataEntity jd = JSONObject.parseObject(rs.getJsonData(), new TypeReference<UserJsonDataEntity>() {
                    });
                    list.add(jd);
                }
                linkedBlockingQueue.put(list);
                count.getAndAdd(list.size());
                log.info("生产者查询数据放入队列完成---->ThreadName:{}", Thread.currentThread().getName());
            }
            countDownLatch.countDown();
        } catch (Exception e) {
            e.printStackTrace();
            log.error("加入队列异常:{}", e.getMessage());
        }
    }
}

​ 消费者代码:

package com.zlf.sb.demo.consumer;
import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.fastjson.JSON;
import com.zlf.sb.demo.entity.UserJsonDataEntity;
import com.zlf.sb.demo.mapper.UserJsonDataMapper;
import com.zlf.sb.demo.util.SpringUtils;
import lombok.extern.slf4j.Slf4j;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.atomic.AtomicInteger;
@Slf4j
public class InsertDataConsumer implements Runnable {
    private LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue;
    private AtomicInteger count;
    private UserJsonDataMapper mapper;
    private CountDownLatch countDownLatch;
    public InsertDataConsumer(LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue, AtomicInteger count, CountDownLatch countDownLatch) {
        this.linkedBlockingQueue = linkedBlockingQueue;
        this.count = count;
        this.mapper = SpringUtils.getBean(UserJsonDataMapper.class);
        this.countDownLatch = countDownLatch;
    }
    @Override
    public void run() {
        if (CollectionUtil.isNotEmpty(linkedBlockingQueue)) {
            try {
                List<UserJsonDataEntity> dataList = linkedBlockingQueue.take();
                if (CollectionUtil.isNotEmpty(dataList)) {
                    log.info("dataList:{}", JSON.toJSONString(dataList));
                    mapper.insertBatchSomeColumn(dataList);
                    log.info("消费者插入数据完成---->ThreadName:{}", Thread.currentThread().getName());
                    count.getAndAdd(dataList.size());
                }
                countDownLatch.countDown();
            } catch (Exception e) {
                e.printStackTrace();
                log.error("消费者插入数据:{}", e.getMessage());
            }
        }
    }
}

​ 阻塞队列在UserJsonDataServiceImpl类中:

package com.zlf.sb.demo.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zlf.sb.demo.entity.UserJsonDataEntity;
import com.zlf.sb.demo.mapper.UserJsonDataMapper;
import com.zlf.sb.demo.service.IUserJsonDataService;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.concurrent.LinkedBlockingQueue;
@Data
@Service
@Slf4j
public class UserJsonDataServiceImpl extends ServiceImpl<UserJsonDataMapper, UserJsonDataEntity> implements IUserJsonDataService {
    private LinkedBlockingQueue<List<UserJsonDataEntity>> linkedBlockingQueue = new LinkedBlockingQueue<>();
}

​ 这种方式如果数据量太大了阻塞队列中的存储了太多的数据,有可能OOM,30万数据本地跑一点压力都没有,轻轻松松就搞定了,但是数据量太大就需要JVM调下优了,这种方式不用管是spring的事务管理,如果使用mysql数据spring的事务传播级别是:ISOLATION_REPEATABLE_READ,可重复读:可以防止脏读和不可重复读,但是幻读仍可能发生,如果使用oracle数据库spring的事务传播级别是:ISOLATION_READ_COMMITTED,读已提交,可防止脏读,幻读和不可重复读仍可能发生,所以事务默认即可。

2.2 游标查询sql注入器批量插入

​ 测试用例:

    @Test
    public void cursorQueryInsertData() {
        userService.cursorQueryInsertData();
    }

​ UserServiceImpl类中的方法:

/**
     * 游标查询
     */
    @Override
    public void cursorQueryInsertData() {
        StopWatch stopWatch = new StopWatch();
        // 开始时间
        stopWatch.start();
        int total = userMapper.countTotal();
        if (total > 0) {
            int pageSize = 10000;
            int pages = (total + pageSize - 1) / pageSize;
            for (int i = 1; i <= pages; i++) {
                List<UserJsonDataEntity> userJsonDataEntities = new ArrayList<>();
                int start = (i - 1) * pageSize;
                try (SqlSession sqlSession = sqlSessionFactory.openSession(); Cursor<UserEntity> pageCursor = sqlSession.getMapper(UserMapper.class).getCursorPageData(start, pageSize)) {
                    int currentIndex = pageCursor.getCurrentIndex();
                    log.info("currentIndex1:{}", currentIndex);
                    Iterator<UserEntity> iterator = pageCursor.iterator();
                    while (iterator.hasNext()) {
                        UserEntity userEntity = iterator.next();
                        // log.info("userEntity:{}", JSON.toJSONString(userEntity));
                        UserJsonDataEntity jd = com.alibaba.fastjson.JSONObject.parseObject(userEntity.getJsonData(), new TypeReference<UserJsonDataEntity>() {
                        });
                        userJsonDataEntities.add(jd);
                    }
                } catch (Exception e) {
                    log.error("游标分页查询异常:{}", e.getMessage());
                }
                log.info("userJsonDataEntities:{}", userJsonDataEntities.size());
                userJsonDataMapper.insertBatchSomeColumn(userJsonDataEntities);
            }
        }
        // 结束时间
        stopWatch.stop();
        log.info("执行时长:{}秒", stopWatch.getTotalTimeSeconds());
        //执行时长:39.5655331秒
    }

2.3 多线程分页查询 、 生产者消费者模型、多线程往ES中按时间维度划分的索引中写入数据

​ 查询user表中的数据可以按照时间段查询,可以根据年、月查询,然后多线程查询出一批数据之后根据原始数据创建一个ES的索引,将各个时间段的数据存放在各自的索引中,这样根据时间段划分数据到对应的索引上,在多线程往ES各自时间段的索引中写入数据的时候就可以并发并行的去操作各自的索引,而不是去并发的去操作一个索引,这是需要特别注意的一点,使用ES要特别注意:有事务场景下不要使用ES,要控制好并发,不要并发的去操作一个索引,这种一定会翻车的。将数据写入到ES各自的索引后,可以根据时间段去从各自的索引中查数据,这种极大的提高了效率和性能,有兴趣的可以去尝试下这个方案。

2.4 查询数据量较小的情况采用List分片的方法

​ 查询数据量小的情况下可以将查询结果的List分片多线程处理每一个分片的数据

2.4.1 使用 Google 的 Guava 框架实现分片

<dependency>
  <groupId>com.google.guava</groupId>
  <artifactId>guava</artifactId>
  <version>31.0.1-jre</version>
</dependency>
import com.google.common.collect.Lists;
import java.util.Arrays;
import java.util.List;
/**
 * Guava 分片
 */
public class PartitionByGuavaExample {
    // 原集合
    private static final List<String> OLD_LIST = Arrays.asList(
            "唐僧,悟空,八戒,沙僧,曹操,刘备,孙权".split(","));
    public static void main(String[] args) {
        // 集合分片
        List<List<String>> newList = Lists.partition(OLD_LIST, 3);
        // 打印分片集合
        newList.forEach(i -> {
            System.out.println("集合长度:" + i.size());
        });
    }
}

2.4.2 使用 Apache 的 commons 框架实现分片

<dependency>
  <groupId>org.apache.commons</groupId>
  <artifactId>commons-collections4</artifactId>
  <version>4.4</version>
</dependency>
import org.apache.commons.collections4.ListUtils;
import java.util.Arrays;
import java.util.List;
/**
 * commons.collections4 集合分片
 */
public class PartitionExample {
    // 原集合
    private static final List<String> OLD_LIST = Arrays.asList(
            "唐僧,悟空,八戒,沙僧,曹操,刘备,孙权".split(","));
    public static void main(String[] args) {
        // 集合分片
        List<List<String>> newList = ListUtils.partition(OLD_LIST, 3);
        newList.forEach(i -> {
            System.out.println("集合长度:" + i.size());
        });
    }
}

2.4.3 使用国产神级框架 Hutool 实现分片

<dependency>
  <groupId>cn.hutool</groupId>
  <artifactId>hutool-all</artifactId>
  <version>5.7.14</version>
</dependency>
import cn.hutool.core.collection.ListUtil;
import java.util.Arrays;
import java.util.List;
public class PartitionByHutoolExample {
    // 原集合
    private static final List<String> OLD_LIST = Arrays.asList(
            "唐僧,悟空,八戒,沙僧,曹操,刘备,孙权".split(","));
    public static void main(String[] args) {
        // 分片处理
        List<List<String>> newList = ListUtil.partition(OLD_LIST, 3);
        newList.forEach(i -> {
            System.out.println("集合长度:" + i.size());
        });
    }
}

2.4.4 自定义实现分片

           if (listServiceResult.isSucceed() && listServiceResult.getData().size() > 0) {
					List<AgentAO> resultData = listServiceResult.getData();
					int totalSize = listServiceResult.getData().size(); // 总记录数
					int pageCount = 0; // 页数
					int pageSize = 50; // 每页50条
					if (totalSize < 50) {
						pageCount = 1;
					}
					int mod = totalSize % pageSize;
					if (pageCount > 0) {
						pageCount = totalSize / pageSize + 1;
					} else {
						pageCount = totalSize / pageSize;
					}
					for (int i = 0; i < pageCount; i++) {
						List<AgentAO> subList1 = shipLogic(resultData, pageCount, pageSize, i, mod);
						if (CollectionUtil.isNotEmpty(subList1)) {
						 // TODO 处理数据
						}
					}
                } 

​ 切分数据的两个方法:

     /**
	 * 切割数据
	 *
	 * @param data
	 * @param pageCount
	 * @param pageSize
	 * @param i
	 * @param mod
	 * @return
	 */
	private List<AgentAO> shipLogic(List<AgentAO> data, int pageCount, int pageSize, int i, int mod) {
		List<AgentAO> subList;
		int startIndex = 0;
		int endIndex = 0;
		if (pageCount == 1) {
			return data;
		} else {
			if (mod == 0) {
				startIndex = i * pageSize;
				endIndex = (i + 1) * pageSize - 1;
			} else {
				startIndex = i * pageSize;
				if (i == pageCount - 1) {
					endIndex = i * pageSize + mod;
				} else {
					endIndex = (i + 1) * pageSize;
				}
			}
		}
		System.out.println("startIndex=" + startIndex + ",endIndex=" + endIndex);
		subList = data.subList(startIndex, endIndex);
		return subList;
	}		
    /**
	 * 切割数据
	 * 使用 JDK 8 中提供 Stream 实现分片
	 */
	private <T> List<List<T>> splitList(List<T> list, int batchSzie) {
		if (CollectionUtils.isEmpty(list)) {
			return null;
		}
		if (list.size() <= batchSzie) {
			return Arrays.asList(list);
		}
		int limit = (list.size() + batchSzie - 1) / batchSzie;
		List<List<T>> splitList = Stream.iterate(0, n -> n + 1).limit(limit).parallel()
				.map(a -> list.stream().skip(a * batchSzie).limit(batchSzie).parallel().collect(Collectors.toList()))
				.collect(Collectors.toList());
		return splitList;
	}
	public static void main(String[] args) {
		List<String> data =new ArrayList<>();
		for(int i=0;i<1000;i++){
			data.add(String.valueOf(i));
		}
		MessagePlanService ms =new MessagePlanService();
		List<List<String>> result = ms.splitList(data,500);
		System.out.println(result.size());
		for(int j= 0;j<result.size();j++){
			List<String> d = result.get(j);
			for(int g=0;g<d.size();g++){
              System.out.println(d.get(g));
			}
		}
	}

​ mybatisPlus将一个List结果转换为一个Page分页对象:

     if (CollectionUtils.isNotEmpty(wmOfpk)) {
                pageInfo.setTotal(wmOfpk.size());
                // 计算总页数
                pages = (wmOfpk.size() + whiteListPageDTO.getSize() - 1) / whiteListPageDTO.getSize();
                pageInfo.setPages(pages);
                if (pageInfo.getCurrent() > pages) {
                    pageInfo.setCurrent(pages);
                }
                long start = (pageInfo.getCurrent() - 1) * whiteListPageDTO.getSize();
                long end = pageInfo.getCurrent() * whiteListPageDTO.getSize();
                if (start != 0) {
                    start = start - 1;
                }
                if (pageInfo.getCurrent() == pages) {
                    end = wmOfpk.size() - 1;
                }
                return pageInfo.setRecords(wmOfpk.subList((int) start, (int) end));
            } else {
                pageInfo.setTotal(0);
                pageInfo.setPages(0);
                return pageInfo.setRecords(null);
      }

3.总结

​ 原来我使用多线程写入遇到了事务问题,导致插入到表中的数据每次执行都会少了,后面才知道是myBatis使用的是DefaultSqlSession获取会话的这个类是不安全,所以我才使用mybatis-Spring包下面的的这个SqlSessionTemplate的类提供的方法,自己写了一个原生的mybatis的批量更新的方法,Dao层的Mapper没有集成myBatisPlus的BaseMapper,批量方法就是要处理全部的字段,然后有插入的实体中空的字段没有做判断的就会报错,所以需要判空:

 <update id="updateBatch"
            parameterType="java.util.List">
        <foreach collection="list" item="item" index="index" open="" close="" separator=";">
            update xxxxxxxx
            <trim prefix="set" suffixOverrides=",">
                <if test=" item.id != null ">
                    id = #{item.id, jdbcType=INTEGER},
                </if>
                <if test=" item.xxxxx != null ">
                    xxxxx = #{item.xxxxx, jdbcType=INTEGER},
                </if>
                <if test="item.xxxxx != null and item.xxxx.trim() neq '' ">
                    order_no = #{item.xxxx, jdbcType=VARCHAR},
                </if>
            </trim>
            <where>
                id = #{item.id, jdbcType=INTEGER}
            </where>
        </foreach>
    </update>

然后调用可以成功的插入:

       //批量更新数据 Dao类的全路径
      String updateStatement = "com.dytzxxxxxxxx.xxxxxx.updateBatch";
       try {
                       if (CollectionUtils.isNotEmpty(noResult) && records1.containsAll(noResult)) {
                                log.info("==========不需要更新的数据=========={}", JSON.toJSONString(noResult));
                                records1.removeAll(noResult);
                            }
                            if (CollectionUtils.isNotEmpty(records1)) {
                                log.info("==========批量更新开始==========");
                                sqlSessionTemplate.update(updateStatement, records1);
                                log.info("==========批量更新结束==========");
                            }
                     } catch (Exception e) {
                        e.getStackTrace();
                  log.error("==========批量更新异常=========={}",                 ExceptionUtils.getMessage(e));
     }

​ 这个是有查询和写入多线程环境下都查询和修改到同一条数据的情况就会出问题,最莫名奇妙的问题就是使用mybaitsPlus的save方法还是saveBatch方法在多线程环境下读写数据会出现报错,单线批量后的数据变少了,原因就是mybaitsPlus使用的DefaultSqlSession,这个类是不安全的,使用mybatis-Spring包下面的的这个SqlSessionTemplate的类session是安全的,原因是每次去获取session都加了jvm的(synchronized)锁,保证了获取session不出现并发操作,使用这个类之后在使用多线程来跑就会直接报错,错误就是提示:有的线程没有获取到session而导致调用方法报错,所以使用mybatis-Spring包下面的的这个SqlSessionTemplate写批量更新不能使用多线程操作,这个也是我之前遇到的坑,希望对大家有帮助,请关注点赞加收藏,一键三连哦。

图片

4.Demo百度网盘

链接: https://pan.baidu.com/s/1bn1feJ0jqaSf6phzV3bvAg

提取码: 8m2g

到此这篇关于mybatisPlus批量插入优化,性能快的飞起的文章就介绍到这了,更多相关mybatisPlus批量插入内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文