IEventHandle Upcast/Downcast

https://a.yandex-team.ru/review/4895257/details#comment-7227783

1 files changed, 24 insertions, 5 deletions

diff --git a/ydb/library/actors/core/event.h b/ydb/library/actors/core/event.h
index 3926312e978..417ef53f337 100644
--- a/ydb/library/actors/core/event.h
+++ b/ydb/library/actors/core/event.h
@@ -53,6 +53,9 @@ namespace NActors {
         virtual TEventSerializationInfo CreateSerializationInfo() const { return {}; }
     };
 
+    template <typename TEventType>
+    class TEventHandle;
+
     // fat handle
     class IEventHandle : TNonCopyable {
         struct TOnNondelivery {
@@ -72,6 +75,27 @@ namespace NActors {
             return fits ? static_cast<TEv*>(Event.Get()) : nullptr;
         }
 
+        template <typename TEv>
+        inline TEv* StaticCastAsLocal() const noexcept {  // blind cast
+            if constexpr (!std::is_same_v<TEv, IEventBase>) {
+                Y_DEBUG_ABORT_UNLESS(GetTypeRewrite() == TEv::EventType);
+            };
+
+            return static_cast<TEv*>(Event.Get());
+        }
+
+        template <typename TEv>
+        static TAutoPtr<TEventHandle<TEv>> Downcast(TAutoPtr<IEventHandle>&& ev) {
+            Y_DEBUG_ABORT_UNLESS(ev->GetTypeRewrite() == TEv::EventType);
+
+            return static_cast<typename TEv::THandle*>(ev.Release());
+        };
+
+        template <typename TEv>
+        static TAutoPtr<IEventHandle> Upcast(TAutoPtr<TEventHandle<TEv>>&& ev) {
+            return ev.Release();
+        };
+
         template <typename TEventType>
         TEventType* Get() {
             if (Type != TEventType::EventType)
@@ -324,11 +348,6 @@ namespace NActors {
         static TAutoPtr<T> Release(THolder<IEventHandle>& ev) {
             return ev->Release<T>();
         }
-
-        template <typename TEv>
-        inline TEv* StaticCastAsLocal() const noexcept { // blind cast
-            return static_cast<TEv*>(Event.Get());
-        }
     };
 
     template <typename TEventType>