diff options
Diffstat (limited to 'src/mongo/db/exec/update_stage.h')
-rw-r--r-- | src/mongo/db/exec/update_stage.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/mongo/db/exec/update_stage.h b/src/mongo/db/exec/update_stage.h index 3080f186e36..5bb55272e9e 100644 --- a/src/mongo/db/exec/update_stage.h +++ b/src/mongo/db/exec/update_stage.h @@ -44,6 +44,7 @@ namespace mongo { class OperationContext; class OpDebug; struct PlanSummaryStats; +class ShardingWriteRouter; struct UpdateStageParams { using DocumentCounter = std::function<size_t(const BSONObj&)>; @@ -186,12 +187,12 @@ private: * If the update changes shard key fields but the new shard key remains on the same node, * returns true. If the update does not change shard key fields, returns false. */ - bool wasExistingShardKeyUpdated(CollectionShardingState* css, + bool wasExistingShardKeyUpdated(const ShardingWriteRouter& shardingWriteRouter, const ScopedCollectionDescription& collDesc, const BSONObj& newObj, const Snapshotted<BSONObj>& oldObj); - bool wasReshardingKeyUpdated(CollectionShardingState* css, + bool wasReshardingKeyUpdated(const ShardingWriteRouter& shardingWriteRouter, const ScopedCollectionDescription& collDesc, const BSONObj& newObj, const Snapshotted<BSONObj>& oldObj); |