执门文章
java项目中多线程并行执行获取汇总结果
于 2024-03-02 21:29:06 发布
访问647
收藏0
文章标签:
CountDownLatch
THread

代码下载地址:

https://download.csdn.net/download/hekf2010/13124627

很多时间,需要多线程并行处理,然后汇总处理结果,例如,因为请参数数据量非常大,所以只能分批查询,然后将分批查询的结果汇总到一个数组中。

一、多线程类

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;
 
/**
 * @param <H> 为被处理的数据类型
 * @param <T>返回数据类型
 */
public abstract class MultiThread<H,T>{
    private final ExecutorService exec; //线程池
    private final BlockingQueue<Future<List<T>>> queue = new LinkedBlockingQueue<>();
    private final CountDownLatch startLock = new CountDownLatch(1); //启动门,当所有线程就绪时调用countDown
    private final CountDownLatch endLock; //结束门
    private final List<List<H>> listData;//被处理的数据
 
    /**
     * @param list list.size()为多少个线程处理,list里面的List<H>为其中一个线程需要理的数据
     */
    public MultiThread(List<List<H>> list){
        if(list!=null&&list.size()>0){
            this.listData = list;
            exec = Executors.newFixedThreadPool(list.size()); //创建线程池,线程池共有nThread个线程
            endLock = new CountDownLatch(list.size());  //设置结束门计数器,当一个线程结束时调用countDown
        }else{
            listData = null;
            exec = null;
            endLock = null;
        }
    }
 
    /**
     *
     * @return 获取每个线程处理结速的数组
     * @throws InterruptedException
     * @throws ExecutionException
     */
    public List<T> getResult() throws InterruptedException, ExecutionException{
        List<T> resultList = new ArrayList<>();
        if(listData!=null&&listData.size()>0){
            int nThread = listData.size(); //线程数量
            for(int i = 0; i < nThread; i++){
                List<H> data = listData.get(i);
                Future<List<T>> future = exec.submit(new Task(i,data){
                    @Override
                    public List<T> execute(int currentThread,List<H> data) {
                        return outExecute(currentThread,data);
                    }
                }); //将任务提交到线程池
                queue.add(future); //将Future实例添加至队列
            }
            startLock.countDown(); //所有任务添加完毕,启动门计数器减1,这时计数器为0,所有添加的任务开始执行
            endLock.await(); //主线程阻塞,直到所有线程执行完成
            for(Future<List<T>> future : queue) {
                List<T> currentThreadList = future.get();
                for(T t:currentThreadList){
                    resultList.add(t);
                }
            }
            exec.shutdown(); //关闭线程池
        }
        return resultList;
    }
    /**
     * 每一个线程执行的功能,需要调用者来实现
     * @param currentThread 线程号
     * @param data 每个线程需要处理的数据
     * @return List<T>返回值为传入的data对应的结果
     */
    public abstract List<T> outExecute(int currentThread,List<H> data);
 
    /**
     * 线程类
     */
    private abstract class Task implements Callable<List<T>>{
        private int currentThread;//当前线程号
        private List<H> data;
        public Task(int currentThread,List<H> data){
            this.currentThread=currentThread;
            this.data=data;
        }
        @Override
        public List<T> call() throws Exception {
            startLock.await(); //线程启动后调用await,当前线程阻塞,只有启动门计数器为0时当前线程才会往下执行
            List<T> t =null;
            try{
                t = execute(currentThread,data);
            }finally{
                endLock.countDown(); //线程执行完毕,结束门计数器减1
            }
            return t;
        }
 
        /**
         * 每一个线程执行的功能
         * @param currentThread 线程号
         * @param data 每个线程被处理的数据
         * @return T返回对象
         */
        public abstract List<T> execute(int currentThread,List<H> data);
    }
}

二、参数类

public class ParamVo {
 
}

三、结果类

public class ResultVo {
 
}

四、测试

import java.util.ArrayList;
import java.util.List;
 
public class Test {
 
	public static void main(String[] args) {
		try {
			List<List<ParamVo>> threadList = new ArrayList<List<ParamVo>>();//线程数组,size就是线程数量
			for(int j=0;j<10;j++) {
				List<ParamVo> oneThreadDealData = new ArrayList<>();//一个线程需要处理的数量
				for (int i = 0; i < 5; i++) {
					oneThreadDealData.add(new ParamVo());
				}
				threadList.add(oneThreadDealData);
			}
 
 
			MultiThread<ParamVo,ResultVo> multiThread = new MultiThread<ParamVo,ResultVo>(threadList){
				@Override
				public List<ResultVo> outExecute(int currentThread, List<ParamVo> data) {
					List<ResultVo> list = new ArrayList<>();
					for(ParamVo vo:data) {
						System.out.println("当前线程号=" + currentThread + " data=" + data);
						list.add(new ResultVo());
					}
					return list;
				}
			};
			List<ResultVo> list = multiThread.getResult();
 
			//获取每一批次处理结果
			System.out.println("获取处理结果........................");
			for(ResultVo vo:list){
				//System.out.println(vo);
			}
			System.out.println(list.size());
		} catch (Exception e) {
			e.printStackTrace();
		}
 
 
	}
 
}


于 2024-03-02 21:29:06 发布
访问647
收藏0

实力雄厚的技术网站

产品

概述

资源

文档

版权问题

请联系客服

联系我们

联系我们


© 财瑞智能科技 2024蜀ICP备2023018175号-2川公网安备51011202000656号