SERVER-45778 Rename getOrphanFilter to getOwnershipFilter and change the return type

1 files changed, 30 insertions, 1 deletions

diff --git a/src/mongo/db/s/scoped_collection_metadata.h b/src/mongo/db/s/scoped_collection_metadata.h
index 31ef088066f..946170074c4 100644
--- a/src/mongo/db/s/scoped_collection_metadata.h
+++ b/src/mongo/db/s/scoped_collection_metadata.h
@@ -64,8 +64,37 @@ public:
         return get();
     }
 
-private:
+    bool isSharded() const {
+        return _impl->get().isSharded();
+    }
+
+    const BSONObj& getKeyPattern() const {
+        return _impl->get().getKeyPattern();
+    }
+
+    const BSONObj extractShardKeyFromDoc(const BSONObj& doc) const {
+        return _impl->get().getChunkManager()->getShardKeyPattern().extractShardKeyFromDoc(doc);
+    }
+
+protected:
     std::shared_ptr<Impl> _impl;
 };
 
+class ScopedCollectionFilter : public ScopedCollectionMetadata {
+public:
+    ScopedCollectionFilter(std::shared_ptr<Impl> impl)
+        : ScopedCollectionMetadata(std::move(impl)) {}
+
+    ScopedCollectionFilter(ScopedCollectionMetadata&& scopedMetadata)
+        : ScopedCollectionMetadata(std::move(scopedMetadata)) {}
+
+    bool keyBelongsToMe(const BSONObj& key) const {
+        return _impl->get().keyBelongsToMe(key);
+    }
+
+    Chunk findIntersectingChunkWithSimpleCollation(const BSONObj& shardKey) const {
+        return _impl->get().getChunkManager()->findIntersectingChunkWithSimpleCollation(shardKey);
+    }
+};
+
 }  // namespace mongo