add the deletion in the new indexer

This commit is contained in:
Tamo
2025-07-17 18:54:37 +02:00
parent d80edead01
commit 44dc64accb
2 changed files with 18 additions and 4 deletions

View File

@ -32,6 +32,7 @@ pub fn write_to_db(
let _entered = span.enter();
let span = tracing::trace_span!(target: "indexing::write_db", "post_merge");
let mut _entered_post_merge = None;
let cellulite = cellulite::Cellulite::new(index.cellulite);
while let Some(action) = writer_receiver.recv_action() {
if _entered_post_merge.is_none()
&& finished_extraction.load(std::sync::atomic::Ordering::Relaxed)
@ -73,8 +74,14 @@ pub fn write_to_db(
writer.add_item_in_store(wtxn, docid, extractor_id, embedding)?;
}
ReceiverAction::GeoJson(docid, geojson) => {
let cellulite = cellulite::Cellulite::new(index.cellulite);
cellulite.add(wtxn, docid, &geojson).map_err(InternalError::CelluliteError)?;
match geojson {
Some(geojson) => {
cellulite.add(wtxn, docid, &geojson).map_err(InternalError::CelluliteError)?;
}
None => {
cellulite.delete(wtxn, docid).map_err(InternalError::CelluliteError)?;
}
}
}
}