Fix bug in RAII jobs, debug session on ts3

3 files changed, 22 insertions, 10 deletions

diff --git a/engine/include/core/priority.hpp b/engine/include/core/priority.hpp
index 1219d26..d6c9b35 100644
--- a/engine/include/core/priority.hpp
+++ b/engine/include/core/priority.hpp
@@ -33,7 +33,7 @@ namespace flat {
                 return lhs.priority() < rhs.priority();
             }
 
-            bool operator()(const std::weak_ptr<prioritized> lhs, const std::weak_ptr<prioritized>& rhs) {
+            bool operator()(const std::weak_ptr<prioritized> lhs, const std::weak_ptr<prioritized> rhs) {
                 if (auto l = lhs.lock()) {
                     if (auto r = rhs.lock()) {
                         // if both valid, check their priority
diff --git a/engine/include/core/task.hpp b/engine/include/core/task.hpp
index d886fcd..570e547 100644
--- a/engine/include/core/task.hpp
+++ b/engine/include/core/task.hpp
@@ -24,9 +24,13 @@ namespace flat {
             callback m_callback;
         };
 
-        struct job : private queue<std::weak_ptr<task>> {
+        struct job : public queue<std::weak_ptr<task>> {
             std::shared_ptr<task> make_task(task::callback f, priority_t p = priority_t::none);
             void invoke_tasks();
+
+            inline void operator()() {
+                invoke_tasks();
+            }
         };
     }
 }
diff --git a/engine/task.cpp b/engine/task.cpp
index f2b6058..c054625 100644
--- a/engine/task.cpp
+++ b/engine/task.cpp
@@ -2,6 +2,8 @@
 
 #include <functional>
 #include <memory>
+#include <algorithm>
+#include <cassert>
 
 
 namespace flat {
@@ -12,20 +14,26 @@ namespace flat {
         std::shared_ptr<task> job::make_task(task::callback f, priority_t p) {
             auto shared = std::make_shared<task>(f, p);
             insert(shared);
+
             return shared;
         }
 
         void job::invoke_tasks() {
-            std::for_each(begin(), end(), [&](auto tp) {
+            // expired tasks to remove
+            std::vector<job::value_type> to_remove;
+
+            for (auto tp : *this) {
                 // check that the task has not expired
-                if (auto t = tp.lock()) {
-                    // run task
+                if (std::shared_ptr<task> t = tp.lock())
                     std::invoke(*t);
-                } else {
-                    // delete task
-                    erase(tp);
-                }
-            });
+                else
+                    to_remove.push_back(tp);
+            }
+
+            // delete expired tasks
+            for (auto tp : to_remove) {
+                erase(tp);
+            }
         }
     }
 }