Java多线程实现

ゝ一世哀愁。 2024-03-31 10:16 76阅读 0赞

### 多线程实现 ###

//按自己需求入参
     private ArrayList<JSONObject> multiThreading(ArrayList list, List<String> errorMsg)     {
        
            // 返回的数据
            // 每10条数据开启一条线程
            int threadSize = 8;
            // 总数据条数
            int dataSize = list.size();
            // 线程数
            int threadNum = dataSize / threadSize + 1;
            // 计数器
            CountDownLatch countDownLatch = new CountDownLatch(threadNum);
            // 定义标记,过滤threadNum为整数
            boolean special = dataSize % threadSize == 0;
            // 创建一个线程池
            ExecutorService exec = Executors.newFixedThreadPool(threadNum);
            // 定义一个任务集合
            List<Callable<Boolean>> tasks = Lists.newArrayList();
            // 定义一个任务
            Callable<Boolean> task;
            // 定义循环处理的i批次数据
            List loopDataList;
            // 确定每条线程的数据
            final ArrayList<JSONObject> finalResult = new ArrayList<>();
            for (int i = 0; i < threadNum; i++) {
        
                if (i == threadNum - 1) {
        
                    if (special) {
        
                        countDownLatch.countDown();
                        break;
                    }
                    loopDataList = list.subList(threadSize * i, dataSize);
                } else {
        
                    loopDataList = list.subList(threadSize * i, threadSize * (i + 1));
                }
                // 当前循环处理的数据集
                final List recipientInfoDTOS = loopDataList;
                task = new Callable<Boolean>() {
        
                    @Override
                    public Boolean call() {
        
                        for (Object recipientInfoDTO : recipientInfoDTOS) {
        
                            //自己业务代码
                        }
                        return true;
                    }
                };
                // 减少计数器的计数，如果计数达到零，则释放所有等待线程。
                // 如果当前计数大于零，则递减。如果新计数为零，则重新启用所有等待线程以进行线程调度。
                countDownLatch.countDown();
                // 任务处理完加入集合
                tasks.add(task);
            }
            try {
        
                // 执行给定的任务，返回一个 Futures 列表，在所有完成时保存它们的状态和结果。
                // Future.isDone对于返回列表的每个元素都是true 。
                // 请注意，已完成的任务可能已经正常终止，也可能通过引发异常终止。
                // 如果在此操作进行时修改了给定的集合，则此方法的结果是不确定的
                exec.invokeAll(tasks);
                // 等待计数器归零
                countDownLatch.await();
            } catch (InterruptedException e) {
        
                e.printStackTrace();
            }
            // 关闭线程池
            exec.shutdown();
            return finalResult;
        }