Count with upper limit support in Fluent API (#100)

abdolence · May 29, 2023 · 930decc · 930decc
1 parent 5c80226
commit 930decc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/fluent_api/select_aggregation_builder.rs b/src/fluent_api/select_aggregation_builder.rs
@@ -49,6 +49,15 @@ impl FirestoreAggregationFieldExpr {
             FirestoreAggregationOperator::Count(FirestoreAggregationOperatorCount::new()),
         ))
     }
+
+    #[inline]
+    pub fn count_up_to(self, up_to: usize) -> Option<FirestoreAggregation> {
+        Some(FirestoreAggregation::new(self.field_name).with_operator(
+            FirestoreAggregationOperator::Count(
+                FirestoreAggregationOperatorCount::new().with_up_to(up_to),
+            ),
+        ))
+    }
 }
 
 impl FirestoreAggregationExpr for FirestoreAggregation {