diff options
Diffstat (limited to 'library/cpp/messagebus/use_count_checker.h')
-rw-r--r-- | library/cpp/messagebus/use_count_checker.h | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/library/cpp/messagebus/use_count_checker.h b/library/cpp/messagebus/use_count_checker.h new file mode 100644 index 0000000000..70bef6fa8a --- /dev/null +++ b/library/cpp/messagebus/use_count_checker.h @@ -0,0 +1,27 @@ +#pragma once + +#include <util/generic/refcount.h> + +class TUseCountChecker { +private: + TAtomicCounter Counter; + +public: + TUseCountChecker(); + ~TUseCountChecker(); + void Inc(); + void Dec(); +}; + +class TUseCountHolder { +private: + TUseCountChecker* CurrentChecker; + +public: + TUseCountHolder(); + explicit TUseCountHolder(TUseCountChecker* currentChecker); + TUseCountHolder& operator=(TUseCountHolder that); + ~TUseCountHolder(); + void Swap(TUseCountHolder&); + void Reset(); +}; |