Add index rename feature

crates/index-scheduler/src/index_mapper/mod.rs

@@ -526,6 +526,20 @@ impl IndexMapper {
         Ok(())
     }
 
+    /// Rename an index.
+    pub fn rename(&self, wtxn: &mut RwTxn, current: &str, new: &str) -> Result<()> {
+        let uuid = self
+            .index_mapping
+            .get(wtxn, current)?
+            .ok_or_else(|| Error::IndexNotFound(current.to_string()))?;
+        if self.index_mapping.get(wtxn, new)?.is_some() {
+            return Err(Error::IndexAlreadyExists(new.to_string()));
+        }
+        self.index_mapping.delete(wtxn, current)?;
+        self.index_mapping.put(wtxn, new, &uuid)?;
+        Ok(())
+    }
+
     /// The stats of an index.
     ///
     /// If available in the cache, they are directly returned.

crates/index-scheduler/src/processing.rs

@@ -125,6 +125,12 @@ make_enum_progress! {
     }
 }
 
+make_enum_progress! {
+    pub enum RenameIndexProgress {
+        RenamingTheIndex,
+    }
+}
+
 make_enum_progress! {
     pub enum DeleteIndexProgress {
         DeletingTheIndex,

crates/index-scheduler/src/scheduler/autobatcher.rs

@@ -24,6 +24,7 @@ enum AutobatchKind {
     IndexCreation,
     IndexDeletion,
     IndexUpdate,
+    IndexRename,
     IndexSwap,
 }
 
@@ -67,6 +68,7 @@ impl From<KindWithContent> for AutobatchKind {
             KindWithContent::IndexDeletion { .. } => AutobatchKind::IndexDeletion,
             KindWithContent::IndexCreation { .. } => AutobatchKind::IndexCreation,
             KindWithContent::IndexUpdate { .. } => AutobatchKind::IndexUpdate,
+            KindWithContent::IndexRename { .. } => AutobatchKind::IndexRename,
             KindWithContent::IndexSwap { .. } => AutobatchKind::IndexSwap,
             KindWithContent::TaskCancelation { .. }
             | KindWithContent::TaskDeletion { .. }
@@ -115,6 +117,9 @@ pub enum BatchKind {
     IndexUpdate {
         id: TaskId,
     },
+    IndexRename {
+        id: TaskId,
+    },
     IndexSwap {
         id: TaskId,
     },
@@ -176,6 +181,13 @@ impl BatchKind {
                 )),
                 false,
             ),
+            K::IndexRename => (
+                Break((
+                    BatchKind::IndexRename { id: task_id },
+                    BatchStopReason::TaskCannotBeBatched { kind, id: task_id },
+                )),
+                false,
+            ),
             K::IndexSwap => (
                 Break((
                     BatchKind::IndexSwap { id: task_id },
@@ -288,7 +300,7 @@ impl BatchKind {
 
         match (self, autobatch_kind) {
             // We don't batch any of these operations
-            (this, K::IndexCreation | K::IndexUpdate | K::IndexSwap | K::DocumentEdition) => Break((this, BatchStopReason::TaskCannotBeBatched { kind, id })),
+            (this, K::IndexCreation | K::IndexUpdate | K::IndexRename | K::IndexSwap | K::DocumentEdition) => Break((this, BatchStopReason::TaskCannotBeBatched { kind, id })),
             // We must not batch tasks that don't have the same index creation rights if the index doesn't already exists.
             (this, kind) if !index_already_exists && this.allow_index_creation() == Some(false) && kind.allow_index_creation() == Some(true) => {
                 Break((this, BatchStopReason::IndexCreationMismatch { id }))

crates/index-scheduler/src/scheduler/create_batch.rs

@@ -40,6 +40,11 @@ pub(crate) enum Batch {
         primary_key: Option<String>,
         task: Task,
     },
+    IndexRename {
+        index_uid: String,
+        new_index_uid: String,
+        task: Task,
+    },
     IndexDeletion {
         index_uid: String,
         tasks: Vec<Task>,
@@ -108,7 +113,8 @@ impl Batch {
             | Batch::Dump(task)
             | Batch::IndexCreation { task, .. }
             | Batch::Export { task }
-            | Batch::IndexUpdate { task, .. } => {
+            | Batch::IndexUpdate { task, .. }
+            | Batch::IndexRename { task, .. } => {
                 RoaringBitmap::from_sorted_iter(std::iter::once(task.uid)).unwrap()
             }
             Batch::SnapshotCreation(tasks)
@@ -153,6 +159,7 @@ impl Batch {
             IndexOperation { op, .. } => Some(op.index_uid()),
             IndexCreation { index_uid, .. }
             | IndexUpdate { index_uid, .. }
+            | IndexRename { index_uid, .. }
             | IndexDeletion { index_uid, .. } => Some(index_uid),
         }
     }
@@ -171,6 +178,7 @@ impl fmt::Display for Batch {
             Batch::IndexOperation { op, .. } => write!(f, "{op}")?,
             Batch::IndexCreation { .. } => f.write_str("IndexCreation")?,
             Batch::IndexUpdate { .. } => f.write_str("IndexUpdate")?,
+            Batch::IndexRename { .. } => f.write_str("IndexRename")?,
             Batch::IndexDeletion { .. } => f.write_str("IndexDeletion")?,
             Batch::IndexSwap { .. } => f.write_str("IndexSwap")?,
             Batch::Export { .. } => f.write_str("Export")?,
@@ -411,6 +419,16 @@ impl IndexScheduler {
                 };
                 Ok(Some(Batch::IndexUpdate { index_uid, primary_key, task }))
             }
+            BatchKind::IndexRename { id } => {
+                let mut task =
+                    self.queue.tasks.get_task(rtxn, id)?.ok_or(Error::CorruptedTaskQueue)?;
+                current_batch.processing(Some(&mut task));
+                let (new_uid) = match &task.kind {
+                    KindWithContent::IndexRename { new_index_uid, .. } => new_index_uid.clone(),
+                    _ => unreachable!(),
+                };
+                Ok(Some(Batch::IndexRename { index_uid, new_index_uid: new_uid, task }))
+            }
             BatchKind::IndexDeletion { ids } => Ok(Some(Batch::IndexDeletion {
                 index_uid,
                 index_has_been_created: must_create_index,

crates/index-scheduler/src/scheduler/process_batch.rs

@@ -15,7 +15,7 @@ use super::create_batch::Batch;
 use crate::processing::{
     AtomicBatchStep, AtomicTaskStep, CreateIndexProgress, DeleteIndexProgress, FinalizingIndexStep,
     InnerSwappingTwoIndexes, SwappingTheIndexes, TaskCancelationProgress, TaskDeletionProgress,
-    UpdateIndexProgress,
+    UpdateIndexProgress, RenameIndexProgress,
 };
 use crate::utils::{
     self, remove_n_tasks_datetime_earlier_than, remove_task_datetime, swap_index_uid_in_task,
@@ -229,6 +229,20 @@ impl IndexScheduler {
                     progress,
                 )
             }
+            Batch::IndexRename { index_uid, new_index_uid, mut task } => {
+                progress.update_progress(RenameIndexProgress::RenamingTheIndex);
+                let mut wtxn = self.env.write_txn()?;
+                self.index_mapper.rename(&mut wtxn, &index_uid, &new_index_uid)?;
+                self.queue.tasks.update_index(&mut wtxn, &new_index_uid, |bm| {
+                    let old = self.queue.tasks.index_tasks(&wtxn, &index_uid).unwrap_or_default();
+                    *bm |= &old;
+                })?;
+                self.queue.tasks.update_index(&mut wtxn, &index_uid, |bm| bm.clear())?;
+                wtxn.commit()?;
+                task.status = Status::Succeeded;
+                task.details = Some(Details::IndexRename(IndexRenameDetails { old_uid: index_uid, new_uid: new_index_uid }));
+                Ok((vec![task], ProcessBatchInfo::default()))
+            }
             Batch::IndexUpdate { index_uid, primary_key, mut task } => {
                 progress.update_progress(UpdateIndexProgress::UpdatingTheIndex);
                 let rtxn = self.env.read_txn()?;