SERVER-52699 oplog fetcher should be able to handle task executor shutdown case while running query.

2 files changed, 5 insertions, 0 deletions

diff --git a/src/mongo/unittest/task_executor_proxy.cpp b/src/mongo/unittest/task_executor_proxy.cpp
index 823328acd1c..4c2804db3b8 100644
--- a/src/mongo/unittest/task_executor_proxy.cpp
+++ b/src/mongo/unittest/task_executor_proxy.cpp
@@ -62,6 +62,10 @@ SharedSemiFuture<void> TaskExecutorProxy::joinAsync() {
     return _executor.load()->joinAsync();
 }
 
+bool TaskExecutorProxy::isShuttingDown() const {
+    return _executor.load()->isShuttingDown();
+}
+
 void TaskExecutorProxy::appendDiagnosticBSON(mongo::BSONObjBuilder* builder) const {
     _executor.load()->appendDiagnosticBSON(builder);
 }
diff --git a/src/mongo/unittest/task_executor_proxy.h b/src/mongo/unittest/task_executor_proxy.h
index 0e9f56bf892..05a2a11a2a3 100644
--- a/src/mongo/unittest/task_executor_proxy.h
+++ b/src/mongo/unittest/task_executor_proxy.h
@@ -55,6 +55,7 @@ public:
     void shutdown() override;
     void join() override;
     SharedSemiFuture<void> joinAsync() override;
+    bool isShuttingDown() const override;
     void appendDiagnosticBSON(BSONObjBuilder* builder) const override;
     Date_t now() override;
     StatusWith<EventHandle> makeEvent() override;