SERVER-29794 Adding a comment to all commands

1 files changed, 13 insertions, 0 deletions

diff --git a/src/mongo/db/operation_context.h b/src/mongo/db/operation_context.h
index a834f89ae1d..43553efae9b 100644
--- a/src/mongo/db/operation_context.h
+++ b/src/mongo/db/operation_context.h
@@ -373,6 +373,15 @@ public:
         _inMultiDocumentTransaction = true;
     }
 
+    void setComment(const BSONObj& comment) {
+        _comment = comment.getOwned();
+    }
+
+    boost::optional<BSONElement> getComment() {
+        // The '_comment' object, if present, will only ever have one field.
+        return _comment ? boost::optional<BSONElement>(_comment->firstElement()) : boost::none;
+    }
+
 private:
     IgnoreInterruptsState pushIgnoreInterrupts() override {
         IgnoreInterruptsState iis{_ignoreInterrupts,
@@ -498,6 +507,10 @@ private:
     bool _writesAreReplicated = true;
     bool _shouldParticipateInFlowControl = true;
     bool _inMultiDocumentTransaction = false;
+
+    // If populated, this is an owned singleton BSONObj whose only field, 'comment', is a copy of
+    // the 'comment' field from the input command object.
+    boost::optional<BSONObj> _comment;
 };
 
 namespace repl {