snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-10-30 15:36:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make sure that the index-scheduler tick loop is rerun after processing

Browse Source
This commit is contained in:
Kerollmops
2022-10-10 16:19:23 +02:00
committed by Clément Renault
parent b311eb3bed
commit 19c6f8303f
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
index-scheduler/src/lib.rs
View File

@@ -217,7 +217,8 @@ impl IndexScheduler {
run.wake_up.wait();
match run.tick() {
Ok(()) => (),
Ok(0) => (),
Ok(_) => run.wake_up.signal(),
Err(e) => log::error!("{}", e),
}
});
@@ -361,14 +362,16 @@ impl IndexScheduler {
}
/// Create and execute and store the result of one batch of registered tasks.
fn tick(&self) -> Result<()> {
///
/// Returns the number of processed tasks.
fn tick(&self) -> Result<usize> {
#[cfg(test)]
self.test_breakpoint_sdr.send(Breakpoint::Start).unwrap();
let rtxn = self.env.read_txn()?;
let batch = match self.create_next_batch(&rtxn)? {
Some(batch) => batch,
None => return Ok(()),
None => return Ok(0),
};
// we don't need this transaction any longer.
drop(rtxn);
@@ -376,6 +379,7 @@ impl IndexScheduler {
// 1. store the starting date with the bitmap of processing tasks.
let mut ids = batch.ids();
ids.sort_unstable();
let processed_tasks = ids.len();
let processing_tasks = RoaringBitmap::from_sorted_iter(ids.iter().copied()).unwrap();
let started_at = OffsetDateTime::now_utc();
*self.processing_tasks.write().unwrap() = (started_at, processing_tasks);
@@ -401,8 +405,7 @@ impl IndexScheduler {
for mut task in tasks {
task.started_at = Some(started_at);
task.finished_at = Some(finished_at);
// the info field should've been set by the process_batch function
// TODO the info field should've been set by the process_batch function
self.update_task(&mut wtxn, &task)?;
task.remove_data()?;
}
@@ -433,7 +436,7 @@ impl IndexScheduler {
.send(Breakpoint::AfterProcessing)
.unwrap();
Ok(())
Ok(processed_tasks)
}
/// Notify the scheduler there is or may be work to do.