Consume updates in the order of insertion

This commit is contained in:
Clément Renault 2019-10-07 15:00:28 +02:00
parent 487411340a
commit 0615c5c52d
No known key found for this signature in database
GPG Key ID: 92ADA4E935E71FA4
2 changed files with 85 additions and 68 deletions

View File

@ -33,6 +33,23 @@ impl Updates {
Ok(Some((number, last_data)))
}
fn first_update_id<'a>(
&self,
reader: &'a impl rkv::Readable,
) -> Result<Option<(u64, Option<Value<'a>>)>, rkv::StoreError>
{
let mut iter = self.updates.iter_start(reader)?;
let (first_key, first_data) = match iter.next() {
Some(result) => result?,
None => return Ok(None),
};
let array = first_key.try_into().unwrap();
let number = u64::from_be_bytes(array);
Ok(Some((number, first_data)))
}
pub fn contains(
&self,
reader: &impl rkv::Readable,
@ -60,12 +77,12 @@ impl Updates {
Ok(last_update_id)
}
pub fn pop_back(
pub fn pop_front(
&self,
writer: &mut rkv::Writer,
) -> MResult<Option<(u64, Update)>>
{
let (last_id, last_data) = match self.last_update_id(writer)? {
let (last_id, last_data) = match self.first_update_id(writer)? {
Some(entry) => entry,
None => return Ok(None),
};

View File

@ -4,10 +4,7 @@ mod documents_deletion;
pub use self::documents_addition::{DocumentsAddition, apply_documents_addition};
pub use self::documents_deletion::{DocumentsDeletion, apply_documents_deletion};
use std::collections::BTreeMap;
use std::sync::{Arc, RwLock};
use std::time::{Duration, Instant};
use serde::{Serialize, Deserialize};
use crate::{store, Error, MResult, DocumentId, RankedMap};
@ -63,7 +60,7 @@ pub fn update_status<T: rkv::Readable>(
}
pub fn push_documents_addition<D: serde::Serialize>(
mut writer: &mut rkv::Writer,
writer: &mut rkv::Writer,
updates_store: store::Updates,
addition: Vec<D>,
) -> Result<u64, Error>
@ -97,9 +94,13 @@ pub fn update_task(
writer: &mut rkv::Writer,
index: store::Index,
mut callback: Option<impl FnOnce(UpdateResult)>,
) -> MResult<()>
) -> MResult<bool>
{
if let Some((update_id, update)) = index.updates.pop_back(writer)? {
let (update_id, update) = match index.updates.pop_front(writer)? {
Some(value) => value,
None => return Ok(false),
};
let (update_type, result, duration) = match update {
Update::DocumentsAddition(documents) => {
let update_type = UpdateType::DocumentsAddition { number: documents.len() };
@ -168,7 +169,6 @@ pub fn update_task(
if let Some(callback) = callback.take() {
(callback)(status);
}
}
Ok(())
Ok(true)
}