intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 53 insertions, 0 deletions

diff --git a/library/cpp/messagebus/use_count_checker.cpp b/library/cpp/messagebus/use_count_checker.cpp
new file mode 100644
index 0000000000..c6243ea21f
--- /dev/null
+++ b/library/cpp/messagebus/use_count_checker.cpp
@@ -0,0 +1,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);
+}