library/cpp/messagebus/session_job_count.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
31
32
33
34
35
36
37
38
39

#pragma once 
 
#include <util/system/atomic.h> 
#include <util/system/condvar.h>
#include <util/system/mutex.h> 
 
namespace NBus {
    namespace NPrivate {
        class TBusSessionJobCount {
        private:
            TAtomic JobCount;
 
            TMutex Mutex;
            TCondVar CondVar;
 
        public:
            TBusSessionJobCount();
            ~TBusSessionJobCount();
 
            void Add(unsigned delta) {
                AtomicAdd(JobCount, delta);
            }
 
            void Increment() {
                Add(1);
            }
 
            void Decrement() {
                if (AtomicDecrement(JobCount) == 0) {
                    TGuard<TMutex> guard(Mutex);
                    CondVar.BroadCast();
                }
            }
 
            void WaitForZero();
        };
 
    } 
}