复习笔记->semaphore/countDownLatch

2020-12-21

前言

开发也经常需要跑并发测试，要是每次需要开jmater/loadrunner那这样太麻烦了，这边简单记录一个单元测试，实际生产使用需要对这快在进行优化。

锁的差别

synchronized是互斥锁，每次只能有一个线程，semaphore也是同步锁维护的是一个许可集，调用acquire()获取一个许可，release()释放一个许可。

信号枪

countDownLatch 信号枪当countDown退到0时候就执行awit的方法。可以用于并发/主线城等待等方式上的使用。

单元测试

@Test
    public void 测试信号量() throws InterruptedException {
        int threadTotal = 20, clientTotal = 100;
//       线程池        
        ExecutorService executorService = Executors.newCachedThreadPool();
//        定义信号量
        final Semaphore semaphore = new Semaphore(threadTotal);
//        定义信号枪-->countDown为0时才执行awit方法，保证主线程的等待
        final CountDownLatch countDownLatch = new CountDownLatch(clientTotal);
        System.out.println("当前时间："+ new Date().toLocaleString());
        for (int i = 0; i < clientTotal; i++) {
            executorService.execute(() -> {
                try {
//                    一次只允许进入一个，多余的会等待
                    semaphore.acquire();
                    // 判断是否正常
//                    System.out.println("当前线程：" + Thread.currentThread().getName());
//                    睡眠2秒
                    TimeUnit.SECONDS.sleep(2);
                    semaphore.release();
                } catch (Exception e) {
                    e.printStackTrace();
                }
                countDownLatch.countDown();
            });
        }
        countDownLatch.await();
        executorService.shutdown();
        System.out.println("结束时间："+ new Date().toLocaleString());
    }