raw_run_command

Author: beckerinj

src/action/run_command.rs

@@ -1,6 +1,6 @@
 use std::time::Duration;
 
-use bson::{Bson, Document};
+use bson::{Bson, Document, RawDocumentBuf};
 
 use crate::{
     client::session::TransactionState,
@@ -9,19 +9,11 @@ use crate::{
     error::{ErrorKind, Result},
     operation::{run_command, run_cursor_command},
     selection_criteria::SelectionCriteria,
-    ClientSession,
-    Cursor,
-    Database,
-    SessionCursor,
+    ClientSession, Cursor, Database, SessionCursor,
 };
 
 use super::{
-    action_impl,
-    deeplink,
-    export_doc,
-    option_setters,
-    options_doc,
-    ExplicitSession,
+    action_impl, deeplink, export_doc, option_setters, options_doc, ExplicitSession,
     ImplicitSession,
 };
 
@@ -46,6 +38,26 @@ impl Database {
         }
     }
 
+    /// Runs a database-level command.
+    ///
+    /// Note that no inspection is done on `doc`, so the command will not use the database's default
+    /// read concern or write concern. If specific read concern or write concern is desired, it must
+    /// be specified manually.
+    /// Please note that run_command doesn't validate WriteConcerns passed into the body of the
+    /// command document.
+    ///
+    /// `await` will return d[`Result<Document>`].
+    #[deeplink]
+    #[options_doc(run_command)]
+    pub fn raw_run_command(&self, command: RawDocumentBuf) -> RawRunCommand {
+        RawRunCommand {
+            db: self,
+            command,
+            options: None,
+            session: None,
+        }
+    }
+
     /// Runs a database-level command and returns a cursor to the response.
     ///
     /// `await` will return d[`Result<Cursor<Document>>`] or a
@@ -152,6 +164,68 @@ impl<'a> Action for RunCommand<'a> {
     }
 }
 
+/// Run a database-level command.  Create with [`Database::run_command`].
+#[must_use]
+pub struct RawRunCommand<'a> {
+    db: &'a Database,
+    command: RawDocumentBuf,
+    options: Option<RunCommandOptions>,
+    session: Option<&'a mut ClientSession>,
+}
+
+#[option_setters(crate::db::options::RunCommandOptions)]
+#[export_doc(run_raw_command)]
+impl<'a> RawRunCommand<'a> {
+    /// Run the command using the provided [`ClientSession`].
+    pub fn session(mut self, value: impl Into<&'a mut ClientSession>) -> Self {
+        self.session = Some(value.into());
+        self
+    }
+}
+
+#[action_impl]
+impl<'a> Action for RawRunCommand<'a> {
+    type Future = RunRawCommandFuture;
+
+    async fn execute(self) -> Result<Document> {
+        let mut selection_criteria = self.options.and_then(|o| o.selection_criteria);
+        if let Some(session) = &self.session {
+            match session.transaction.state {
+                TransactionState::Starting | TransactionState::InProgress => {
+                    if self.command.get("readConcern").is_ok() {
+                        return Err(ErrorKind::InvalidArgument {
+                            message: "Cannot set read concern after starting a transaction".into(),
+                        }
+                        .into());
+                    }
+                    selection_criteria = match selection_criteria {
+                        Some(selection_criteria) => Some(selection_criteria),
+                        None => {
+                            if let Some(ref options) = session.transaction.options {
+                                options.selection_criteria.clone()
+                            } else {
+                                None
+                            }
+                        }
+                    };
+                }
+                _ => {}
+            }
+        }
+
+        let operation = run_command::RunCommand::new_raw(
+            self.db.name().into(),
+            self.command,
+            selection_criteria,
+            None,
+        )?;
+        self.db
+            .client()
+            .execute_operation(operation, self.session)
+            .await
+    }
+}
+
 /// Runs a database-level command and returns a cursor to the response.  Create with
 /// [`Database::run_cursor_command`].
 #[must_use]

src/operation/run_command.rs

@@ -33,7 +33,6 @@ impl<'conn> RunCommand<'conn> {
         })
     }
 
-    #[cfg(feature = "in-use-encryption")]
     pub(crate) fn new_raw(
         db: String,
         command: RawDocumentBuf,