aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/messagebus/use_count_checker.h
diff options
context:
space:
mode:
Diffstat (limited to 'library/cpp/messagebus/use_count_checker.h')
-rw-r--r--library/cpp/messagebus/use_count_checker.h27
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();
+};