From 7d85753573e856c598cc0d317371bd3aca066aea Mon Sep 17 00:00:00 2001
From: Kerollmops <clement@meilisearch.com>
Date: Mon, 4 Sep 2023 17:38:15 +0200
Subject: [PATCH] Make the snapshot download work

---
 index-scheduler/src/lib.rs | 52 +++++++++++++++++++++++---------------
 1 file changed, 32 insertions(+), 20 deletions(-)

diff --git a/index-scheduler/src/lib.rs b/index-scheduler/src/lib.rs
index fb7f1c6e9..e085bc25e 100644
--- a/index-scheduler/src/lib.rs
+++ b/index-scheduler/src/lib.rs
@@ -419,12 +419,11 @@ impl IndexScheduler {
 
                                     log::info!("Downloading the index scheduler database.");
                                     let tasks_snapshot = snapshot_dir.join("tasks.mdb");
-                                    std::fs::copy(&tasks_snapshot, tasks_file).unwrap();
+                                    std::fs::copy(&tasks_snapshot, &tasks_file).unwrap();
 
                                     log::info!("Downloading the indexes databases");
                                     let indexes_files =
-                                        tempfile::TempDir::new_in(&inner.index_mapper.base_path)
-                                            .unwrap();
+                                        tempfile::TempDir::new_in(&base_path).unwrap();
 
                                     let mut indexes = Vec::new();
                                     let src = snapshot_dir.join("indexes");
@@ -437,9 +436,17 @@ impl IndexScheduler {
                                             .unwrap()
                                             .to_string();
                                         log::info!("\tDownloading the index {}", uuid.to_string());
+                                        std::fs::create_dir_all(
+                                            indexes_files.path().join(&uuid).with_extension(""),
+                                        )
+                                        .unwrap();
                                         std::fs::copy(
-                                            src.join(&uuid),
-                                            indexes_files.path().join(&uuid),
+                                            src.join(&uuid).with_extension("mdb"),
+                                            indexes_files
+                                                .path()
+                                                .join(&uuid)
+                                                .with_extension("")
+                                                .join("data.mdb"),
                                         )
                                         .unwrap();
                                         indexes.push(uuid);
@@ -460,24 +467,29 @@ impl IndexScheduler {
                                     pfcs.into_iter().for_each(|pfc| pfc.wait());
 
                                     // Let's replace all the folders/files.
-                                    std::fs::rename(&tasks_snapshot, base_path.join("tasks"))
-                                        .unwrap();
-                                    std::fs::rename(indexes_files, base_path.join("indexes"))
+                                    std::fs::rename(
+                                        &tasks_file,
+                                        base_path.join("tasks").join("data.mdb"),
+                                    )
+                                    .unwrap();
+                                    let dst_indexes = base_path.join("indexes");
+                                    std::fs::remove_dir_all(&dst_indexes).unwrap();
+                                    std::fs::create_dir_all(&dst_indexes).unwrap();
+                                    std::fs::rename(indexes_files.into_path(), dst_indexes)
                                         .unwrap();
 
-                                    // let inner = IndexSchedulerInner::new();
+                                    let mut inner = IndexSchedulerInner::new(
+                                        raw_inner.options,
+                                        #[cfg(test)]
+                                        raw_inner.test_breakpoint_sdr,
+                                        #[cfg(test)]
+                                        raw_inner.planned_failures,
+                                    )
+                                    .unwrap();
 
-                                    *lock = Some(todo!());
-
-                                    // run.env.close();
-
-                                    // 4. Move all the databases
-
-                                    // 5. Unlock the index-mapper
-
-                                    // 2. Download and import the index-scheduler database
-
-                                    // 3. Snapshot every indexes
+                                    // We replace the newly created wake-up signal with the old one
+                                    inner.wake_up = raw_inner.wake_up;
+                                    *lock = Some(inner);
                                 }
                                 otherwise => panic!("{otherwise:?}"),
                             }