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();
        };

    }
}