Java批量更新太慢?多線程+List分段完美解決!
寫(xiě)在前面
步驟
寫(xiě)在最后
寫(xiě)在前面
相信不少開(kāi)發(fā)者在遇到項(xiàng)目對(duì)數(shù)據(jù)進(jìn)行批量操作的時(shí)候,都會(huì)有不少的煩惱,尤其是針對(duì)數(shù)據(jù)量極大的情況下,效率問(wèn)題就直接提上了菜板。
因此,開(kāi)多線程來(lái)執(zhí)行批量任務(wù)是十分重要的一種批量操作思路,其實(shí)這種思路實(shí)現(xiàn)起來(lái)也十分簡(jiǎn)單,就拿批量更新的操作舉例。
整體流程圖如下:

步驟
步驟如下:
獲取需要進(jìn)行批量更新的大集合 A,對(duì)大集合進(jìn)行拆分操作,分成 N 個(gè)小集合 A-1 ~ A-N 。 開(kāi)啟線程池,針對(duì)集合的大小進(jìn)行調(diào)參,對(duì)小集合進(jìn)行批量更新操作。 對(duì)流程進(jìn)行控制,控制線程執(zhí)行順序。
按照指定大小拆分集合的工具類:
import com.google.common.collect.Lists;
import org.apache.commons.collections.CollectionUtils;
import java.util.List;
/**
* 拆分結(jié)合工具類
*
* @author shiwen
* @date 2020/12/27
*/
public class SplitListUtils {
/**
* 拆分集合
*
* @param <T> 泛型對(duì)象
* @param resList 需要拆分的集合
* @param subListLength 每個(gè)子集合的元素個(gè)數(shù)
* @return 返回拆分后的各個(gè)集合組成的列表
* 代碼里面用到了guava和common的結(jié)合工具類
**/
public static <T> List<List<T>> split(List<T> resList, int subListLength) {
if (CollectionUtils.isEmpty(resList) || subListLength <= 0) {
return Lists.newArrayList();
}
List<List<T>> ret = Lists.newArrayList();
int size = resList.size();
if (size <= subListLength) {
// 數(shù)據(jù)量不足 subListLength 指定的大小
ret.add(resList);
} else {
int pre = size / subListLength;
int last = size % subListLength;
// 前面pre個(gè)集合,每個(gè)大小都是 subListLength 個(gè)元素
for (int i = 0; i < pre; i++) {
List<T> itemList = Lists.newArrayList();
for (int j = 0; j < subListLength; j++) {
itemList.add(resList.get(i * subListLength + j));
}
ret.add(itemList);
}
// last的進(jìn)行處理
if (last > 0) {
List<T> itemList = Lists.newArrayList();
for (int i = 0; i < last; i++) {
itemList.add(resList.get(pre * subListLength + i));
}
ret.add(itemList);
}
}
return ret;
}
// 運(yùn)行代碼
public static void main(String[] args) {
List<String> list = Lists.newArrayList();
int size = 1099;
for (int i = 0; i < size; i++) {
list.add("hello-" + i);
}
// 大集合里面包含多個(gè)小集合
List<List<String>> temps = split(list, 100);
int j = 0;
// 對(duì)大集合里面的每一個(gè)小集合進(jìn)行操作
for (List<String> obj : temps) {
System.out.println(String.format("row:%s -> size:%s,data:%s", ++j, obj.size(), obj));
}
}
}
public void threadMethod() {
List<T> updateList = new ArrayList();
// 初始化線程池, 參數(shù)一定要一定要一定要調(diào)好?。。?!
ThreadPoolExecutor threadPool = new ThreadPoolExecutor(20, 50,
4, TimeUnit.SECONDS, new ArrayBlockingQueue(10), new ThreadPoolExecutor.AbortPolicy());
// 大集合拆分成N個(gè)小集合, 這里集合的size可以稍微小一些(這里我用100剛剛好), 以保證多線程異步執(zhí)行, 過(guò)大容易回到單線程
List<T> splitNList = SplitListUtils.split(totalList, 100);
// 記錄單個(gè)任務(wù)的執(zhí)行次數(shù)
CountDownLatch countDownLatch = new CountDownLatch(splitNList.size());
// 對(duì)拆分的集合進(jìn)行批量處理, 先拆分的集合, 再多線程執(zhí)行
for (List<T> singleList : splitNList) {
// 線程池執(zhí)行
threadPool.execute(new Thread(new Runnable(){
@Override
public void run() {
for (Entity yangshiwen : singleList) {
// 將每一個(gè)對(duì)象進(jìn)行數(shù)據(jù)封裝, 并添加到一個(gè)用于存儲(chǔ)更新數(shù)據(jù)的list
// ......
}
}
}));
// 任務(wù)個(gè)數(shù) - 1, 直至為0時(shí)喚醒a(bǔ)wait()
countDownLatch.countDown();
}
try {
// 讓當(dāng)前線程處于阻塞狀態(tài),直到鎖存器計(jì)數(shù)為零
countDownLatch.await();
} catch (InterruptedException e) {
throw new BusinessLogException(ResponseEnum.FAIL);
}
// 通過(guò)mybatis的批量插入的方式來(lái)進(jìn)行數(shù)據(jù)的插入, 這一步還是要做判空
if (GeneralUtil.listNotNull(updateList)) {
batchUpdateEntity(updateList);
LogUtil.info("xxxxxxxxxxxxxxx");
}
}
寫(xiě)在最后
多線程是 Java 的一個(gè)難點(diǎn),但是它也很有趣,聽(tīng)說(shuō)玩得溜得起飛的人,人生都開(kāi)啟多線程模式了…
評(píng)論
圖片
表情
