snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 18:45:06 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #484 from meilisearch/fix-reindex-by-chunk

Browse Source 
Stop reindexing by chunk during complete reindexing
This commit is contained in:
Clément Renault 2020-02-28 18:29:25 +01:00 committed by GitHub
commit 72450c765d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 34 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
meilisearch-core/src/update/documents_addition.rs
View File

@ -279,19 +279,17 @@ pub fn reindex_all_documents(writer: &mut heed::RwTxn<MainT>, index: &store::Ind
index.postings_lists.clear(writer)?; index.postings_lists.clear(writer)?;
index.docs_words.clear(writer)?; index.docs_words.clear(writer)?;
// 3. re-index chunks of documents (otherwise we make the borrow checker unhappy)
for documents_ids in documents_ids_to_reindex.chunks(100) {
let stop_words = match index.main.stop_words_fst(writer)? { let stop_words = match index.main.stop_words_fst(writer)? {
Some(stop_words) => stop_words, Some(stop_words) => stop_words,
None => fst::Set::default(), None => fst::Set::default(),
}; };
let number_of_inserted_documents = documents_ids.len(); let number_of_inserted_documents = documents_ids_to_reindex.len();
let mut indexer = RawIndexer::new(stop_words); let mut indexer = RawIndexer::new(stop_words);
let mut ram_store = HashMap::new(); let mut ram_store = HashMap::new();
for document_id in documents_ids { for document_id in documents_ids_to_reindex {
for result in index.documents_fields.document_fields(writer, *document_id)? { for result in index.documents_fields.document_fields(writer, document_id)? {
let (field_id, bytes) = result?; let (field_id, bytes) = result?;
let value: serde_json::Value = serde_json::from_slice(bytes)?; let value: serde_json::Value = serde_json::from_slice(bytes)?;
ram_store.insert((document_id, field_id), value); ram_store.insert((document_id, field_id), value);
@ -302,7 +300,7 @@ pub fn reindex_all_documents(writer: &mut heed::RwTxn<MainT>, index: &store::Ind
writer, writer,
field_id, field_id,
&schema, &schema,
*docid, docid,
index.documents_fields, index.documents_fields,
index.documents_fields_counts, index.documents_fields_counts,
&mut indexer, &mut indexer,
@ -320,7 +318,6 @@ pub fn reindex_all_documents(writer: &mut heed::RwTxn<MainT>, index: &store::Ind
number_of_inserted_documents, number_of_inserted_documents,
indexer, indexer,
)?; )?;
}
index.main.put_schema(writer, &schema)?; index.main.put_schema(writer, &schema)?;