SERVER-45029 Add getCurrentServiceContext()

(cherry picked from commit 0649034e411fc20d26efce791af276e727db62d5)

2 files changed, 18 insertions, 0 deletions

diff --git a/src/mongo/db/service_context.cpp b/src/mongo/db/service_context.cpp
index c3d486fa100..f14fce1404c 100644
--- a/src/mongo/db/service_context.cpp
+++ b/src/mongo/db/service_context.cpp
@@ -72,6 +72,15 @@ ServiceContext* getGlobalServiceContext() {
     return globalServiceContext;
 }
 
+ServiceContext* getCurrentServiceContext() {
+    auto client = Client::getCurrent();
+    if (client) {
+        return client->getServiceContext();
+    }
+
+    return nullptr;
+}
+
 void setGlobalServiceContext(ServiceContext::UniqueServiceContext&& serviceContext) {
     if (globalServiceContext) {
         // Make sure that calling getGlobalServiceContext() during the destructor results in
diff --git a/src/mongo/db/service_context.h b/src/mongo/db/service_context.h
index fb350baad55..792066d571d 100644
--- a/src/mongo/db/service_context.h
+++ b/src/mongo/db/service_context.h
@@ -609,6 +609,15 @@ bool hasGlobalServiceContext();
 ServiceContext* getGlobalServiceContext();
 
 /**
+ * Returns the ServiceContext associated with the current Client.
+ *
+ * Returns a nullptr if there is not a current Client
+ *
+ * Caller does not own pointer.
+ */
+ServiceContext* getCurrentServiceContext();
+
+/**
  * Sets the global ServiceContext.  If 'serviceContext' is NULL, un-sets and deletes
  * the current global ServiceContext.
  *