Merge branch 'debug_work'

1 files changed, 18 insertions, 1 deletions

diff --git a/lib/block.cpp b/lib/block.cpp
index 4f7c7c9..85f7654 100644
--- a/lib/block.cpp
+++ b/lib/block.cpp
@@ -3,6 +3,7 @@
 #include "element_impl.hpp"
 #include <gras/block.hpp>
 #include <boost/foreach.hpp>
+#include <boost/thread/thread.hpp> //yield
 
 using namespace gras;
 
@@ -28,7 +29,7 @@ Block::Block(void)
 Block::Block(const std::string &name):
     Element(name)
 {
-    (*this)->block = boost::shared_ptr<BlockActor>(new BlockActor());
+    (*this)->block.reset(new BlockActor());
     (*this)->thread_pool = (*this)->block->thread_pool; //ref copy of pool
     (*this)->block->name = name; //for debug purposes
 
@@ -44,6 +45,22 @@ Block::Block(const std::string &name):
     this->set_buffer_affinity(-1);
 }
 
+void ElementImpl::block_cleanup(void)
+{
+    //wait for actor to chew through enqueued messages
+    while (this->block->GetNumQueuedMessages())
+    {
+        //TODO timeout if this does not stop
+        boost::this_thread::yield();
+    }
+
+    //delete the actor
+    this->block.reset();
+
+    //unref actor's framework
+    this->thread_pool.reset(); //must be deleted after actor
+}
+
 template <typename V, typename T>
 void vector_set(V &v, const T &t, const size_t index)
 {