snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-22 01:57:41 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Make sure we edit the task statuses

Browse Source
This commit is contained in:
Clément Renault 2024-10-14 16:48:15 +02:00
parent 7e1dc8439b
commit 52b95c4e59
No known key found for this signature in database
GPG Key ID: F250A4C4E3AE5F5F
2 changed files with 41 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
index-scheduler/src/batch.rs
View File

@ -1424,6 +1424,8 @@ impl IndexScheduler {
retrieve_or_guess_primary_key(&rtxn, index, &mut new_fields_ids_map, None)?
.unwrap();
let result_count = Ok((candidates.len(), candidates.len())) as Result<_>;
if task.error.is_none() {
/// TODO create a pool if needed
// let pool = indexer_config.thread_pool.unwrap();
@ -1444,42 +1446,42 @@ impl IndexScheduler {
// tracing::info!(indexing_result = ?addition, processed_in = ?started_processing_at.elapsed(), "document indexing done");
}
// let (original_filter, context, function) = if let Some(Details::DocumentEdition {
// original_filter,
// context,
// function,
// ..
// }) = task.details
// {
// (original_filter, context, function)
// } else {
// // In the case of a `documentEdition` the details MUST be set
// unreachable!();
// };
let (original_filter, context, function) = if let Some(Details::DocumentEdition {
original_filter,
context,
function,
..
}) = task.details
{
(original_filter, context, function)
} else {
// In the case of a `documentEdition` the details MUST be set
unreachable!();
};
// match result_count {
// Ok((deleted_documents, edited_documents)) => {
// task.status = Status::Succeeded;
// task.details = Some(Details::DocumentEdition {
// original_filter,
// context,
// function,
// deleted_documents: Some(deleted_documents),
// edited_documents: Some(edited_documents),
// });
// }
// Err(e) => {
// task.status = Status::Failed;
// task.details = Some(Details::DocumentEdition {
// original_filter,
// context,
// function,
// deleted_documents: Some(0),
// edited_documents: Some(0),
// });
// task.error = Some(e.into());
// }
// }
match result_count {
Ok((deleted_documents, edited_documents)) => {
task.status = Status::Succeeded;
task.details = Some(Details::DocumentEdition {
original_filter,
context,
function,
deleted_documents: Some(deleted_documents),
edited_documents: Some(edited_documents),
});
}
Err(e) => {
task.status = Status::Failed;
task.details = Some(Details::DocumentEdition {
original_filter,
context,
function,
deleted_documents: Some(0),
edited_documents: Some(0),
});
task.error = Some(e.into());
}
}
Ok(vec![task])
}

9
milli/src/update/new/indexer/update_by_function.rs
View File

@ -113,16 +113,15 @@ impl<'index> DocumentChanges<'index> for UpdateByFunctionChanges<'index> {
scope.push_constant_dynamic("context", context.clone());
}
scope.push("doc", rhai_document);
// That's were the magic happens. We run the user script
// which edits "doc" scope variable reprensenting the document
// and ignore the output and even the type of it, i.e., Dynamic.
// We run the user script which edits "doc" scope variable reprensenting
// the document and ignore the output and even the type of it, i.e., Dynamic.
let _ = self
.engine
.eval_ast_with_scope::<Dynamic>(&mut scope, &self.ast)
.map_err(UserError::DocumentEditionRuntimeError)?;
match scope.remove::<Dynamic>("doc") {
// If the "doc" variable has set to (), we effectively delete the document.
// If the "doc" variable has been set to (), we effectively delete the document.
Some(doc) if doc.is_unit() => Ok(Some(DocumentChange::Deletion(Deletion::create(
docid,
doc_alloc.alloc_str(&document_id),
@ -142,7 +141,7 @@ impl<'index> DocumentChanges<'index> for UpdateByFunctionChanges<'index> {
//
// Future: Use a custom function rhai function to track changes.
// <https://docs.rs/rhai/latest/rhai/struct.Engine.html#method.register_indexer_set>
if json_document != rhaimap_to_object(new_rhai_document) {
if dbg!(json_document) != dbg!(rhaimap_to_object(new_rhai_document)) {
let mut global_fields_ids_map = new_fields_ids_map.borrow_mut();
let new_document_id = self
.primary_key