CountDownLatch正确使用

在一些List处理场景中，为了加速处理，一般使用CountDownLatch+threadPoolExecutor实现并发处理。

但线程池处理中，可能出现未知异常，导致CountDownLatch减1缺少，进一步导致await一直阻塞等待。这种情况，就需要合理设置await的超时等待时间，避免请求一直阻塞等待，影响到其他。

当然，也可以不设置await的超时等待时间，但是需要保证线程执行正常或不正常结束，CountDownLatch一定会减1。

建议两方面都考虑，既处理线程正常或不正常结束的CountDownLatch减1，又设置await合适的时间。

本文提供一个demo，仅供参考。

CountDownLatch.java

// 循环处理节点列表
CountDownLatch countDownLatch = new CountDownLatch(chainNodeEntityList.size());
chainNodeEntityList.forEach(
        chainNodeEntity -> {
            threadPoolExecutor.execute(
                    () -> {
                        // 查询节点根据节点id
                        LambdaQueryWrapper<NodeEntity> lqwNode = new LambdaQueryWrapper<>();
                        lqwNode.eq(NodeEntity::getId, chainNodeEntity.getNodeId());
                        lqwNode.eq(NodeEntity::getDel, 0);
                        NodeEntity nodeEntity = nodeService.getOne(lqwNode);
                        // 节点不存在
                        if (nodeEntity == null) {
                            log.error(
                                    "node not exist，节点id：{}", chainNodeEntity.getNodeId());

                            // 减少计数
                            countDownLatch.countDown();
                            // 结束线程
                            return;
                        }

                        // 定义节点返回实体
                        NodeSingleInfo nodeInfo = new NodeSingleInfo();
                        // copy属性
                        // 设置节点运营机构
                        // 设置节点状态
                        BeanUtils.copyProperties(nodeEntity, nodeInfo);

                        // 添加到返回列表
                        nodeInfoList.add(nodeInfo);

                        // 减少计数
                        countDownLatch.countDown();
                    });
        });
// 等待所有线程执行完毕
try {
    if (!countDownLatch.await(60, TimeUnit.SECONDS)) {
        log.error("countDownLatch.await return false");
    }
} catch (InterruptedException e) {
    log.error(
            "getChainNode error，链：{}，原因：{}", chainNodeSingleReq.getChain(), e.getMessage());

    Thread.currentThread().interrupt();
}

可以看到，await设置了超时时间，以防线程内未知异常，导致countDownLatch一直阻塞。

线程池里的线程，如果结束，使用的是return。在结束之前，一定需要**countDownLatch.countDown()**去减少计数。