fix attributes to retrieve bug

This commit is contained in:
mpostma 2020-12-10 15:57:19 +01:00
parent 7a27f9b610
commit 2bd5d2474e
2 changed files with 3 additions and 4 deletions

View File

@ -193,9 +193,7 @@ impl<'a> SearchBuilder<'a> {
.index .index
.document(reader, Some(&all_attributes), doc.id) .document(reader, Some(&all_attributes), doc.id)
.map_err(|e| Error::retrieve_document(doc.id.0, e))? .map_err(|e| Error::retrieve_document(doc.id.0, e))?
.ok_or(Error::internal( .unwrap_or_default();
"Impossible to retrieve the document; Corrupted data",
))?;
let mut formatted = document.iter() let mut formatted = document.iter()
.filter(|(key, _)| all_formatted.contains(key.as_str())) .filter(|(key, _)| all_formatted.contains(key.as_str()))

View File

@ -130,12 +130,13 @@ impl SearchQuery {
restricted_attributes = available_attributes.clone(); restricted_attributes = available_attributes.clone();
} else { } else {
restricted_attributes = HashSet::new(); restricted_attributes = HashSet::new();
search_builder.attributes_to_retrieve(HashSet::new());
for attr in attributes_to_retrieve { for attr in attributes_to_retrieve {
if available_attributes.contains(attr) { if available_attributes.contains(attr) {
restricted_attributes.insert(attr); restricted_attributes.insert(attr);
search_builder.add_retrievable_field(attr.to_string()); search_builder.add_retrievable_field(attr.to_string());
} else { } else {
warn!("The attributes {:?} present in attributesToCrop parameter doesn't exist", attr); warn!("The attributes {:?} present in attributesToRetrieve parameter doesn't exist", attr);
} }
} }
} }