美文网首页
CountDownLatch 注意事项

CountDownLatch 注意事项

作者: 赵荆州 | 来源:发表于2019-07-10 10:05 被阅读0次

先看一段代码:


CountDownLatch countDownLatch = new CountDownLatch(params.size());
        try {
            params.forEach(regionNameMapping -> {
                CompletableFuture.runAsync(() -> {
                        Region region = this.getRegionById(regionNameMapping.getRegionId());
 regionNameMapping.getRegionNameConsumer().accept(region.getName());
                        countDownLatch.countDown();
                });
            });
            countDownLatch.await();
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }

上面的代码逻辑很简单,并发去执行getRegionById这个方法。然后await等待结果。
但是里面有一个隐患,当region查出来为null时,会出现NPE,就会导致countDown()无法被执行,于是程序就一直阻塞在

countDownLatch.await();

正确的写法:

CountDownLatch countDownLatch = new CountDownLatch(params.size());
        try {
            params.forEach(regionNameMapping -> {
                CompletableFuture.runAsync(() -> {
                      try{
                        Region region = this.getRegionById(regionNameMapping.getRegionId());
 regionNameMapping.getRegionNameConsumer().accept(region.getName());
                      }finally{
                        countDownLatch.countDown();
                      }
                });
            });
            countDownLatch.await(3,TimeUnit.SECONDS);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }

添加try、finally,保证最终锁会被释放,以及设置等待超时时间,避免程序挂掉。

所有使用AbstractQueuedSynchronizer实现的同步器,例如Lock,Semaphore、等都应该如此。

相关文章

网友评论

      本文标题:CountDownLatch 注意事项

      本文链接:https://www.haomeiwen.com/subject/mqrckctx.html