snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-09-06 04:36:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make backend change cancelable

Browse Source
This commit is contained in:
Louis Dureuil
2025-09-01 12:10:36 +02:00
parent 231f86decf
commit 8933d87031
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
crates/milli/src/vector/store.rs
View File

@ -104,7 +104,15 @@ impl VectorStore {
} }
} }
pub fn change_backend(&mut self, wtxn: &mut RwTxn, progress: Progress) -> crate::Result<()> { pub fn change_backend<MSP>(
&mut self,
wtxn: &mut RwTxn,
progress: Progress,
must_stop_processing: MSP,
) -> crate::Result<()>
where
MSP: Fn() -> bool + Sync + Send,
{
if self.backend == VectorStoreBackend::Arroy { if self.backend == VectorStoreBackend::Arroy {
self.backend = VectorStoreBackend::Hannoy; self.backend = VectorStoreBackend::Hannoy;
if self.quantized { if self.quantized {
@ -139,6 +147,7 @@ impl VectorStore {
let mut rng = rand::rngs::StdRng::from_entropy(); let mut rng = rand::rngs::StdRng::from_entropy();
let writer = hannoy::Writer::new(self._hannoy_angular_db(), index, dimensions); let writer = hannoy::Writer::new(self._hannoy_angular_db(), index, dimensions);
let mut builder = writer.builder(&mut rng).progress(progress.clone()); let mut builder = writer.builder(&mut rng).progress(progress.clone());
builder.cancel(must_stop_processing);
builder.prepare_arroy_conversion(wtxn)?; builder.prepare_arroy_conversion(wtxn)?;
builder.build::<HANNOY_M, HANNOY_M0>(wtxn)?; builder.build::<HANNOY_M, HANNOY_M0>(wtxn)?;
} }