intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 36 insertions, 0 deletions

diff --git a/library/cpp/messagebus/extra_ref.h b/library/cpp/messagebus/extra_ref.h
new file mode 100644
index 0000000000..2927123266
--- /dev/null
+++ b/library/cpp/messagebus/extra_ref.h
@@ -0,0 +1,36 @@
+#pragma once
+
+#include <util/system/yassert.h>
+
+class TExtraRef {
+    TAtomic Holds;
+
+public:
+    TExtraRef()
+        : Holds(false)
+    {
+    }
+    ~TExtraRef() {
+        Y_VERIFY(!Holds);
+    }
+
+    template <typename TThis>
+    void Retain(TThis* thiz) {
+        if (AtomicGet(Holds)) {
+            return;
+        }
+        if (AtomicCas(&Holds, 1, 0)) {
+            thiz->Ref();
+        }
+    }
+
+    template <typename TThis>
+    void Release(TThis* thiz) {
+        if (!AtomicGet(Holds)) {
+            return;
+        }
+        if (AtomicCas(&Holds, 0, 1)) {
+            thiz->UnRef();
+        }
+    }
+};