本篇文章给大家分享的是有关使用CountDownLatch怎么实现一个并发框架，小编觉得挺实用的，因此分享给大家学习，希望大家阅读完这篇文章后可以有所收获，话不多说，跟着小编一起来看看吧。
成都创新互联公司专业为企业提供金秀网站建设、金秀做网站、金秀网站设计、金秀网站制作等企业网站建设、网页设计与制作、金秀企业网站模板建站服务，10余年金秀做网站经验，不只是建网站，更提供有价值的思路和整体网络服务。
目录结构
package com.**.**.base.support.executor;import lombok.NoArgsConstructor;import org.apache.commons.logging.Log;import org.apache.commons.logging.LogFactory;import org.springframework.util.Assert;import java.util.ArrayList;import java.util.List;import java.util.concurrent.CountDownLatch;import java.util.concurrent.Executor;import java.util.concurrent.Executors;import java.util.stream.Collectors;/** * @Author lbm * @Date 2019/10/9 10:32 上午 * @Description YJ执行器 **/@NoArgsConstructorpublic class YjExecutor {private static final Log log = LogFactory.getLog(YjExecutor.class);private static final int FIRST_RETURN = 1;private CountDownLatch latch;private List executeUnits = new ArrayList<>();private Executor pool = Executors.newFixedThreadPool(10);public static YjExecutor build() {return new YjExecutor();
}public YjExecutor add(ExecuteUnit paramUnit) {
paramCheck(paramUnit);executeUnits.add(paramUnit);return this;
}public YjExecutor addAll(List paramUnits) {
paramCheck(paramUnits);this.executeUnits.addAll(paramUnits);return this;
}/** * 第一个执行完返回 * * @return 执行结果 */ public ReturnT firstPriorityReturn() {this.latch = new CountDownLatch(FIRST_RETURN);
List results = this.submit();return results.stream()
.filter(res -> res != null && res.getRes() != null)
.findFirst()
.orElse(null);
}/** * 所有执行完才返回 * * @return 执行结果 */ public List allReturn() {this.latch = new CountDownLatch(executeUnits.size());return submit();
}/** * 参数校验 * * @param paramUnits 参数列表 */ private void paramCheck(List paramUnits) {
Assert.notNull(paramUnits, "Argument can not be null !!");for (ExecuteUnit paramUnit : paramUnits) {
paramCheck(paramUnit);
}
}/** * 参数校验 * * @param paramUnit 单个参数 */ private void paramCheck(ExecuteUnit paramUnit) {
Assert.notNull(paramUnit, "Argument can not be null !!");
Assert.notNull(paramUnit.getFunc(), "Argument function can not be null !!");
}/** * 执行 */ private void execute() {try {for (ExecuteUnit executeUnit : executeUnits) {pool.execute(executeUnit.getYjRunner());
}latch.await();
} catch (Exception e) {log.error("execute error {}", e);
}
}/** * 提交 * * @return 返回执行结果 */ private List submit() {
buildExecuteUnit();
execute();return executeUnits.stream()
.map(ExecuteUnit::getResult)
.collect(Collectors.toList());
}/** * 创建执行单元 */ private void buildExecuteUnit() {for (ExecuteUnit executeUnit : executeUnits) {
executeUnit.setYjRunner(executeUnit.yjRunnerBuild(latch));
}
}
} package com.**.**.base.support.executor;import lombok.Builder;import lombok.Data;/** * @Author lbm * @Date 2019/10/9 10:10 上午 * @Description 执行结果 **/@Data@Builderpublic class ReturnT{private ExecuteEnum resCode;private R res; }
