format the whole project

milli/src/search/distinct/facet_distinct.rs

@@ -3,11 +3,11 @@ use std::mem::size_of;
 use heed::types::ByteSlice;
 use roaring::RoaringBitmap;
 
+use super::{Distinct, DocIter};
 use crate::error::InternalError;
 use crate::heed_codec::facet::*;
 use crate::index::db_name;
 use crate::{DocumentId, FieldId, Index, Result};
-use super::{Distinct, DocIter};
 
 const FID_SIZE: usize = size_of::<FieldId>();
 const DOCID_SIZE: usize = size_of::<DocumentId>();
@@ -28,11 +28,7 @@ pub struct FacetDistinct<'a> {
 
 impl<'a> FacetDistinct<'a> {
     pub fn new(distinct: FieldId, index: &'a Index, txn: &'a heed::RoTxn<'a>) -> Self {
-        Self {
-            distinct,
-            index,
-            txn,
-        }
+        Self { distinct, index, txn }
     }
 }
 
@@ -47,16 +43,12 @@ pub struct FacetDistinctIter<'a> {
 
 impl<'a> FacetDistinctIter<'a> {
     fn facet_string_docids(&self, key: &str) -> heed::Result<Option<RoaringBitmap>> {
-        self.index
-            .facet_id_string_docids
-            .get(self.txn, &(self.distinct, key))
+        self.index.facet_id_string_docids.get(self.txn, &(self.distinct, key))
     }
 
     fn facet_number_docids(&self, key: f64) -> heed::Result<Option<RoaringBitmap>> {
         // get facet docids on level 0
-        self.index
-            .facet_id_f64_docids
-            .get(self.txn, &(self.distinct, 0, key, key))
+        self.index.facet_id_f64_docids.get(self.txn, &(self.distinct, 0, key, key))
     }
 
     fn distinct_string(&mut self, id: DocumentId) -> Result<()> {
@@ -64,9 +56,8 @@ impl<'a> FacetDistinctIter<'a> {
 
         for item in iter {
             let ((_, _, value), _) = item?;
-            let facet_docids = self
-                .facet_string_docids(value)?
-                .ok_or(InternalError::DatabaseMissingEntry {
+            let facet_docids =
+                self.facet_string_docids(value)?.ok_or(InternalError::DatabaseMissingEntry {
                     db_name: db_name::FACET_ID_STRING_DOCIDS,
                     key: None,
                 })?;
@@ -83,9 +74,8 @@ impl<'a> FacetDistinctIter<'a> {
 
         for item in iter {
             let ((_, _, value), _) = item?;
-            let facet_docids = self
-                .facet_number_docids(value)?
-                .ok_or(InternalError::DatabaseMissingEntry {
+            let facet_docids =
+                self.facet_number_docids(value)?.ok_or(InternalError::DatabaseMissingEntry {
                     db_name: db_name::FACET_ID_F64_DOCIDS,
                     key: None,
                 })?;

milli/src/search/distinct/mod.rs

@@ -1,11 +1,11 @@
 mod facet_distinct;
 mod noop_distinct;
 
+pub use facet_distinct::FacetDistinct;
+pub use noop_distinct::NoopDistinct;
 use roaring::RoaringBitmap;
 
 use crate::{DocumentId, Result};
-pub use facet_distinct::FacetDistinct;
-pub use noop_distinct::NoopDistinct;
 
 /// A trait implemented by document interators that are returned by calls to `Distinct::distinct`.
 /// It provides a way to get back the ownership to the excluded set.
@@ -29,13 +29,15 @@ mod test {
     use std::collections::HashSet;
 
     use once_cell::sync::Lazy;
-    use rand::{seq::SliceRandom, Rng};
+    use rand::seq::SliceRandom;
+    use rand::Rng;
     use roaring::RoaringBitmap;
     use serde_json::{json, Value};
 
-    use crate::index::{Index, tests::TempIndex};
+    use crate::index::tests::TempIndex;
+    use crate::index::Index;
     use crate::update::{IndexDocumentsMethod, UpdateBuilder, UpdateFormat};
-    use crate::{BEU32, FieldId, DocumentId};
+    use crate::{DocumentId, FieldId, BEU32};
 
     static JSON: Lazy<Value> = Lazy::new(generate_json);
 
@@ -89,9 +91,7 @@ mod test {
         addition.index_documents_method(IndexDocumentsMethod::ReplaceDocuments);
         addition.update_format(UpdateFormat::Json);
 
-        addition
-            .execute(JSON.to_string().as_bytes(), |_, _| ())
-            .unwrap();
+        addition.execute(JSON.to_string().as_bytes(), |_, _| ()).unwrap();
 
         let fields_map = index.fields_ids_map(&txn).unwrap();
         let fid = fields_map.id(&distinct).unwrap();
@@ -103,13 +103,12 @@ mod test {
         (index, fid, map)
     }
 
-
     /// Checks that all the candidates are distinct, and returns the candidates number.
     pub(crate) fn validate_distinct_candidates(
         candidates: impl Iterator<Item = crate::Result<DocumentId>>,
         distinct: FieldId,
         index: &Index,
-        ) -> usize {
+    ) -> usize {
         fn test(seen: &mut HashSet<String>, value: &Value) {
             match value {
                 Value::Null | Value::Object(_) | Value::Bool(_) => (),
@@ -117,7 +116,7 @@ mod test {
                     let s = value.to_string();
                     assert!(seen.insert(s));
                 }
-                Value::Array(values) => {values.into_iter().for_each(|value| test(seen, value))}
+                Value::Array(values) => values.into_iter().for_each(|value| test(seen, value)),
             }
         }
 

milli/src/search/distinct/noop_distinct.rs

@@ -1,7 +1,8 @@
-use roaring::{RoaringBitmap, bitmap::IntoIter};
+use roaring::bitmap::IntoIter;
+use roaring::RoaringBitmap;
 
+use super::{Distinct, DocIter};
 use crate::{DocumentId, Result};
-use super::{DocIter, Distinct};
 
 /// A distinct implementer that does not perform any distinct,
 /// and simply returns an iterator to the candidates.
@@ -30,10 +31,7 @@ impl Distinct for NoopDistinct {
     type Iter = NoopDistinctIter;
 
     fn distinct(&mut self, candidates: RoaringBitmap, excluded: RoaringBitmap) -> Self::Iter {
-        NoopDistinctIter {
-            candidates: candidates.into_iter(),
-            excluded,
-        }
+        NoopDistinctIter { candidates: candidates.into_iter(), excluded }
     }
 }