Merge pull request #522 from PrimeIntellect-ai/imp/ability-to-delete-all-tasks

feat(orchestrator): ability to delete all tasks from orchestrator

Author: JannikSt

crates/orchestrator/src/api/routes/task.rs

@@ -74,10 +74,20 @@ async fn delete_task(id: web::Path<String>, app_state: Data<AppState>) -> HttpRe
     HttpResponse::Ok().json(json!({"success": true}))
 }
 
+async fn delete_all_tasks(app_state: Data<AppState>) -> HttpResponse {
+    let task_store = app_state.store_context.task_store.clone();
+    if let Err(e) = task_store.delete_all_tasks().await {
+        return HttpResponse::InternalServerError()
+            .json(json!({"success": false, "error": e.to_string()}));
+    }
+    HttpResponse::Ok().json(json!({"success": true}))
+}
+
 pub fn tasks_routes() -> Scope {
     web::scope("/tasks")
         .route("", get().to(get_all_tasks))
         .route("", post().to(create_task))
+        .route("/all", delete().to(delete_all_tasks))
         .route("/{id}", delete().to(delete_task))
 }
 

crates/orchestrator/src/store/domains/task_store.rs

@@ -93,6 +93,34 @@ impl TaskStore {
         Ok(())
     }
 
+    pub async fn delete_all_tasks(&self) -> Result<()> {
+        let mut con = self.redis.client.get_multiplexed_async_connection().await?;
+
+        // Get all tasks first for observer notifications
+        let tasks = self.get_all_tasks().await?;
+
+        // Delete all individual task keys
+        for task in &tasks {
+            let task_key = format!("{}{}", TASK_KEY_PREFIX, task.id);
+            let _: () = con.del(&task_key).await?;
+        }
+
+        // Clear the task list
+        let _: () = con.del(TASK_LIST_KEY).await?;
+
+        // Notify observers synchronously
+        let observers = self.observers.lock().unwrap().clone();
+        for task in tasks {
+            for observer in observers.iter() {
+                if let Err(e) = observer.on_task_deleted(Some(task.clone())) {
+                    error!("Error notifying observer: {}", e);
+                }
+            }
+        }
+
+        Ok(())
+    }
+
     pub async fn get_task(&self, id: &str) -> Result<Option<Task>> {
         let mut con = self.redis.client.get_multiplexed_async_connection().await?;
         let task_key = format!("{}{}", TASK_KEY_PREFIX, id);