java实现限定时间countdownlatch并行场景-pg电子游戏官网

热度：11℃ 发布时间：2022-01-21 14:34:47

目录业务场景：
pg电子游戏官网官方网站的解决方案：
总结

业务场景：

一个用户数据接口，要求在20ms内返回数据，它的调用逻辑复杂，关联接口多，需要从3个接口汇总数据，这些汇总接口最小耗时也需要16ms,全部汇总接口最优状态耗时需要16ms*3=48ms

pg电子游戏官网官方网站的解决方案：

使用并行调用接口，通过多线程同时获取结果集，最后进行结果整合。在这种场景下，使用concurrent包的countdownlatch完成相关操作。countdownlatch本质上是一个计数器，把它初始化为与执行任务相同的数量，当一个任务执行完时，就将计数器的值减1，直到计算器达到0时，表示完成了所有任务，在await上等待线程就继续执行。

为上述业务场景封装的工具类，传入两个参数：一个参数是计算的task数量，另外一个参数是整个大任务超时的毫秒数。

import .util.concurrent.arrayblockingqueue;import java.util.concurrent.countdownlatch;import java.util.concurrent.threadpoolexecutor;import java.util.concurrent.timeunit;public class parallelcollector { private long timeout; private countdownlatch countdownlatch; threadpoolexecutor executor = new threadpoolexecutor(100, 200, 1, timeunit.hours, new arrayblockingqueue<>(100)); public parallelcollector(int tasksize, long timeoutmill) { countdownlatch = new countdownlatch(tasksize); timeout = timeoutmill; } public void submittask(runnable runnable) { executor.execute(() -> { runnable.run(); countdownlatch.countdown(); }); } public void await() { try { this.countdownlatch.await(timeout, timeunit.milliseconds); } catch (interruptedexception e) { e.printstacktrace(); } } public void destroy() { this.executor.shutdown(); }}

当任务运行时间超过了任务的时间上限，就被直接停止，这就是await()的功能。

interface是一个模拟远程服务的超时的测试类，程序运行后，会输出执行结果到map集合。

public class interfacemock { private volatile int num=1; public string slowmethod1() { try { thread.sleep(2000); } catch (interruptedexception e) { e.printstacktrace(); } return string.valueof(num 1); }; public string slowmethod2() { return string.valueof(num 1); }; public string slowmethod3() { return string.valueof(num 1); };}

并行执行获取结果测试类

@springboottestclass threadpoolapplicationtests { @test void testtask() { interfacemock interfacemock = new interfacemock(); parallelcollector collector = new parallelcollector(3, 20l); concurrenthashmap map = new concurrenthashmap<>(); collector.submittask(()->map.put("method1",interfacemock.slowmethod1())); collector.submittask(()->map.put("method2",interfacemock.slowmethod2())); collector.submittask(()->map.put("method3",interfacemock.slowmethod3())); collector.await(); system.out.println(map.tostring()); collector.destroy(); }}

当method1()执行时间大于20ms,则该方法直接被终止，结果map集没有method1()的结果,结果如下：