diff --git a/crates/index-scheduler/src/batch.rs b/crates/index-scheduler/src/batch.rs index fc6fb194c..cc730e286 100644 --- a/crates/index-scheduler/src/batch.rs +++ b/crates/index-scheduler/src/batch.rs @@ -496,7 +496,7 @@ impl IndexScheduler { // 5. We make a batch from the unprioritised tasks. Start by taking the next enqueued task. let task_id = if let Some(task_id) = enqueued.min() { task_id } else { return Ok(None) }; - let task = self.get_task(rtxn, task_id)?.ok_or(Error::CorruptedTaskQueue)?; + let mut task = self.get_task(rtxn, task_id)?.ok_or(Error::CorruptedTaskQueue)?; // If the task is not associated with any index, verify that it is an index swap and // create the batch directly. Otherwise, get the index name associated with the task @@ -506,6 +506,7 @@ impl IndexScheduler { index_name } else { assert!(matches!(&task.kind, KindWithContent::IndexSwap { swaps } if swaps.is_empty())); + current_batch.processing(Some(&mut task)); return Ok(Some((Batch::IndexSwap { task }, current_batch))); }; diff --git a/crates/index-scheduler/src/snapshots/lib.rs/swap_indexes/third_empty_swap_processed.snap b/crates/index-scheduler/src/snapshots/lib.rs/swap_indexes/third_empty_swap_processed.snap index 0f126b33a..77b1193a5 100644 --- a/crates/index-scheduler/src/snapshots/lib.rs/swap_indexes/third_empty_swap_processed.snap +++ b/crates/index-scheduler/src/snapshots/lib.rs/swap_indexes/third_empty_swap_processed.snap @@ -89,7 +89,7 @@ succeeded [0,1,2,3,4,5,6,] ---------------------------------------------------------------------- ### Batches Kind: "indexCreation" [0,1,2,3,] -"indexSwap" [4,5,] +"indexSwap" [4,5,6,] ---------------------------------------------------------------------- ### Batches Index Tasks: a [0,4,5,] @@ -104,6 +104,7 @@ d [3,4,] [timestamp] [3,] [timestamp] [4,] [timestamp] [5,] +[timestamp] [6,] ---------------------------------------------------------------------- ### Batches Started At: [timestamp] [0,]