diff options
Diffstat (limited to 'src/mongo/s/write_ops/chunk_manager_targeter.h')
| -rw-r--r-- | src/mongo/s/write_ops/chunk_manager_targeter.h | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/src/mongo/s/write_ops/chunk_manager_targeter.h b/src/mongo/s/write_ops/chunk_manager_targeter.h index d280d6383b9..1e62bc2eeb5 100644 --- a/src/mongo/s/write_ops/chunk_manager_targeter.h +++ b/src/mongo/s/write_ops/chunk_manager_targeter.h @@ -37,6 +37,7 @@ #include "mongo/db/namespace_string.h" #include "mongo/s/catalog_cache.h" #include "mongo/s/ns_targeter.h" +#include "mongo/s/write_ops/batched_command_request.h" namespace mongo { @@ -75,11 +76,11 @@ public: ShardEndpoint targetInsert(OperationContext* opCtx, const BSONObj& doc) const override; - std::vector<ShardEndpoint> targetUpdate( - OperationContext* opCtx, const write_ops::UpdateOpEntry& updateOp) const override; + std::vector<ShardEndpoint> targetUpdate(OperationContext* opCtx, + const BatchItemRef& itemRef) const override; - std::vector<ShardEndpoint> targetDelete( - OperationContext* opCtx, const write_ops::DeleteOpEntry& deleteOp) const override; + std::vector<ShardEndpoint> targetDelete(OperationContext* opCtx, + const BatchItemRef& itemRef) const override; std::vector<ShardEndpoint> targetAllShards(OperationContext* opCtx) const override; @@ -126,9 +127,10 @@ private: * * If 'collation' is empty, we use the collection default collation for targeting. */ - StatusWith<std::vector<ShardEndpoint>> _targetQuery(OperationContext* opCtx, - const BSONObj& query, - const BSONObj& collation) const; + StatusWith<std::vector<ShardEndpoint>> _targetQuery( + boost::intrusive_ptr<ExpressionContext> expCtx, + const BSONObj& query, + const BSONObj& collation) const; /** * Returns a ShardEndpoint for an exact shard key query. |