summaryrefslogtreecommitdiff
path: root/src/mongo/dbtests/query_stage_multiplan.cpp
diff options
context:
space:
mode:
authorRuoxin Xu <ruoxin.xu@mongodb.com>2021-02-02 21:03:25 +0000
committerEvergreen Agent <no-reply@evergreen.mongodb.com>2021-02-17 11:23:05 +0000
commit5ada96cbe318734f9c7dfa26dea3f28dbe4627bf (patch)
tree5a63211f065b1e40e5eeeb239d241355e1720ffa /src/mongo/dbtests/query_stage_multiplan.cpp
parentd4ff82a11019aef87701db9053499461601e75d6 (diff)
downloadmongo-5ada96cbe318734f9c7dfa26dea3f28dbe4627bf.tar.gz
SERVER-53060 Remove QueryRequest class
Diffstat (limited to 'src/mongo/dbtests/query_stage_multiplan.cpp')
-rw-r--r--src/mongo/dbtests/query_stage_multiplan.cpp46
1 files changed, 23 insertions, 23 deletions
diff --git a/src/mongo/dbtests/query_stage_multiplan.cpp b/src/mongo/dbtests/query_stage_multiplan.cpp
index f4300805bd2..5ef147ed590 100644
--- a/src/mongo/dbtests/query_stage_multiplan.cpp
+++ b/src/mongo/dbtests/query_stage_multiplan.cpp
@@ -128,9 +128,9 @@ protected:
std::unique_ptr<CanonicalQuery> makeCanonicalQuery(OperationContext* opCtx,
NamespaceString nss,
BSONObj filter) {
- auto qr = std::make_unique<QueryRequest>(nss);
- qr->setFilter(filter);
- auto statusWithCQ = CanonicalQuery::canonicalize(opCtx, std::move(qr));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(filter);
+ auto statusWithCQ = CanonicalQuery::canonicalize(opCtx, std::move(findCommand));
ASSERT_OK(statusWithCQ.getStatus());
unique_ptr<CanonicalQuery> cq = std::move(statusWithCQ.getValue());
ASSERT(cq);
@@ -374,10 +374,10 @@ TEST_F(QueryStageMultiPlanTest, MPSBackupPlan) {
AutoGetCollectionForReadCommand collection(_opCtx.get(), nss);
// Query for both 'a' and 'b' and sort on 'b'.
- auto qr = std::make_unique<QueryRequest>(nss);
- qr->setFilter(BSON("a" << 1 << "b" << 1));
- qr->setSort(BSON("b" << 1));
- auto statusWithCQ = CanonicalQuery::canonicalize(opCtx(), std::move(qr));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(BSON("a" << 1 << "b" << 1));
+ findCommand->setSort(BSON("b" << 1));
+ auto statusWithCQ = CanonicalQuery::canonicalize(opCtx(), std::move(findCommand));
verify(statusWithCQ.isOK());
unique_ptr<CanonicalQuery> cq = std::move(statusWithCQ.getValue());
ASSERT(nullptr != cq.get());
@@ -486,9 +486,9 @@ TEST_F(QueryStageMultiPlanTest, MPSExplainAllPlans) {
AutoGetCollectionForReadCommand ctx(_opCtx.get(), nss);
- auto qr = std::make_unique<QueryRequest>(nss);
- qr->setFilter(BSON("x" << 1));
- auto cq = uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(qr)));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(BSON("x" << 1));
+ auto cq = uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(findCommand)));
unique_ptr<MultiPlanStage> mps =
std::make_unique<MultiPlanStage>(_expCtx.get(), ctx.getCollection(), cq.get());
@@ -558,9 +558,9 @@ TEST_F(QueryStageMultiPlanTest, MPSSummaryStats) {
const CollectionPtr& coll = ctx.getCollection();
// Create the executor (Matching all documents).
- auto qr = std::make_unique<QueryRequest>(nss);
- qr->setFilter(BSON("foo" << BSON("$gte" << 0)));
- auto cq = uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(qr)));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(BSON("foo" << BSON("$gte" << 0)));
+ auto cq = uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(findCommand)));
auto exec = uassertStatusOK(
getExecutor(opCtx(), &coll, std::move(cq), PlanYieldPolicy::YieldPolicy::NO_YIELD, 0));
@@ -611,10 +611,10 @@ TEST_F(QueryStageMultiPlanTest, ShouldReportErrorIfExceedsTimeLimitDuringPlannin
getCollScanPlan(_expCtx.get(), coll.getCollection(), sharedWs.get(), filter.get());
- auto queryRequest = std::make_unique<QueryRequest>(nss);
- queryRequest->setFilter(filterObj);
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(filterObj);
auto canonicalQuery =
- uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(queryRequest)));
+ uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(findCommand)));
MultiPlanStage multiPlanStage(
_expCtx.get(), coll.getCollection(), canonicalQuery.get(), PlanCachingMode::NeverCache);
multiPlanStage.addPlan(createQuerySolution(), std::move(ixScanRoot), sharedWs.get());
@@ -651,10 +651,10 @@ TEST_F(QueryStageMultiPlanTest, ShouldReportErrorIfKilledDuringPlanning) {
unique_ptr<PlanStage> collScanRoot =
getCollScanPlan(_expCtx.get(), coll.getCollection(), sharedWs.get(), filter.get());
- auto queryRequest = std::make_unique<QueryRequest>(nss);
- queryRequest->setFilter(BSON("foo" << BSON("$gte" << 0)));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(BSON("foo" << BSON("$gte" << 0)));
auto canonicalQuery =
- uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(queryRequest)));
+ uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(findCommand)));
MultiPlanStage multiPlanStage(
_expCtx.get(), coll.getCollection(), canonicalQuery.get(), PlanCachingMode::NeverCache);
multiPlanStage.addPlan(createQuerySolution(), std::move(ixScanRoot), sharedWs.get());
@@ -694,11 +694,11 @@ TEST_F(QueryStageMultiPlanTest, AddsContextDuringException) {
insert(BSON("foo" << 10));
AutoGetCollectionForReadCommand ctx(_opCtx.get(), nss);
- auto queryRequest = std::make_unique<QueryRequest>(nss);
- queryRequest->setFilter(BSON("fake"
- << "query"));
+ auto findCommand = std::make_unique<FindCommand>(nss);
+ findCommand->setFilter(BSON("fake"
+ << "query"));
auto canonicalQuery =
- uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(queryRequest)));
+ uassertStatusOK(CanonicalQuery::canonicalize(opCtx(), std::move(findCommand)));
MultiPlanStage multiPlanStage(
_expCtx.get(), ctx.getCollection(), canonicalQuery.get(), PlanCachingMode::NeverCache);
unique_ptr<WorkingSet> sharedWs(new WorkingSet());