SERVER-25050 Set maxTimeMS on OperationContext on mongos

This change makes mongos commands to retrieve the maxTimeMS from the
command and set it on the OperationContext.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/mongo/db/operation_context.cpp b/src/mongo/db/operation_context.cpp
index 58529dffedf..2a5c8814bf3 100644
--- a/src/mongo/db/operation_context.cpp
+++ b/src/mongo/db/operation_context.cpp
@@ -139,6 +139,15 @@ bool OperationContext::hasDeadlineExpired() const {
     return now >= getDeadline();
 }
 
+Milliseconds OperationContext::getRemainingMaxTimeMillis() const {
+    if (!hasDeadline()) {
+        return Milliseconds::max();
+    }
+
+    return std::max(Milliseconds{0},
+                    getDeadline() - getServiceContext()->getFastClockSource()->now());
+}
+
 Microseconds OperationContext::getRemainingMaxTimeMicros() const {
     if (!hasDeadline()) {
         return Microseconds::max();