Explain options support

abdolence · Apr 2, 2024 · 94dd224 · 94dd224
1 parent 9ff2ed7
commit 94dd224
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/db/query.rs b/src/db/query.rs
@@ -85,8 +85,12 @@ impl FirestoreDb {
                 .as_ref()
                 .map(|selector| selector.try_into())
                 .transpose()?,
+            explain_options: params
+                .explain_options
+                .as_ref()
+                .map(|eo| eo.try_into())
+                .transpose()?,
             query_type: Some(run_query_request::QueryType::StructuredQuery(params.into())),
-            explain_options: None,
         }))
     }
 

diff --git a/src/db/query_models.rs b/src/db/query_models.rs
@@ -1,5 +1,6 @@
 #![allow(clippy::derive_partial_eq_without_eq)] // Since we may not be able to implement Eq for the changes coming from Firestore protos
 
+use crate::errors::FirestoreError;
 use crate::FirestoreValue;
 use gcloud_sdk::google::firestore::v1::*;
 use rsb_derive::Builder;
@@ -37,6 +38,7 @@ pub struct FirestoreQueryParams {
     pub return_only_fields: Option<Vec<String>>,
     pub start_at: Option<FirestoreQueryCursor>,
     pub end_at: Option<FirestoreQueryCursor>,
+    pub explain_options: Option<FirestoreExplainOptions>,
 }
 
 impl From<FirestoreQueryParams> for StructuredQuery {
@@ -409,3 +411,17 @@ pub struct FirestorePartition {
     pub start_at: Option<FirestoreQueryCursor>,
     pub end_at: Option<FirestoreQueryCursor>,
 }
+
+#[derive(Debug, PartialEq, Clone, Builder)]
+pub struct FirestoreExplainOptions {
+    pub analyze: Option<bool>,
+}
+
+impl TryFrom<&FirestoreExplainOptions> for gcloud_sdk::google::firestore::v1::ExplainOptions {
+    type Error = FirestoreError;
+    fn try_from(explain_options: &FirestoreExplainOptions) -> Result<Self, Self::Error> {
+        Ok(ExplainOptions {
+            analyze: explain_options.analyze.unwrap_or(false),
+        })
+    }
+}