Attempt to call shutdown

nv-morpheus · Oct 6, 2023 · e3f243f · e3f243f
1 parent b068686
commit e3f243f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/cpp/mrc/include/mrc/node/rx_runnable.hpp b/cpp/mrc/include/mrc/node/rx_runnable.hpp
@@ -33,6 +33,8 @@ class RxRunnable : public runnable::RunnableWithContext<ContextT>, public RxSubs
     RxRunnable()           = default;
     ~RxRunnable() override = default;
 
+    void shutdown(ContextT& ctx);
+
   private:
     // implemented by node objects and will be final
     virtual void on_shutdown_critical_section()      = 0;
@@ -54,8 +56,6 @@ class RxRunnable : public runnable::RunnableWithContext<ContextT>, public RxSubs
     void run(ContextT& ctx) final;
     void on_state_update(const state_t& state) final;
 
-    void shutdown(ContextT& ctx);
-
     rxcpp::composite_subscription m_subscription;
 };
 

diff --git a/cpp/mrc/src/internal/segment/builder_definition.cpp b/cpp/mrc/src/internal/segment/builder_definition.cpp
@@ -299,6 +299,11 @@ void BuilderDefinition::shutdown()
         if (obj_prop->is_source() && !obj_prop->is_sink())
         {
             DVLOG(10) << "Destroying: " << name;
+            auto& ctx          = Context<>::get_runtime_context();
+            auto source_object = std::dynamic_pointer_cast<Object<node::RxRunnable<decltype(ctx)>>>(obj_prop);
+            auto& runnable     = source_object->object();
+            runnable.shutdown(ctx);
+
             obj_prop->destroy();
         }
     }