SERVER-37560 Add API for allowing commands to support speculative majority reads

This patch lays the groundwork for allowing read queries to take advantage of "speculative" majority reads, which is a mechanism for satisfying majority read guarantees without storage engine support for reading from a historical snapshot. This patch adds a flag on the ReadConcernArgs object to indicate whether a query should use the speculative behavior, and it also adds a method to the CommandInvocation interface that allows commands to optionally support speculative majority reads. The intention is to initially only utilize this behavior for change stream queries i.e. 'aggregate' and 'find' commands, but the feature is, in theory, generic.

2 files changed, 44 insertions, 0 deletions

diff --git a/src/mongo/db/repl/read_concern_args.cpp b/src/mongo/db/repl/read_concern_args.cpp
index a2014bdb1b2..4ae80592c11 100644
--- a/src/mongo/db/repl/read_concern_args.cpp
+++ b/src/mongo/db/repl/read_concern_args.cpp
@@ -262,6 +262,16 @@ Status ReadConcernArgs::initialize(const BSONElement& readConcernElem) {
     return Status::OK();
 }
 
+void ReadConcernArgs::setMajorityReadMechanism(MajorityReadMechanism mechanism) {
+    invariant(*_level == ReadConcernLevel::kMajorityReadConcern);
+    _majorityReadMechanism = mechanism;
+}
+
+ReadConcernArgs::MajorityReadMechanism ReadConcernArgs::getMajorityReadMechanism() const {
+    invariant(*_level == ReadConcernLevel::kMajorityReadConcern);
+    return _majorityReadMechanism;
+}
+
 Status ReadConcernArgs::upconvertReadConcernLevelToSnapshot() {
     if (_level && *_level != ReadConcernLevel::kSnapshotReadConcern &&
         *_level != ReadConcernLevel::kMajorityReadConcern &&
diff --git a/src/mongo/db/repl/read_concern_args.h b/src/mongo/db/repl/read_concern_args.h
index e5f3381cfb4..9cd82318aa1 100644
--- a/src/mongo/db/repl/read_concern_args.h
+++ b/src/mongo/db/repl/read_concern_args.h
@@ -55,6 +55,20 @@ public:
     static const std::string kAtClusterTimeFieldName;
     static const std::string kLevelFieldName;
 
+    /**
+     * Represents the internal mechanism an operation uses to satisfy 'majority' read concern.
+     */
+    enum class MajorityReadMechanism {
+        // The storage engine will read from a historical, majority committed snapshot of data. This
+        // is the default mechanism.
+        kMajoritySnapshot,
+
+        // A mechanism to be used when the storage engine does not support reading from a historical
+        // snapshot. A query will read from a local (potentially uncommitted) snapshot, and, after
+        // reading data, will block until it knows that data has become majority committed.
+        kSpeculative
+    };
+
     static ReadConcernArgs& get(OperationContext* opCtx);
     static const ReadConcernArgs& get(const OperationContext* opCtx);
 
@@ -97,6 +111,20 @@ public:
     Status upconvertReadConcernLevelToSnapshot();
 
     /**
+     * Sets the mechanism we should use to satisfy 'majority' reads.
+     *
+     * Invalid to call unless the read concern level is 'kMajorityReadConcern'.
+     */
+    void setMajorityReadMechanism(MajorityReadMechanism m);
+
+    /**
+     * Returns the mechanism to use for satisfying 'majority' read concern.
+     *
+     * Invalid to call unless the read concern level is 'kMajorityReadConcern'.
+     */
+    MajorityReadMechanism getMajorityReadMechanism() const;
+
+    /**
      * Appends level and afterOpTime.
      */
     void appendInfo(BSONObjBuilder* builder) const;
@@ -152,6 +180,12 @@ private:
      * If the read concern was upconverted, the original read concern level.
      */
     boost::optional<ReadConcernLevel> _originalLevel;
+
+    /**
+     * The mechanism to use for satisfying 'majority' reads. Only meaningful if the read concern
+     * level is 'majority'.
+     */
+    MajorityReadMechanism _majorityReadMechanism{MajorityReadMechanism::kMajoritySnapshot};
 };
 
 }  // namespace repl