Adding force stopping mechanism for always online services

Author: SergioMartin86

include/taskr/runtime.hpp

@@ -344,6 +344,9 @@ class Runtime
    */
   __INLINE__ void run()
   {
+    // Clear force terminate flag
+    _forceTerminate = false;
+
     // Verify taskr is correctly initialized and not running
     if (_state == state_t::uninitialized) HICR_THROW_LOGIC("Trying to run TaskR, but it was not initialized");
     if (_state == state_t::running) HICR_THROW_LOGIC("Trying to run TaskR, but it is currently running");
@@ -434,6 +437,15 @@ class Runtime
    */
   __INLINE__ void addService(taskr::service_t *service) { _serviceQueue->push(service); }
 
+
+  /**
+   * Funtion to force termination in case the application does not have its own termination logic
+   */
+  __INLINE__ void forceTermination()
+  {
+    _forceTerminate = true;
+  }
+
   private:
 
   __INLINE__ taskr::Task *serviceWorkerLoop(const workerId_t serviceWorkerId)
@@ -447,7 +459,7 @@ class Runtime
 
     // Getting worker pointer
     auto worker = _serviceWorkers[serviceWorkerId];
-
+    
     // Checking for termination
     auto terminated = checkTermination(worker.get());
 
@@ -493,6 +505,22 @@ class Runtime
     // Getting worker pointer
     auto worker = _taskWorkers[taskWorkerId];
 
+    // If force termination is set, clearing all tasks from the general and worker's queue
+    if (_forceTerminate == true)
+    {
+      while (_commonReadyTaskQueue->wasEmpty() == false)
+      { 
+        auto task = _commonReadyTaskQueue->pop();
+        if (task != nullptr) _activeTaskCount--;
+      } 
+
+      while (worker->getReadyTaskQueue()->wasEmpty() == false)
+      { 
+        auto task = worker->getReadyTaskQueue()->pop();
+        if (task != nullptr) _activeTaskCount--;
+      } 
+    }
+
     // If required, perform a service task
     if (_makeTaskWorkersRunServices == true)
     {
@@ -723,6 +751,14 @@ class Runtime
     this->_workerCallbackMap.trigger(taskrWorker, HiCR::tasking::Worker::callback_t::onWorkerTerminate);
   }
 
+
+
+  /**
+   * Flag to indicate whether execution must be forcibly terminated. It is discouraged to use this if the application
+   * has implemented a clear ending logic. This is only useful for always-on services-like applications
+   */
+  bool _forceTerminate;
+
   /**
    * A flag to indicate whether taskR was initialized
    */