add the version to the index-scheduler snapshots + fix a bug when opening an index scheduler for the first time

crates/index-scheduler/src/insta_snapshot.rs

@@ -6,6 +6,7 @@ use meilisearch_types::heed::types::{SerdeBincode, SerdeJson, Str};
 use meilisearch_types::heed::{Database, RoTxn};
 use meilisearch_types::milli::{CboRoaringBitmapCodec, RoaringBitmapCodec, BEU32};
 use meilisearch_types::tasks::{Details, Kind, Status, Task};
+use meilisearch_types::versioning;
 use roaring::RoaringBitmap;
 
 use crate::index_mapper::IndexMapper;
@@ -21,7 +22,7 @@ pub fn snapshot_index_scheduler(scheduler: &IndexScheduler) -> String {
         cleanup_enabled: _,
         processing_tasks,
         env,
-        version: _,
+        version,
         queue,
         scheduler,
 
@@ -39,6 +40,16 @@ pub fn snapshot_index_scheduler(scheduler: &IndexScheduler) -> String {
 
     let mut snap = String::new();
 
+    let indx_sched_version = version.get_version(&rtxn).unwrap();
+    let latest_version = (
+        versioning::VERSION_MAJOR.parse().unwrap(),
+        versioning::VERSION_MINOR.parse().unwrap(),
+        versioning::VERSION_PATCH.parse().unwrap(),
+    );
+    if indx_sched_version != Some(latest_version) {
+        snap.push_str(&format!("index scheduler running on version {indx_sched_version:?}\n"));
+    }
+
     let processing = processing_tasks.read().unwrap().clone();
     snap.push_str(&format!("### Autobatching Enabled = {}\n", scheduler.autobatching_enabled));
     snap.push_str(&format!(

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

@@ -6,6 +6,7 @@ use time::OffsetDateTime;
 use tracing::info;
 
 use crate::queue::TaskQueue;
+use crate::versioning::Versioning;
 
 trait UpgradeIndexScheduler {
     fn upgrade(&self, env: &Env, wtxn: &mut RwTxn, original: (u32, u32, u32))
@@ -15,6 +16,7 @@ trait UpgradeIndexScheduler {
 
 pub fn upgrade_index_scheduler(
     env: &Env,
+    versioning: &Versioning,
     from: (u32, u32, u32),
     to: (u32, u32, u32),
 ) -> anyhow::Result<()> {
@@ -55,6 +57,7 @@ pub fn upgrade_index_scheduler(
         );
         let mut wtxn = env.write_txn()?;
         upgrade.upgrade(env, &mut wtxn, from)?;
+        versioning.set_version(&mut wtxn, target)?;
         wtxn.commit()?;
         current_version = target;
     }

crates/index-scheduler/src/versioning.rs

@@ -54,8 +54,15 @@ impl Versioning {
         let to = (bin_major, bin_minor, bin_patch);
 
         if from != to {
-            upgrade_index_scheduler(env, from, to)?;
+            upgrade_index_scheduler(env, &this, from, to)?;
         }
+
+        // Once we reach this point it means the upgrade process, if there was one is entirely finished
+        // we can safely say we reached the latest version of the index scheduler
+        let mut wtxn = env.write_txn()?;
+        this.set_current_version(&mut wtxn)?;
+        wtxn.commit()?;
+
         Ok(this)
     }
 }