library/cpp/messagebus/test/ut/count_down_latch.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

#pragma once 
 
#include <util/system/atomic.h> 
#include <util/system/event.h> 
 
class TCountDownLatch { 
private: 
    TAtomic Current; 
    TSystemEvent EventObject;

public: 
    TCountDownLatch(unsigned initial) 
        : Current(initial) 
    {
    }
 
    void CountDown() { 
        if (AtomicDecrement(Current) == 0) { 
            EventObject.Signal(); 
        } 
    } 
 
    void Await() { 
        EventObject.Wait(); 
    } 
 
    bool Await(TDuration timeout) { 
        return EventObject.WaitT(timeout); 
    } 
};