多线程设计模式-Latch模式

概念

Latch(阀门)设计模式也叫做 Count Down 设计模式。当若干个线程并发执行完某个特 定的任务,然后等到所有的子任务都执行结束之后再统一汇总。

CountDownLatch 能够使一个线程在等待另外一些线程完成各自工作之后,再继续执 行。使用一个计数器进行实现。计数器初始值为线程的数量。当每一个线程完成自己任务后, 计数器的值就会减一。当计数器的值为 0 时,表示所有的线程都已经完成了任务,然后在

CountDownLatch 上等待的线程就可以恢复执行任务。示例代码如下:

/**
 * 自定义CustomCountDownLatch
 */
public class CustomCountDownLatch {
    //阀门值
    private final int latchNum;

    //计数器
    private int  count;

    public CustomCountDownLatch(int latch_num) {
        this.latchNum = latch_num;
        this.count = latch_num;
    }

    //计数方法 进行count--
    public synchronized void countDown(){
        this.count--;
        notifyAll();
    }

    //阻止线程继续执行 依据计数器判断线程能否执行
    public synchronized void await() {
        if(this.count!=0){
            try {
                this.wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
针对多线程编程的单元测试,可以使用Mockito的一些特性和技巧来完成。以下是一些常用的技巧: 1. 使用Mockito的异步调用技术,可以模拟多线程环境下的方法调用。例如,使用Mockito的when().thenAnswer()方法来模拟异步调用: ```java @Test public void testAsyncMethod() throws Exception { CompletableFuture<String> future = new CompletableFuture<>(); when(mock.asyncMethod()).thenAnswer(invocation -> { new Thread(() -> { future.complete("mock result"); }).start(); return future; }); String result = mock.asyncMethod().get(); assertEquals("mock result", result); } ``` 2. 使用CountDownLatch多线程工具类来控制多线程的执行顺序。例如,使用CountDownLatch来等待所有线程执行完毕: ```java @Test public void testMultiThread() throws Exception { CountDownLatch latch = new CountDownLatch(2); new Thread(() -> { // do something latch.countDown(); }).start(); new Thread(() -> { // do something latch.countDown(); }).start(); latch.await(); // assert something } ``` 3. 使用Spy和ArgumentMatchers等技巧来模拟多线程环境下的方法调用和参数传递。例如,使用Spy来模拟一个多线程环境下的对象: ```java @Test public void testMultiThreadWithSpy() throws Exception { MyClass myClass = new MyClass(); MyClass spy = spy(myClass); CountDownLatch latch = new CountDownLatch(2); new Thread(() -> { doAnswer(invocation -> { // do something latch.countDown(); return null; }).when(spy).myMethod(anyString()); }).start(); new Thread(() -> { doAnswer(invocation -> { // do something latch.countDown(); return null; }).when(spy).myMethod(anyString()); }).start(); latch.await(); verify(spy, times(2)).myMethod(anyString()); } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值