SERVER-46048 IndexBuildsCoordinator::inProgForCollection() allows filtering on single/two phase index builds in a collection

2 files changed, 11 insertions, 0 deletions

diff --git a/src/mongo/db/index_builds_coordinator.cpp b/src/mongo/db/index_builds_coordinator.cpp
index 2abaf9261cb..90226395a44 100644
--- a/src/mongo/db/index_builds_coordinator.cpp
+++ b/src/mongo/db/index_builds_coordinator.cpp
@@ -1105,6 +1105,16 @@ void IndexBuildsCoordinator::dump(std::ostream& ss) const {
     }
 }
 
+bool IndexBuildsCoordinator::inProgForCollection(const UUID& collectionUUID,
+                                                 IndexBuildProtocol protocol) const {
+    stdx::unique_lock<Latch> lk(_mutex);
+    auto indexBuildFilter = [=](const auto& replState) {
+        return collectionUUID == replState.collectionUUID && protocol == replState.protocol;
+    };
+    auto indexBuilds = _filterIndexBuilds_inlock(lk, indexBuildFilter);
+    return !indexBuilds.empty();
+}
+
 bool IndexBuildsCoordinator::inProgForCollection(const UUID& collectionUUID) const {
     stdx::unique_lock<Latch> lk(_mutex);
     return _collectionIndexBuilds.find(collectionUUID) != _collectionIndexBuilds.end();
diff --git a/src/mongo/db/index_builds_coordinator.h b/src/mongo/db/index_builds_coordinator.h
index 20e5d34a741..a538aecf882 100644
--- a/src/mongo/db/index_builds_coordinator.h
+++ b/src/mongo/db/index_builds_coordinator.h
@@ -335,6 +335,7 @@ public:
     /**
      * Returns true if an index build is in progress on the specified collection.
      */
+    bool inProgForCollection(const UUID& collectionUUID, IndexBuildProtocol protocol) const;
     bool inProgForCollection(const UUID& collectionUUID) const;
 
     /**