From 445e5aff02c8d16f8029584e4b18e8d7d3fe83b5 Mon Sep 17 00:00:00 2001 From: Tamo Date: Wed, 15 Jan 2025 12:38:40 +0100 Subject: [PATCH] fix the corruption --- .../index-scheduler/src/index_mapper/mod.rs | 5 ++++ crates/index-scheduler/src/scheduler/test.rs | 24 ++++++++++++++++--- 2 files changed, 26 insertions(+), 3 deletions(-) diff --git a/crates/index-scheduler/src/index_mapper/mod.rs b/crates/index-scheduler/src/index_mapper/mod.rs index d8624d7b9..77cccf9b1 100644 --- a/crates/index-scheduler/src/index_mapper/mod.rs +++ b/crates/index-scheduler/src/index_mapper/mod.rs @@ -191,6 +191,11 @@ impl IndexMapper { self.index_base_map_size, ) .map_err(|e| Error::from_milli(e, Some(uuid.to_string())))?; + let index_rtxn = index.read_txn()?; + let stats = crate::index_mapper::IndexStats::new(&index, &index_rtxn) + .map_err(|e| Error::from_milli(e, Some(name.to_string())))?; + self.store_stats_of(&mut wtxn, name, &stats)?; + drop(index_rtxn); wtxn.commit()?; diff --git a/crates/index-scheduler/src/scheduler/test.rs b/crates/index-scheduler/src/scheduler/test.rs index 6cea6e2e2..de12cb25d 100644 --- a/crates/index-scheduler/src/scheduler/test.rs +++ b/crates/index-scheduler/src/scheduler/test.rs @@ -897,7 +897,25 @@ fn create_and_list_index() { // but the indexUpdate task has not been processed yet handle.advance_till([InsideProcessBatch]); - index_scheduler.index("kefir").unwrap(); // Crash on corrupted task queue - let empty = index_scheduler.get_paginated_indexes_stats(&AuthFilter::default(), 0, 20).unwrap(); - snapshot!(format!("{empty:?}"), @""); + index_scheduler.index("kefir").unwrap(); + let list = index_scheduler.get_paginated_indexes_stats(&AuthFilter::default(), 0, 20).unwrap(); + snapshot!(json_string!(list, { "[1][0][1].created_at" => "[date]", "[1][0][1].updated_at" => "[date]" }), @r#" + [ + 1, + [ + [ + "kefir", + { + "number_of_documents": 0, + "database_size": 24576, + "used_database_size": 8192, + "primary_key": null, + "field_distribution": {}, + "created_at": "[date]", + "updated_at": "[date]" + } + ] + ] + ] + "#); }