工作的时候遇到一个这么一个问题,信息的群发任务,当代码写完,发现执行太慢了,主要是任务数太多导致的,但是这个结果又是要同步返回的,不能说只提交任务,不用管执行结果是什么。因为我这边的做法是,使用CountDownLatch控制,代码:
private final Semaphore semaphore = new Semaphore(10);
private ExecutorService cachedThreadPool = Executors.newCachedThreadPool();
CountDownLatch countDownLatch = new CountDownLatch(xtYwmbpzs.size());
for (XtYwmbpz xtYwmbpz : xtYwmbpzs) {
cachedThreadPool.execute(() -> {
XtMbpz xtMbpz = xtMbpzService.selectByPrimaryKey(xtYwmbpz.getMbpzid());
if (xtMbpz != null) {
log.info("找到模板配置:{}", xtMbpz.getMbid());
System.out.println(Thread.currentThread().getName() + "正在被执行");
try {
semaphore.acquire();
SendMessResUtils.complete(response, getMessageImpl(xtMbpz.getMblx()).sendMessage(param, xtMbpz, xtYwmbpz, originCode));
response.setMessage("发送完成");
semaphore.release();
} catch (Exception ex) {
ex.printStackTrace();
}
}
countDownLatch.countDown();
});
}
try {
countDownLatch.await(1, TimeUnit.MINUTES);
} catch (InterruptedException e) {
e.printStackTrace();
}