diff options
Diffstat (limited to 'src/mongo/db/s/range_deleter_service_test.h')
-rw-r--r-- | src/mongo/db/s/range_deleter_service_test.h | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/mongo/db/s/range_deleter_service_test.h b/src/mongo/db/s/range_deleter_service_test.h index 7c689178ab1..cc44ac03225 100644 --- a/src/mongo/db/s/range_deleter_service_test.h +++ b/src/mongo/db/s/range_deleter_service_test.h @@ -75,7 +75,7 @@ public: inline static const BSONObj kShardKeyPattern = BSON(kShardKey << 1); private: - void _setFilteringMetadataWithUUID(OperationContext* opCtx, const UUID& uuid); + void _setFilteringMetadataByUUID(OperationContext* opCtx, const UUID& uuid); // Scoped objects RAIIServerParameterControllerForTest enableFeatureFlag{"featureFlagRangeDeleterService", true}; @@ -87,14 +87,16 @@ RangeDeletionTask createRangeDeletionTask(const UUID& collectionUUID, const BSONObj& min, const BSONObj& max, CleanWhenEnum whenToClean = CleanWhenEnum::kNow, - bool pending = true); + bool pending = true, + boost::optional<KeyPattern> keyPattern = boost::none); std::shared_ptr<RangeDeletionWithOngoingQueries> createRangeDeletionTaskWithOngoingQueries( const UUID& collectionUUID, const BSONObj& min, const BSONObj& max, CleanWhenEnum whenToClean = CleanWhenEnum::kNow, - bool pending = true); + bool pending = true, + boost::optional<KeyPattern> keyPattern = boost::none); SharedSemiFuture<void> registerAndCreatePersistentTask( OperationContext* opCtx, @@ -108,6 +110,9 @@ void verifyRangeDeletionTasks(OperationContext* opCtx, UUID uuidColl, std::vector<ChunkRange> expectedChunkRanges); +/* Unset any filtering metadata associated with the specified collection */ +void _clearFilteringMetadataByUUID(OperationContext* opCtx, const UUID& uuid); + // CRUD operation over `config.rangeDeletions` void insertRangeDeletionTaskDocument(OperationContext* opCtx, const RangeDeletionTask& rdt); void updatePendingField(OperationContext* opCtx, UUID rdtId, bool pending); |