多线程之批量插入
背景
昨天在测试mysql的两种批量更新时，由于需要入库大量测试数据，反复执行插入脚本，过程繁琐，档次很低，测试完后我就想着写个批量插入的小demo，然后又想写个多线程的批量插入的demo，然后就有了下面的东西了……
环境
spring-boot 1.5.6 集成 mysql druid mybits 还有一些无关紧要的东西
代码
线程类：
/**
* @ClassName InsertDataThread
* @Description <插入数据类>
* @Author zhaiyt
* @Date 2018/8/29 17:04
* @Version 1.0
*/
public class InsertDataThread extends Thread {
//日志
private final Logger logger = LoggerFactory.getLogger(InsertDataThread.class);
//数据访问层
private UserEntityMapper userEntityMapper;
//具体插入批次
private int batch;
//插入的数据
private List<UserEntity> list;
public InsertDataThread(UserEntityMapper userMpper, List<UserEntity> li, int batch) {
this.userEntityMapper = userMpper;
this.list = li;
this.batch = batch;
}
@Override
public void run() {
try {
this.userEntityMapper.insertBatch(this.list);
logger.info("第" + this.batch + "批次插入成功");
} catch (Exception e) {
logger.error("第" + this.batch + "批次插入失败");
}
}
}
===============================================================================
service层的多线程批量插入方法：
/**
* @param list
* @return int
* @Description <批量插入>
* @Author zhaiyt
* @Date 9:51 2018/8/29
* @Param [list]
*/
@Override
public int insertBatch(List<UserEntity> list) throws Exception {
PageHelper.offsetPage(0, 500000);
long start = System.currentTimeMillis();
List<UserEntity> listUser = userEntityMapper.selectAllUser();
int betch = 0;
if (CollectionUtils.isEmpty(listUser)) {
logger.error("表中无数据，需要改造测试");
return 0;
}
//根据数据量判断是否使用多线程 选择开启线程数
if (listUser.size() > 1000000) {
betch = 10;
} else if (listUser.size() > 100000) {
betch = 5;
} else if (listUser.size() > 10000) {
betch = 3;
} else {
//不走多线程
long end = System.currentTimeMillis();
logger.error("查询耗时:" + (end - start));
start = System.currentTimeMillis();
int count = userEntityMapper.insertBatch(listUser);
end = System.currentTimeMillis();
logger.error("插入耗时:" + (end - start));
return count;
}
//计数器
int size = 0;
//创建线程池
ExecutorService fixedThreadPool = Executors.newFixedThreadPool(betch);
for (int i = 0; i < (Math.ceil(listUser.size() / 5000)); i++) {
int startLen = i * 5000;
int endLen = ((i + 1) * 5000 > listUser.size() ? listUser.size() - 1 : (i + 1) * 5000);
// 该线程处理
List<UserEntity> threadList = listUser.subList(startLen, endLen);
size = size + threadList.size();
fixedThreadPool.execute(new InsertDataThread(userEntityMapper, threadList, i));
}
System.err.println("插入数据总条数：" + size);
long end = System.currentTimeMillis();
logger.error("查询耗时:" + (end - start));
return size;
}