support setting recovery token in Command

patrickfreed · patrickfreed · commit e2db5d78b88d · 2021-08-02T16:20:33.000-04:00
diff --git a/src/cmap/conn/command.rs b/src/cmap/conn/command.rs
@@ -51,6 +51,8 @@ pub(crate) struct Command<T = Document> {
     autocommit: Option<bool>,
 
     read_concern: Option<ReadConcern>,
+
+    recovery_token: Option<Document>,
 }
 
 impl<T> Command<T> {
@@ -67,6 +69,7 @@ impl<T> Command<T> {
             start_transaction: None,
             autocommit: None,
             read_concern: None,
+            recovery_token: None,
         }
     }
 
@@ -79,7 +82,7 @@ impl<T> Command<T> {
     }
 
     pub(crate) fn set_recovery_token(&mut self, recovery_token: &Document) {
-        self.body.insert("recoveryToken", recovery_token);
+        self.recovery_token = Some(recovery_token.clone());
     }
 
     pub(crate) fn set_txn_number(&mut self, txn_number: i64) {

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,8 @@ pub(crate) struct Command<T = Document> {`
`51`	`51`	`autocommit: Option<bool>,`
`52`	`52`
`53`	`53`	`read_concern: Option<ReadConcern>,`
	`54`	`+`
	`55`	`+ recovery_token: Option<Document>,`
`54`	`56`	`}`
`55`	`57`
`56`	`58`	`impl<T> Command<T> {`
`@@ -67,6 +69,7 @@ impl<T> Command<T> {`
`67`	`69`	`start_transaction: None,`
`68`	`70`	`autocommit: None,`
`69`	`71`	`read_concern: None,`
	`72`	`+ recovery_token: None,`
`70`	`73`	`}`
`71`	`74`	`}`
`72`	`75`
`@@ -79,7 +82,7 @@ impl<T> Command<T> {`
`79`	`82`	`}`
`80`	`83`
`81`	`84`	`pub(crate) fn set_recovery_token(&mut self, recovery_token: &Document) {`
`82`		`- self.body.insert("recoveryToken", recovery_token);`
	`85`	`+ self.recovery_token = Some(recovery_token.clone());`
`83`	`86`	`}`
`84`	`87`
`85`	`88`	`pub(crate) fn set_txn_number(&mut self, txn_number: i64) {`