SERVER-29720 Provide a libmongodbcapi database wrapper for service context

1 files changed, 15 insertions, 0 deletions

diff --git a/src/mongo/db/service_context.cpp b/src/mongo/db/service_context.cpp
index 05d89c4cc6f..805a6f0802e 100644
--- a/src/mongo/db/service_context.cpp
+++ b/src/mongo/db/service_context.cpp
@@ -46,6 +46,8 @@ namespace mongo {
 namespace {
 
 ServiceContext* globalServiceContext = nullptr;
+stdx::mutex globalServiceContextMutex;
+stdx::condition_variable globalServiceContextCV;
 
 }  // namespace
 
@@ -58,11 +60,24 @@ ServiceContext* getGlobalServiceContext() {
     return globalServiceContext;
 }
 
+ServiceContext* waitAndGetGlobalServiceContext() {
+    stdx::unique_lock<stdx::mutex> lk(globalServiceContextMutex);
+    globalServiceContextCV.wait(lk, [] { return globalServiceContext; });
+    fassert(40549, globalServiceContext);
+    return globalServiceContext;
+}
+
 void setGlobalServiceContext(std::unique_ptr<ServiceContext>&& serviceContext) {
     fassert(17509, serviceContext.get());
 
     delete globalServiceContext;
 
+    stdx::lock_guard<stdx::mutex> lk(globalServiceContextMutex);
+
+    if (!globalServiceContext) {
+        globalServiceContextCV.notify_all();
+    }
+
     globalServiceContext = serviceContext.release();
 }