Cancel automatic pending tasks that are removed during new assignments (

#301) * Cancel automatic pending tasks that are removed during new assignments Signed-off-by: Aaron Chong <[email protected]> * Lint Signed-off-by: Aaron Chong <[email protected]> * Use unordered_set Signed-off-by: Aaron Chong <[email protected]> --------- Signed-off-by: Aaron Chong <[email protected]> (cherry picked from commit fb23496) Signed-off-by: Aaron Chong <[email protected]>
open-rmf · Sep 28, 2023 · d7d2758 · d7d2758
1 parent 09d76b8
commit d7d2758
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/rmf_fleet_adapter/src/rmf_fleet_adapter/TaskManager.cpp b/rmf_fleet_adapter/src/rmf_fleet_adapter/TaskManager.cpp
@@ -922,6 +922,29 @@ void TaskManager::set_queue(
     {
       return;
     }
+
+    // Cancel pending automatic tasks that will be removed from the new
+    // assignments.
+    std::unordered_set<std::string> new_automatic_task_ids;
+    const std::vector<std::string> cancellation_labels =
+    {"New task assignments received."};
+    for (const auto& a : assignments)
+    {
+      if (a.request()->booking()->automatic())
+      {
+        new_automatic_task_ids.insert(a.request()->booking()->id());
+      }
+    }
+    for (const auto& a : _queue)
+    {
+      if (a.request()->booking()->automatic() &&
+        new_automatic_task_ids.find(a.request()->booking()->id()) ==
+        new_automatic_task_ids.end())
+      {
+        _publish_canceled_pending_task(a, cancellation_labels);
+      }
+    }
+
     _queue = assignments;
     _publish_task_queue();
   }