快捷導(dǎo)航

Java使用多線程異步執(zhí)行批量更新操作方法

更新時(shí)間：2021年01月14日 08:45:00 作者：ADAMs.

這篇文章主要介紹了Java使用多線程異步執(zhí)行批量更新操作,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

寫在前面：

相信不少開發(fā)者在遇到項(xiàng)目對數(shù)據(jù)進(jìn)行批量操作的時(shí)候，都會有不少的煩惱，尤其是針對數(shù)據(jù)量極大的情況下，效率問題就直接提上了菜板。因此，開多線程來執(zhí)行批量任務(wù)是十分重要的一種批量操作思路，其實(shí)這種思路實(shí)現(xiàn)起來也十分簡單，就拿批量更新的操作舉例：

整體流程圖

在這里插入圖片描述

步驟獲取需要進(jìn)行批量更新的大集合A，對大集合進(jìn)行拆分操作，分成N個小集合A-1 ~ A-N 。
開啟線程池，針對集合的大小進(jìn)行調(diào)參，對小集合進(jìn)行批量更新操作。
對流程進(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> 泛型對象
  * @param resList 需要拆分的集合
  * @param subListLength 每個子集合的元素個數(shù)
  * @return 返回拆分后的各個集合組成的列表
  * 代碼里面用到了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個集合，每個大小都是 subListLength 個元素
   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);
  }
  // 大集合里面包含多個小集合
  List<List<String>> temps = split(list, 100);
  int j = 0;
  // 對大集合里面的每一個小集合進(jìn)行操作
  for (List<String> obj : temps) {
   System.out.println(String.format("row:%s -> size:%s,data:%s", ++j, obj.size(), obj));
  }
 }

}

開啟異步執(zhí)行任務(wù)的線程池

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個小集合, 這里集合的size可以稍微小一些（這里我用100剛剛好）, 以保證多線程異步執(zhí)行, 過大容易回到單線程
 List<T> splitNList = SplitListUtils.split(totalList, 100);
 // 記錄單個任務(wù)的執(zhí)行次數(shù)
 CountDownLatch countDownLatch = new CountDownLatch(splitNList.size());
 // 對拆分的集合進(jìn)行批量處理, 先拆分的集合, 再多線程執(zhí)行
 for (List<T> singleList : splitNList) {
  // 線程池執(zhí)行
  threadPool.execute(new Thread(new Runnable(){
   @Override
   public void run() {
    for (Entity yangshiwen : singleList) {
     // 將每一個對象進(jìn)行數(shù)據(jù)封裝, 并添加到一個用于存儲更新數(shù)據(jù)的list
     // ......
     
     // 任務(wù)個數(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);
 }
 // 通過mybatis的批量插入的方式來進(jìn)行數(shù)據(jù)的插入, 這一步還是要做判空
 if (GeneralUtil.listNotNull(updateList)) {
  batchUpdateEntity(updateList);
  LogUtil.info("xxxxxxxxxxxxxxx");
 }
}