library/cpp/messagebus/use_count_checker.cpp


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
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#include "use_count_checker.h"

#include <util/generic/utility.h>
#include <util/system/yassert.h>

TUseCountChecker::TUseCountChecker() {
}

TUseCountChecker::~TUseCountChecker() {
    TAtomicBase count = Counter.Val();
    Y_VERIFY(count == 0, "must not release when count is not zero: %ld", (long)count); 
}

void TUseCountChecker::Inc() {
    Counter.Inc();
}

void TUseCountChecker::Dec() {
    Counter.Dec();
}

TUseCountHolder::TUseCountHolder()
    : CurrentChecker(nullptr)
{ 
} 

TUseCountHolder::TUseCountHolder(TUseCountChecker* currentChecker)
    : CurrentChecker(currentChecker)
{
    if (!!CurrentChecker) {
        CurrentChecker->Inc();
    }
}

TUseCountHolder::~TUseCountHolder() { 
    if (!!CurrentChecker) {
        CurrentChecker->Dec();
    }
}

TUseCountHolder& TUseCountHolder::operator=(TUseCountHolder that) {
    Swap(that);
    return *this;
}

void TUseCountHolder::Swap(TUseCountHolder& that) {
    DoSwap(CurrentChecker, that.CurrentChecker);
}

void TUseCountHolder::Reset() {
    TUseCountHolder tmp;
    Swap(tmp);
}