SERVER-25062 Implement interruptible sleep and waitFor methods on OperationContext.

1 files changed, 20 insertions, 0 deletions

diff --git a/src/mongo/db/operation_context.cpp b/src/mongo/db/operation_context.cpp
index 8151fe97dfa..d0cb8e244ad 100644
--- a/src/mongo/db/operation_context.cpp
+++ b/src/mongo/db/operation_context.cpp
@@ -210,6 +210,22 @@ Status OperationContext::checkForInterruptNoAssert() {
     return Status::OK();
 }
 
+void OperationContext::sleepUntil(Date_t deadline) {
+    stdx::mutex m;
+    stdx::condition_variable cv;
+    stdx::unique_lock<stdx::mutex> lk(m);
+    invariant(stdx::cv_status::timeout ==
+              waitForConditionOrInterruptUntil(cv, lk, deadline, [] { return false; }));
+}
+
+void OperationContext::sleepFor(Milliseconds duration) {
+    stdx::mutex m;
+    stdx::condition_variable cv;
+    stdx::unique_lock<stdx::mutex> lk(m);
+    invariant(stdx::cv_status::timeout ==
+              waitForConditionOrInterruptFor(cv, lk, duration, [] { return false; }));
+}
+
 void OperationContext::waitForConditionOrInterrupt(stdx::condition_variable& cv,
                                                    stdx::unique_lock<stdx::mutex>& m) {
     uassertStatusOK(waitForConditionOrInterruptNoAssert(cv, m));
@@ -408,4 +424,8 @@ void OperationContext::setLockState(std::unique_ptr<Locker> locker) {
     _locker = std::move(locker);
 }
 
+Date_t OperationContext::getExpirationDateForWaitForValue(Milliseconds waitFor) {
+    return getServiceContext()->getPreciseClockSource()->now() + waitFor;
+}
+
 }  // namespace mongo