From 737b891a41ac2deb0954da88554bfda398afa001 Mon Sep 17 00:00:00 2001 From: ad hoc Date: Tue, 17 May 2022 10:58:15 +0200 Subject: [PATCH] introduce Dump TaskListIdentifier variant --- meilisearch-lib/src/index_controller/mod.rs | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/meilisearch-lib/src/index_controller/mod.rs b/meilisearch-lib/src/index_controller/mod.rs index 7ba91dfca..4be90489a 100644 --- a/meilisearch-lib/src/index_controller/mod.rs +++ b/meilisearch-lib/src/index_controller/mod.rs @@ -76,6 +76,7 @@ pub struct IndexController { index_resolver: Arc>, scheduler: Arc>, task_store: TaskStore, + dump_path: PathBuf, dump_handle: dump_actor::DumpActorHandleImpl, update_file_store: UpdateFileStore, } @@ -89,6 +90,7 @@ impl Clone for IndexController { dump_handle: self.dump_handle.clone(), update_file_store: self.update_file_store.clone(), task_store: self.task_store.clone(), + dump_path: self.dump_path.clone(), } } } @@ -234,7 +236,7 @@ impl IndexControllerBuilder { receiver, update_file_store.clone(), scheduler.clone(), - dump_path, + dump_path.clone(), analytics_path, index_size, task_store_size, @@ -269,6 +271,7 @@ impl IndexControllerBuilder { index_resolver, scheduler, dump_handle, + dump_path, update_file_store, task_store, }) @@ -425,6 +428,15 @@ where Ok(task) } + pub async fn register_dump_task(&self) -> Result { + let content = TaskContent::Dump { + path: self.dump_path.clone(), + }; + let task = self.task_store.register(None, content).await?; + self.scheduler.read().await.notify(); + Ok(task) + } + pub async fn get_task(&self, id: TaskId, filter: Option) -> Result { let task = self.scheduler.read().await.get_task(id, filter).await?; Ok(task)