SERVER-40406 Add support for $$NOW and $$CLUSTER_TIME in the update command

1 files changed, 16 insertions, 0 deletions

diff --git a/src/mongo/db/ops/update_request.h b/src/mongo/db/ops/update_request.h
index 587fdb1b718..2e84c575f11 100644
--- a/src/mongo/db/ops/update_request.h
+++ b/src/mongo/db/ops/update_request.h
@@ -34,6 +34,7 @@
 #include "mongo/db/logical_session_id.h"
 #include "mongo/db/namespace_string.h"
 #include "mongo/db/ops/write_ops_parsers.h"
+#include "mongo/db/pipeline/runtime_constants_gen.h"
 #include "mongo/db/query/explain.h"
 #include "mongo/util/str.h"
 
@@ -110,6 +111,14 @@ public:
         return _updateMod;
     }
 
+    inline void setRuntimeConstants(const RuntimeConstants& runtimeConstants) {
+        _runtimeConstants = runtimeConstants;
+    }
+
+    inline const boost::optional<RuntimeConstants>& getRuntimeConstants() const {
+        return _runtimeConstants;
+    }
+
     inline void setArrayFilters(const std::vector<BSONObj>& arrayFilters) {
         _arrayFilters = arrayFilters;
     }
@@ -210,6 +219,10 @@ public:
         builder << " updateModification: " << _updateMod.toString();
         builder << " stmtId: " << _stmtId;
 
+        if (_runtimeConstants) {
+            builder << "runtimeConstants: " << _runtimeConstants->toBSON().toString();
+        }
+
         builder << " arrayFilters: [";
         bool first = true;
         for (auto arrayFilter : _arrayFilters) {
@@ -248,6 +261,9 @@ private:
     // Contains the modifiers to apply to matched objects, or a replacement document.
     write_ops::UpdateModification _updateMod;
 
+    // Contains any constant values which may be required by the query or update operation.
+    boost::optional<RuntimeConstants> _runtimeConstants;
+
     // Filters to specify which array elements should be updated.
     std::vector<BSONObj> _arrayFilters;