DPL: reset the async queue on STOP

AliceO2Group · Nov 14, 2023 · 6102a52 · 6102a52
1 parent f207642
commit 6102a52
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/Framework/Core/include/Framework/AsyncQueue.h b/Framework/Core/include/Framework/AsyncQueue.h
@@ -59,6 +59,9 @@ struct AsyncQueueHelpers {
   /// 2. then priority
   /// 3. only execute the highest (timeslice, debounce) value
   static void run(AsyncQueue& queue, TimesliceId oldestPossibleTimeslice);
+
+  /// Reset the queue to its initial state
+  static void reset(AsyncQueue& queue);
 };
 
 } // namespace o2::framework

diff --git a/Framework/Core/src/AsyncQueue.cxx b/Framework/Core/src/AsyncQueue.cxx
@@ -107,4 +107,10 @@ auto AsyncQueueHelpers::run(AsyncQueue& queue, TimesliceId oldestPossible) -> vo
                     queue.tasks.end());
 }
 
+auto AsyncQueueHelpers::reset(AsyncQueue& queue) -> void
+{
+  queue.tasks.clear();
+  queue.iteration = 0;
+}
+
 } // namespace o2::framework
diff --git a/Framework/Core/src/CommonServices.cxx b/Framework/Core/src/CommonServices.cxx
@@ -135,6 +135,10 @@ o2::framework::ServiceSpec CommonServices::asyncQueue()
     .name = "async-queue",
     .init = simpleServiceInit<AsyncQueue, AsyncQueue>(),
     .configure = noConfiguration(),
+    .stop = [](ServiceRegistryRef services, void* service) {
+      auto& queue = services.get<AsyncQueue>();
+      AsyncQueueHelpers::reset(queue);
+    },
     .kind = ServiceKind::Serial};
 }