summaryrefslogtreecommitdiff
path: root/src/mongo
diff options
context:
space:
mode:
authorDavid Storch <david.storch@mongodb.com>2022-02-23 18:00:56 +0000
committerEvergreen Agent <no-reply@evergreen.mongodb.com>2022-02-23 19:43:16 +0000
commit149e5e01ecadaf3361d31e7ac850977b66db0751 (patch)
treebb743a4835e52e578506b30d4a04d88b100d6f10 /src/mongo
parent23d4be01ca042cd28d0d616f9334e02598e79510 (diff)
downloadmongo-149e5e01ecadaf3361d31e7ac850977b66db0751.tar.gz
SERVER-62797 Clear SBE plan cache when query knobs are changed
This necessary for any parameters which could affect the plans which are generated and cached by the optimizer. This commit also updates the encoding of the SBE plan cache key so that it no longer encodes the values of these parameters.
Diffstat (limited to 'src/mongo')
-rw-r--r--src/mongo/db/query/SConscript5
-rw-r--r--src/mongo/db/query/canonical_query_encoder.cpp21
-rw-r--r--src/mongo/db/query/canonical_query_encoder_test.cpp208
-rw-r--r--src/mongo/db/query/plan_cache_size_parameter.cpp30
-rw-r--r--src/mongo/db/query/plan_cache_size_parameter.h27
-rw-r--r--src/mongo/db/query/query_knobs.idl140
-rw-r--r--src/mongo/db/query/sbe_plan_cache.cpp17
-rw-r--r--src/mongo/db/query/sbe_plan_cache_on_parameter_change.cpp83
-rw-r--r--src/mongo/db/query/sbe_plan_cache_on_parameter_change.h91
9 files changed, 368 insertions, 254 deletions
diff --git a/src/mongo/db/query/SConscript b/src/mongo/db/query/SConscript
index 1428041ae89..9ca3f4021ec 100644
--- a/src/mongo/db/query/SConscript
+++ b/src/mongo/db/query/SConscript
@@ -248,9 +248,10 @@ env.Library(
env.Library(
target="query_knobs",
source=[
- 'query_knobs.idl',
- 'query_feature_flags.idl',
'plan_cache_size_parameter.cpp',
+ 'query_feature_flags.idl',
+ 'query_knobs.idl',
+ 'sbe_plan_cache_on_parameter_change.cpp',
],
LIBDEPS_PRIVATE=[
"$BUILD_DIR/mongo/db/service_context",
diff --git a/src/mongo/db/query/canonical_query_encoder.cpp b/src/mongo/db/query/canonical_query_encoder.cpp
index 51dbff5d00e..4892e62fdc2 100644
--- a/src/mongo/db/query/canonical_query_encoder.cpp
+++ b/src/mongo/db/query/canonical_query_encoder.cpp
@@ -39,7 +39,6 @@
#include "mongo/db/matcher/expression_array.h"
#include "mongo/db/matcher/expression_geo.h"
#include "mongo/db/query/projection.h"
-#include "mongo/db/query/query_knobs_gen.h"
#include "mongo/logv2/log.h"
#include "mongo/util/base64.h"
@@ -604,25 +603,6 @@ void encodeFindCommandRequest(const FindCommandRequest& findCommand, BufBuilder*
encodeBSONObj(findCommand.getMin());
encodeBSONObj(findCommand.getMax());
}
-
-void encodeQueryParameters(BufBuilder* bufBuilder) {
- auto encodeBool = [bufBuilder](bool val) { bufBuilder->appendChar(val ? 't' : 'f'); };
-
- encodeBool(internalQueryForceIntersectionPlans.load());
- encodeBool(internalQueryPlannerEnableIndexIntersection.load());
- encodeBool(internalQueryPlannerEnableHashIntersection.load());
-
- bufBuilder->appendNum(internalQueryPlannerMaxIndexedSolutions.load());
- encodeBool(internalQueryEnumerationPreferLockstepOrEnumeration.load());
- bufBuilder->appendNum(internalQueryEnumerationMaxOrSolutions.load());
- bufBuilder->appendNum(internalQueryEnumerationMaxIntersectPerAnd.load());
- encodeBool(internalQueryPlanOrChildrenIndependently.load());
- bufBuilder->appendNum(internalQueryMaxScansToExplode.load());
- encodeBool(internalQueryPlannerGenerateCoveredWholeIndexScans.load());
-
- bufBuilder->appendNum(internalQueryMaxBlockingSortMemoryUsageBytes.load());
- bufBuilder->appendNum(internalQuerySlotBasedExecutionMaxStaticIndexScanIntervals.load());
-}
} // namespace
namespace canonical_query_encoder {
@@ -668,7 +648,6 @@ std::string encodeSBE(const CanonicalQuery& cq) {
bufBuilder.appendStr(strBuilderEncoded, false /* includeEndingNull */);
encodeFindCommandRequest(cq.getFindCommandRequest(), &bufBuilder);
- encodeQueryParameters(&bufBuilder);
return base64::encode(StringData(bufBuilder.buf(), bufBuilder.len()));
}
diff --git a/src/mongo/db/query/canonical_query_encoder_test.cpp b/src/mongo/db/query/canonical_query_encoder_test.cpp
index 7cbeed6c05a..64cf9d111a4 100644
--- a/src/mongo/db/query/canonical_query_encoder_test.cpp
+++ b/src/mongo/db/query/canonical_query_encoder_test.cpp
@@ -363,178 +363,130 @@ TEST(CanonicalQueryEncoderTest, ComputeKeySBE) {
// Generated cache keys should be treated as opaque to the user.
RAIIServerParameterControllerForTest controller("featureFlagSbePlanCache", true);
- testComputeSBEKey("{}",
- "{}",
- "{}",
- "BQAAAAAFAAAAAAAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAGZ0ZkAAAABmCgAAAAMAAAB0yAAA"
- "AGYAAEAG6AMAAA==");
+ testComputeSBEKey("{}", "{}", "{}", "BQAAAAAFAAAAAAAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAA==");
testComputeSBEKey("{$or: [{a: 1}, {b: 2}]}",
"{}",
"{}",
"LQAAAAQkb3IAIwAAAAMwAAwAAAAQYQABAAAAAAMxAAwAAAAQYgACAAAAAAAABQAAAAAAAAAAAAAA"
- "AG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=");
- testComputeSBEKey("{a: 1}",
- "{}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAAD"
- "AAAAdMgAAABmAABABugDAAA=");
- testComputeSBEKey("{b: 1}",
- "{}",
- "{}",
- "DAAAABBiAAEAAAAABQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAAD"
- "AAAAdMgAAABmAABABugDAAA=");
- testComputeSBEKey("{a: 1, b: 1, c: 1}",
- "{}",
- "{}",
- "GgAAABBhAAEAAAAQYgABAAAAEGMAAQAAAAAFAAAAAAAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAA"
- "AGZ0ZkAAAABmCgAAAAMAAAB0yAAAAGYAAEAG6AMAAA==");
+ "AG5ubm4FAAAAAAUAAAAABQAAAAA=");
+ testComputeSBEKey(
+ "{a: 1}", "{}", "{}", "DAAAABBhAAEAAAAABQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
+ testComputeSBEKey(
+ "{b: 1}", "{}", "{}", "DAAAABBiAAEAAAAABQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
+ testComputeSBEKey(
+ "{a: 1, b: 1, c: 1}",
+ "{}",
+ "{}",
+ "GgAAABBhAAEAAAAQYgABAAAAEGMAAQAAAAAFAAAAAAAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAA==");
// With sort
- testComputeSBEKey("{}",
+ testComputeSBEKey(
+ "{}", "{a: 1}", "{}", "BQAAAAAFAAAAAH5hYQAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAA==");
+ testComputeSBEKey(
+ "{}", "{a: -1}", "{}", "BQAAAAAFAAAAAH5kYQAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAA==");
+ testComputeSBEKey("{a: 1}",
"{a: 1}",
"{}",
- "BQAAAAAFAAAAAH5hYQAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAGZ0ZkAAAABmCgAAAAMAAAB0"
- "yAAAAGYAAEAG6AMAAA==");
- testComputeSBEKey("{}",
- "{a: -1}",
- "{}",
- "BQAAAAAFAAAAAH5kYQAAAAAAAAAAbm5ubgUAAAAABQAAAAAFAAAAAGZ0ZkAAAABmCgAAAAMAAAB0"
- "yAAAAGYAAEAG6AMAAA==");
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=");
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
// With projection
testComputeSBEKey("{a: 1}",
"{a: 1}",
"{a: 1}",
- "DAAAABBhAAEAAAAADAAAABBhAAEAAAAAfmFhAAAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAAZnRm"
- "QAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA");
- testComputeSBEKey(
- "{}",
- "{a: 1}",
- "{a: 1}",
- "BQAAAAAMAAAAEGEAAQAAAAB+"
- "YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=");
+ "DAAAABBhAAEAAAAADAAAABBhAAEAAAAAfmFhAAAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAA");
testComputeSBEKey("{}",
"{a: 1}",
- "{a: 1, b: [{$const: 1}]}",
- "BQAAAAAoAAAAEGEAAQAAAARiABkAAAADMAARAAAAECRjb25zdAABAAAAAAAAfmFhAAAAAAAAAABu"
- "bm5uBQAAAAAFAAAAAAUAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA");
- testComputeSBEKey("{}",
- "{}",
"{a: 1}",
- "BQAAAAAMAAAAEGEAAQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAAD"
- "AAAAdMgAAABmAABABugDAAA=");
+ "BQAAAAAMAAAAEGEAAQAAAAB+YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
testComputeSBEKey("{}",
- "{}",
- "{a: true}",
- "BQAAAAAJAAAACGEAAQAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAA"
- "dMgAAABmAABABugDAAA=");
- testComputeSBEKey("{}",
- "{}",
- "{a: false}",
- "BQAAAAAJAAAACGEAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAA"
- "dMgAAABmAABABugDAAA=");
+ "{a: 1}",
+ "{a: 1, b: [{$const: 1}]}",
+ "BQAAAAAoAAAAEGEAAQAAAARiABkAAAADMAARAAAAECRjb25zdAABAAAAAAAAfmFhAAAAAAAAAABu"
+ "bm5uBQAAAAAFAAAAAAUAAAAA");
+ testComputeSBEKey(
+ "{}", "{}", "{a: 1}", "BQAAAAAMAAAAEGEAAQAAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
+ testComputeSBEKey(
+ "{}", "{}", "{a: true}", "BQAAAAAJAAAACGEAAQAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
+ testComputeSBEKey(
+ "{}", "{}", "{a: false}", "BQAAAAAJAAAACGEAAAAAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=");
// With FindCommandRequest
auto findCommand = std::make_unique<FindCommandRequest>(nss);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAABQAAAAA=",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setAllowDiskUse(true);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAHRubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAHRubm4FAAAAAAUAAAAABQAAAAA=",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setAllowDiskUse(false);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAGZubm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAGZubm4FAAAAAAUAAAAABQAAAAA=",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setReturnKey(true);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG50bm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG50bm4FAAAAAAUAAAAABQAAAAA=",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setRequestResumeToken(false);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG5uZm4FAAAAAAUAAAAABQAAAABmdGZAAAAAZgoAAAADAAAAdMgAAABmAABABugDAAA=",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG5uZm4FAAAAAAUAAAAABQAAAAA=",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setSkip(10);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEKAAAAAAAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEKAAAAAAAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAA",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setLimit(10);
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAACgAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAACgAAAAAAAABubm5uBQAAAAAFAAAAAAUAAAAA",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setMin(mongo::fromjson("{ a : 1 }"));
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG5ubm4FAAAAAAwAAAAQYQABAAAAAAUAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG5ubm4FAAAAAAwAAAAQYQABAAAAAAUAAAAA",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setMax(mongo::fromjson("{ a : 1 }"));
- testComputeSBEKey(
- "{a: 1}",
- "{a: 1}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAB+"
- "YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAADAAAABBhAAEAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA",
- std::move(findCommand));
+ testComputeSBEKey("{a: 1}",
+ "{a: 1}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAB+YWEAAAAAAAAAAG5ubm4FAAAAAAUAAAAADAAAABBhAAEAAAAA",
+ std::move(findCommand));
findCommand = std::make_unique<FindCommandRequest>(nss);
findCommand->setRequestResumeToken(true);
// "hint" must be {$natural:1} if 'requestResumeToken' is enabled.
findCommand->setHint(fromjson("{$natural: 1}"));
findCommand->setResumeAfter(mongo::fromjson("{ $recordId: NumberLong(1) }"));
- testComputeSBEKey("{a: 1}",
- "{}",
- "{}",
- "DAAAABBhAAEAAAAABQAAAAAAAAAAAAAAAG5udG4YAAAAEiRyZWNvcmRJZAABAAAAAAAAAAAFAAAA"
- "AAUAAAAAZnRmQAAAAGYKAAAAAwAAAHTIAAAAZgAAQAboAwAA",
- std::move(findCommand));
+ testComputeSBEKey(
+ "{a: 1}",
+ "{}",
+ "{}",
+ "DAAAABBhAAEAAAAABQAAAAAAAAAAAAAAAG5udG4YAAAAEiRyZWNvcmRJZAABAAAAAAAAAAAFAAAAAAUAAAAA",
+ std::move(findCommand));
}
} // namespace
diff --git a/src/mongo/db/query/plan_cache_size_parameter.cpp b/src/mongo/db/query/plan_cache_size_parameter.cpp
index 46302c8334e..46e42efafbf 100644
--- a/src/mongo/db/query/plan_cache_size_parameter.cpp
+++ b/src/mongo/db/query/plan_cache_size_parameter.cpp
@@ -31,7 +31,6 @@
#include <pcrecpp.h>
-#include "mongo/db/client.h"
#include "mongo/db/query/query_knobs_gen.h"
namespace mongo::plan_cache_util {
@@ -72,33 +71,4 @@ StatusWith<PlanCacheSizeParameter> PlanCacheSizeParameter::parse(const std::stri
return PlanCacheSizeParameter{size, statusWithUnit.getValue()};
}
-Status onPlanCacheSizeUpdate(const std::string& str) {
- auto newSize = PlanCacheSizeParameter::parse(str);
- if (!newSize.isOK()) {
- return newSize.getStatus();
- }
-
- // The client is nullptr if the parameter is supplied from the command line. In this case, we
- // ignore the update event, the parameter will be processed when initializing the service
- // context.
- if (auto client = Client::getCurrent()) {
- auto serviceCtx = client->getServiceContext();
- tassert(6007013, "ServiceContext must be non null", serviceCtx);
-
- auto updater = sbePlanCacheSizeUpdaterDecoration(serviceCtx).get();
- tassert(6007014, "Plan cache size updater must be non null", serviceCtx);
- updater->update(serviceCtx, newSize.getValue());
- }
-
- return Status::OK();
-}
-
-Status validatePlanCacheSize(const std::string& str) {
- return PlanCacheSizeParameter::parse(str).getStatus();
-}
-
-const Decorable<ServiceContext>::Decoration<std::unique_ptr<PlanCacheSizeUpdater>>
- sbePlanCacheSizeUpdaterDecoration =
- ServiceContext::declareDecoration<std::unique_ptr<PlanCacheSizeUpdater>>();
-
} // namespace mongo::plan_cache_util
diff --git a/src/mongo/db/query/plan_cache_size_parameter.h b/src/mongo/db/query/plan_cache_size_parameter.h
index ae3968e17e7..322a1fff564 100644
--- a/src/mongo/db/query/plan_cache_size_parameter.h
+++ b/src/mongo/db/query/plan_cache_size_parameter.h
@@ -32,7 +32,6 @@
#include <string>
#include "mongo/base/status_with.h"
-#include "mongo/db/service_context.h"
namespace mongo::plan_cache_util {
@@ -57,30 +56,4 @@ struct PlanCacheSizeParameter {
const PlanCacheSizeUnits units;
};
-/**
- * Callback called on a change of planCacheSize parameter.
- */
-Status onPlanCacheSizeUpdate(const std::string& str);
-
-/**
- * Callback called on validation of planCacheSize parameter.
- */
-Status validatePlanCacheSize(const std::string& str);
-
-/**
- * Encapsulates a callback function used to update the PlanCache size when the planCacheSize
- * parameter is updated.
- */
-class PlanCacheSizeUpdater {
-public:
- virtual ~PlanCacheSizeUpdater() = default;
- virtual void update(ServiceContext* serviceCtx, PlanCacheSizeParameter parameter) = 0;
-};
-
-/**
- * Decorated accessor to the PlanCacheSizeUpdater stored in ServiceContext.
- */
-extern const Decorable<ServiceContext>::Decoration<std::unique_ptr<PlanCacheSizeUpdater>>
- sbePlanCacheSizeUpdaterDecoration;
-
} // namespace mongo::plan_cache_util
diff --git a/src/mongo/db/query/query_knobs.idl b/src/mongo/db/query/query_knobs.idl
index 35b04fceda0..2150f93d3c3 100644
--- a/src/mongo/db/query/query_knobs.idl
+++ b/src/mongo/db/query/query_knobs.idl
@@ -30,6 +30,7 @@ global:
cpp_namespace: "mongo"
cpp_includes:
- "mongo/db/query/plan_cache_size_parameter.h"
+ - "mongo/db/query/sbe_plan_cache_on_parameter_change.h"
- "mongo/platform/atomic_proxy.h"
- "mongo/platform/atomic_word.h"
@@ -62,6 +63,7 @@ server_parameters:
default: 10000
validator:
gt: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlanEvaluationCollFraction:
description: "For large collections, the ceiling for the number times we work() candidate plans
@@ -86,6 +88,7 @@ server_parameters:
validator:
gte: 0.0
lte: 1.0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlanEvaluationMaxResults:
description: "Stop working plans once a plan returns this many results."
@@ -95,20 +98,24 @@ server_parameters:
default: 101
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryForceIntersectionPlans:
- description: "Do we give a big ranking bonus to intersection plans?"
+ description: "Gives a large ranking bonus to index intersection plans, forcing intersection
+ plans to be chosen when possible."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryForceIntersectionPlans"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlannerEnableIndexIntersection:
- description: "Do we have ixisect on at all?"
+ description: "Controls whether the planner will generate and consider index intersection plans."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryPlannerEnableIndexIntersection"
cpp_vartype: AtomicWord<bool>
default: true
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlannerEnableHashIntersection:
description: "Do we use hash-based intersection for rooted $and queries?"
@@ -116,13 +123,15 @@ server_parameters:
cpp_varname: "internalQueryPlannerEnableHashIntersection"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
#
# Plan cache
#
internalQueryCacheMaxEntriesPerCollection:
- description: "The maximum number of entries allowed in a given collection's plan cache."
+ description: "The maximum number of entries allowed in a given collection's plan cache. Applies
+ only to the classic plan cache, not to the SBE plan cache."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryCacheMaxEntriesPerCollection"
cpp_vartype: AtomicWord<int>
@@ -136,7 +145,8 @@ server_parameters:
the estimate of the number of bytes used across all plan caches exceeds this threshold, then
debug info is not stored alongside new cache entries, in order to limit plan cache memory
consumption. If plan cache entries are freed and the estimate once again dips below this
- threshold, then new cache entries will once again have debug info associated with them."
+ threshold, then new cache entries will once again have debug info associated with them. Applies
+ only to the classic plan cache, not to the SBE plan cache."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryCacheMaxSizeBytesBeforeStripDebugInfo"
cpp_vartype: AtomicWord<long long>
@@ -146,22 +156,26 @@ server_parameters:
gte: 0
internalQueryCacheEvictionRatio:
- description: "How many times more works must we perform in order to justify plan cache eviction and replanning?"
+ description: "How many times more works must we perform in order to justify plan cache eviction
+ and replanning?"
set_at: [ startup, runtime ]
cpp_varname: "internalQueryCacheEvictionRatio"
cpp_vartype: AtomicDouble
default: 10.0
validator:
gte: 0.0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryCacheWorksGrowthCoefficient:
- description: "How quickly the the 'works' value in an inactive cache entry will grow. It grows exponentially. The value of this server parameter is the base."
+ description: "How quickly the the 'works' value in an inactive cache entry will grow. It grows
+ exponentially. The value of this server parameter is the base."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryCacheWorksGrowthCoefficient"
cpp_vartype: AtomicDouble
default: 2.0
validator:
gt: 1.0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryCacheDisableInactiveEntries:
description: "Whether or not cache entries can be marked as 'inactive'."
@@ -169,6 +183,7 @@ server_parameters:
cpp_varname: "internalQueryCacheDisableInactiveEntries"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
planCacheSize:
description: "The maximum amount of memory that the system will allocate for the plan cache.
@@ -228,6 +243,7 @@ server_parameters:
default: 64
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryEnumerationPreferLockstepOrEnumeration:
description: "If set to true, instructs the plan enumerator to enumerate contained $ors in a
@@ -247,6 +263,7 @@ server_parameters:
cpp_varname: "internalQueryEnumerationPreferLockstepOrEnumeration"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryEnumerationMaxOrSolutions:
description: "How many solutions will the enumerator consider at each OR?"
@@ -256,6 +273,7 @@ server_parameters:
default: 10
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryEnumerationMaxIntersectPerAnd:
description: "How many intersections will the enumerator consider at each AND?"
@@ -265,6 +283,7 @@ server_parameters:
default: 3
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlanOrChildrenIndependently:
description: "Do we want to plan each child of the OR independently?"
@@ -272,22 +291,27 @@ server_parameters:
cpp_varname: "internalQueryPlanOrChildrenIndependently"
cpp_vartype: AtomicWord<bool>
default: true
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryMaxScansToExplode:
- description: "How many index scans are we willing to produce in order to obtain a sort order during explodeForSort?"
+ description: "How many index scans are we willing to produce in order to obtain a sort order
+ during explodeForSort?"
set_at: [ startup, runtime ]
cpp_varname: "internalQueryMaxScansToExplode"
cpp_vartype: AtomicWord<int>
default: 200
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryPlannerGenerateCoveredWholeIndexScans:
- description: "Allow the planner to generate covered whole index scans, rather than falling back to a COLLSCAN."
+ description: "Allow the planner to generate covered whole index scans, rather than falling back
+ to a COLLSCAN."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryPlannerGenerateCoveredWholeIndexScans"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryIgnoreUnknownJSONSchemaKeywords:
description: "Ignore unknown JSON Schema keywords."
@@ -310,6 +334,7 @@ server_parameters:
expr: 100 * 1024 * 1024
validator:
gte: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryExecYieldIterations:
description: "Yield after this many \"should yield?\" checks."
@@ -348,7 +373,8 @@ server_parameters:
gt: 0
internalLookupStageIntermediateDocumentMaxSizeBytes:
- description: "Maximum size of the result set that we cache from the foreign collection during a $lookup."
+ description: "Maximum size of the result set that we cache from the foreign collection during a
+ $lookup."
set_at: [ startup, runtime ]
cpp_varname: "internalLookupStageIntermediateDocumentMaxSizeBytes"
cpp_vartype: AtomicWord<long long>
@@ -358,7 +384,8 @@ server_parameters:
gte: { expr: BSONObjMaxInternalSize}
internalDocumentSourceGroupMaxMemoryBytes:
- description: "Maximum size of the data that the $group aggregation stage will cache in-memory before spilling to disk."
+ description: "Maximum size of the data that the $group aggregation stage will cache in-memory
+ before spilling to disk."
set_at: [ startup, runtime ]
cpp_varname: "internalDocumentSourceGroupMaxMemoryBytes"
cpp_vartype: AtomicWord<long long>
@@ -368,7 +395,8 @@ server_parameters:
gt: 0
internalDocumentSourceSetWindowFieldsMaxMemoryBytes:
- description: "Maximum size of the data that the $setWindowFields aggregation stage will cache in-memory before throwing an error."
+ description: "Maximum size of the data that the $setWindowFields aggregation stage will cache
+ in-memory before throwing an error."
set_at: [ startup, runtime ]
cpp_varname: "internalDocumentSourceSetWindowFieldsMaxMemoryBytes"
cpp_vartype: AtomicWord<long long>
@@ -389,7 +417,8 @@ server_parameters:
gt: 0
internalDocumentSourceCursorBatchSizeBytes:
- description: "Maximum amount of data that DocumentSourceCursor will cache from the underlying PlanExecutor before pipeline processing."
+ description: "Maximum amount of data that DocumentSourceCursor will cache from the underlying
+ PlanExecutor before pipeline processing."
set_at: [ startup, runtime ]
cpp_varname: "internalDocumentSourceCursorBatchSizeBytes"
cpp_vartype: AtomicWord<int>
@@ -399,7 +428,8 @@ server_parameters:
gte: 0
internalDocumentSourceLookupCacheSizeBytes:
- description: "Maximum amount of non-correlated foreign-collection data that the $lookup stage will cache before abandoning the cache and executing the full pipeline on each iteration."
+ description: "Maximum amount of non-correlated foreign-collection data that the $lookup stage
+ will cache before abandoning the cache and executing the full pipeline on each iteration."
set_at: [ startup, runtime ]
cpp_varname: "internalDocumentSourceLookupCacheSizeBytes"
cpp_vartype: AtomicWord<int>
@@ -409,7 +439,8 @@ server_parameters:
gte: 0
internalQueryProhibitBlockingMergeOnMongoS:
- description: "If true, blocking stages such as $group or non-merging $sort will be prohibited from running on mongoS."
+ description: "If true, blocking stages such as $group or non-merging $sort will be prohibited
+ from running on mongoS."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryProhibitBlockingMergeOnMongoS"
cpp_vartype: AtomicWord<bool>
@@ -428,7 +459,7 @@ server_parameters:
internalQueryMaxPushBytes:
description: "Limits the vector of values pushed into a single array while grouping with the
- $push accumulator."
+ $push accumulator."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryMaxPushBytes"
cpp_vartype: AtomicWord<int>
@@ -439,7 +470,7 @@ server_parameters:
internalQueryMaxRangeBytes:
description: "Limits the vector of values pushed into a single array while generating $range
- result."
+ result."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryMaxRangeBytes"
cpp_vartype: AtomicWord<int>
@@ -450,7 +481,7 @@ server_parameters:
internalQueryMaxAddToSetBytes:
description: "Limits the vector of values pushed into a single array while grouping with the
- $addToSet accumulator."
+ $addToSet accumulator."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryMaxAddToSetBytes"
cpp_vartype: AtomicWord<int>
@@ -460,7 +491,8 @@ server_parameters:
gt: 0
internalQueryJavaScriptHeapSizeLimitMB:
- description: "Limits the JavaScript heap size used in aggregation. Will defer to the global 'jsHeapLimitMB' limit if the global limit is smaller."
+ description: "Limits the JavaScript heap size used in aggregation. Will defer to the global
+ 'jsHeapLimitMB' limit if the global limit is smaller."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryJavaScriptHeapSizeLimitMB"
cpp_vartype: AtomicWord<int>
@@ -468,7 +500,8 @@ server_parameters:
expr: 100
internalQueryJavaScriptFnTimeoutMillis:
- description: "Limits the maximum allowed time a user-defined javascript function can run in a query."
+ description: "Limits the maximum allowed time a user-defined javascript function can run in a
+ query."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryJavaScriptFnTimeoutMillis"
cpp_vartype: AtomicWord<int>
@@ -485,7 +518,8 @@ server_parameters:
default: false
internalQueryDefaultDOP:
- description: "Default degree of parallelism. This an internal experimental parameter and should not be changed on live systems."
+ description: "Default degree of parallelism. This an internal experimental parameter and should
+ not be changed on live systems."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryDefaultDOP"
cpp_vartype: AtomicWord<int>
@@ -493,6 +527,7 @@ server_parameters:
test_only: true
validator:
gt: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryEnableLoggingV2OplogEntries:
description: "If true, this node may log $v:2 delta-style oplog entries."
@@ -502,7 +537,8 @@ server_parameters:
default: true
internalQuerySlotBasedExecutionMaxStaticIndexScanIntervals:
- description: "Limits the number of statically known intervals that SBE can decompose index bounds into when possible."
+ description: "Limits the number of statically known intervals that SBE can decompose index
+ bounds into when possible."
set_at: [ startup, runtime ]
cpp_varname: "internalQuerySlotBasedExecutionMaxStaticIndexScanIntervals"
cpp_vartype: AtomicWord<int>
@@ -510,6 +546,7 @@ server_parameters:
expr: 1000
validator:
gt: 0
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryEnableCSTParser:
description: "If true, use the grammar-based parser and CST to parse queries."
@@ -545,7 +582,8 @@ server_parameters:
description: "The memory check in HashAgg stage is done on every T'th processed record, where T
is calculated adaptively based on the estimated memory used and its recent growth. This setting
defines the percent of the remaining available memory to be used before the next check, given
- the estimated growth speed per advance [see internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
+ the estimated growth speed per advance [see
+ internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
set_at: [ startup, runtime ]
cpp_varname: "internalQuerySBEAggMemoryUseCheckMargin"
cpp_vartype: AtomicDouble
@@ -558,7 +596,8 @@ server_parameters:
internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtMost:
description: "The memory check in HashAgg stage is done on every T'th processed record, where T
is calculated adaptively based on the estimated memory used and its recent growth. This setting
- defines the lower bound for T [see internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
+ defines the lower bound for T [see
+ internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
set_at: [ startup, runtime ]
cpp_varname: "internalQuerySBEAggMemoryCheckPerAdvanceAtMost"
cpp_vartype: AtomicWord<long long>
@@ -570,9 +609,10 @@ server_parameters:
internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtLeast:
description: "The memory check in HashAgg stage is done on every T'th processed record, where T
is calculated adaptively based on the estimated memory used and its recent growth. This setting
- defines the upper bound for T. If this setting is less than [see internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtMost],
- the check will be done on every internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtLeast'th
- processed record [see internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
+ defines the upper bound for T. If this setting is less than [see
+ internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtMost], the check will be done on
+ every internalQuerySlotBasedExecutionHashAggMemoryCheckPerAdvanceAtLeast'th processed record
+ [see internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill]."
set_at: [ startup, runtime ]
cpp_varname: "internalQuerySBEAggMemoryCheckPerAdvanceAtLeast"
cpp_vartype: AtomicWord<long long>
@@ -582,8 +622,8 @@ server_parameters:
gt: 0
internalQuerySlotBasedExecutionHashAggApproxMemoryUseInBytesBeforeSpill:
- description: "The max size in bytes that the hash table in a HashAgg stage can be
- estimated to be before we spill to disk."
+ description: "The max size in bytes that the hash table in a HashAgg stage can be estimated to
+ be before we spill to disk."
set_at: [ startup, runtime ]
cpp_varname: "internalQuerySBEAggApproxMemoryUseInBytesBeforeSpill"
cpp_vartype: AtomicWord<long long>
@@ -599,16 +639,19 @@ server_parameters:
cpp_varname: "internalQueryForceClassicEngine"
cpp_vartype: AtomicWord<bool>
default: false
+ on_update: plan_cache_util::clearSbeCacheOnParameterChange
internalQueryAppendIdToSetWindowFieldsSort:
- description: "If true, appends _id to the sort stage generated by desugaring $setWindowFields to ensure deterministic sort order."
+ description: "If true, appends _id to the sort stage generated by desugaring $setWindowFields to
+ ensure deterministic sort order."
set_at: [startup, runtime]
cpp_varname: "internalQueryAppendIdToSetWindowFieldsSort"
cpp_vartype: AtomicWord<bool>
default: false
internalQueryTopNAccumulatorBytes:
- description: "Limits the vector of values pushed into a single array while grouping with the 'N' family of accumulators."
+ description: "Limits the vector of values pushed into a single array while grouping with the 'N'
+ family of accumulators."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryTopNAccumulatorBytes"
cpp_vartype: AtomicWord<int>
@@ -618,8 +661,8 @@ server_parameters:
gt: 0
enableSearchMeta:
- description: "Exists for backwards compatibility in startup parameters,
- enabling this was required on 4.4 to access SEARCH_META variables. Does not do anything."
+ description: "Exists for backwards compatibility in startup parameters, enabling this was
+ required on 4.4 to access SEARCH_META variables. Does not do anything."
set_at: [ startup, runtime ]
cpp_varname: "enableSearchMeta"
cpp_vartype: AtomicWord<bool>
@@ -659,64 +702,79 @@ server_parameters:
default: false
internalQueryEnableSamplingCardinalityEstimator:
- description: "Set to use the sampling-based method for estimating cardinality."
+ description: "Set to use the sampling-based method for estimating cardinality in the Cascades
+ optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryEnableSamplingCardinalityEstimator"
cpp_vartype: AtomicWord<bool>
default: true
internalQueryEnableCascadesOptimizer:
- description: "Set to use the new optimizer path, must be used in conjunction with the feature flag."
+ description: "Set to use the new optimizer path, must be used in conjunction with the feature
+ flag."
set_at: [ startup, runtime ]
cpp_varname: "internalQueryEnableCascadesOptimizer"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableScan:
- description: "Disable full collection scans."
+ description: "Disable full collection scans in the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableScan"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableIndexes:
- description: "Disable index scan plans."
+ description: "Disable index scan plans in the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableIndexes"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableMergeJoinRIDIntersect:
- description: "Disable index RID intersection via merge join."
+ description: "Disable index RID intersection via merge join in the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableMergeJoinRIDIntersect"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableHashJoinRIDIntersect:
- description: "Disable index RID intersection via hash join."
+ description: "Disable index RID intersection via hash join in the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableHashJoinRIDIntersect"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableGroupByAndUnionRIDIntersect:
- description: "Disable index RID intersection via group by and union."
+ description: "Disable index RID intersection via group by and union in the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableGroupByAndUnionRIDIntersect"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerKeepRejectedPlans:
- description: "Keep track of rejected plans in the memo."
+ description: "Keep track of rejected plans in the memo. Applies only to the Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerKeepRejectedPlans"
cpp_vartype: AtomicWord<bool>
default: false
internalCascadesOptimizerDisableBranchAndBound:
- description: "Disable cascades branch-and-bound strategy, and fully evaluate all plans."
+ description: "Disable cascades branch-and-bound strategy, and fully evaluate all plans in the
+ Cascades optimizer."
set_at: [ startup, runtime ]
cpp_varname: "internalCascadesOptimizerDisableBranchAndBound"
cpp_vartype: AtomicWord<bool>
default: false
+
+# Note for adding additional query knobs:
+#
+# When adding a new query knob, you should consider whether or not you need to add an 'on_update'
+# hook to flush the SBE plan cache. If your knob affects the contents of SBE plans which may be
+# cached, then the SBE cache should be flushed when the value of the knob changes. This will ensure
+# that the application actually starts to get plans which reflect the new value of the knob, rather
+# than continuing to use stale cached plans.
+#
+# When adding a new query knob which flushes the SBE plan cache on update, you should test this
+# behavior by including the name of the new knob in the list at the top of the jstest
+# 'sbe_plan_cache_clear_on_param_change.js'.
diff --git a/src/mongo/db/query/sbe_plan_cache.cpp b/src/mongo/db/query/sbe_plan_cache.cpp
index 28f97cd886d..64e1767db80 100644
--- a/src/mongo/db/query/sbe_plan_cache.cpp
+++ b/src/mongo/db/query/sbe_plan_cache.cpp
@@ -99,22 +99,29 @@ size_t getPlanCacheSizeInBytes(const plan_cache_util::PlanCacheSizeParameter& pa
return capPlanCacheSize(planCacheSize);
}
-class PlanCacheSizeUpdaterImpl final : public plan_cache_util::PlanCacheSizeUpdater {
+class PlanCacheOnParamChangeUpdaterImpl final : public plan_cache_util::OnParamChangeUpdater {
public:
- void update(ServiceContext* serviceCtx,
- plan_cache_util::PlanCacheSizeParameter parameter) final {
+ void updateCacheSize(ServiceContext* serviceCtx,
+ plan_cache_util::PlanCacheSizeParameter parameter) final {
if (feature_flags::gFeatureFlagSbePlanCache.isEnabledAndIgnoreFCV()) {
auto size = getPlanCacheSizeInBytes(parameter);
auto& globalPlanCache = sbePlanCacheDecoration(serviceCtx);
globalPlanCache->reset(size);
}
}
+
+ void clearCache(ServiceContext* serviceCtx) final {
+ if (feature_flags::gFeatureFlagSbePlanCache.isEnabledAndIgnoreFCV()) {
+ auto& globalPlanCache = sbePlanCacheDecoration(serviceCtx);
+ globalPlanCache->clear();
+ }
+ }
};
ServiceContext::ConstructorActionRegisterer planCacheRegisterer{
"PlanCacheRegisterer", [](ServiceContext* serviceCtx) {
- plan_cache_util::sbePlanCacheSizeUpdaterDecoration(serviceCtx) =
- std::make_unique<PlanCacheSizeUpdaterImpl>();
+ plan_cache_util::sbePlanCacheOnParamChangeUpdater(serviceCtx) =
+ std::make_unique<PlanCacheOnParamChangeUpdaterImpl>();
if (feature_flags::gFeatureFlagSbePlanCache.isEnabledAndIgnoreFCV()) {
auto status = plan_cache_util::PlanCacheSizeParameter::parse(planCacheSize.get());
diff --git a/src/mongo/db/query/sbe_plan_cache_on_parameter_change.cpp b/src/mongo/db/query/sbe_plan_cache_on_parameter_change.cpp
new file mode 100644
index 00000000000..399d5210128
--- /dev/null
+++ b/src/mongo/db/query/sbe_plan_cache_on_parameter_change.cpp
@@ -0,0 +1,83 @@
+/**
+ * Copyright (C) 2022-present MongoDB, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the Server Side Public License, version 1,
+ * as published by MongoDB, Inc.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * Server Side Public License for more details.
+ *
+ * You should have received a copy of the Server Side Public License
+ * along with this program. If not, see
+ * <http://www.mongodb.com/licensing/server-side-public-license>.
+ *
+ * As a special exception, the copyright holders give permission to link the
+ * code of portions of this program with the OpenSSL library under certain
+ * conditions as described in each individual source file and distribute
+ * linked combinations including the program with the OpenSSL library. You
+ * must comply with the Server Side Public License in all respects for
+ * all of the code used other than as permitted herein. If you modify file(s)
+ * with this exception, you may extend this exception to your version of the
+ * file(s), but you are not obligated to do so. If you do not wish to do so,
+ * delete this exception statement from your version. If you delete this
+ * exception statement from all source files in the program, then also delete
+ * it in the license file.
+ */
+
+#include "mongo/db/query/sbe_plan_cache_on_parameter_change.h"
+
+#include "mongo/db/client.h"
+
+namespace mongo::plan_cache_util {
+namespace {
+/**
+ * Given the current 'Client', returns a pointer to the 'ServiceContext' and an interface for
+ * updating the SBE plan cache,
+ */
+std::pair<ServiceContext*, OnParamChangeUpdater*> getUpdater(const Client& client) {
+ auto serviceCtx = client.getServiceContext();
+ tassert(6007013, "ServiceContext must be non null", serviceCtx);
+
+ auto updater = sbePlanCacheOnParamChangeUpdater(serviceCtx).get();
+ tassert(6007014, "Plan cache size updater must be non null", updater);
+ return {serviceCtx, updater};
+}
+} // namespace
+
+Status clearSbeCacheOnParameterChangeHelper() {
+ if (auto client = Client::getCurrent()) {
+ auto&& [serviceCtx, updater] = getUpdater(*client);
+ updater->clearCache(serviceCtx);
+ }
+ return Status::OK();
+}
+
+Status onPlanCacheSizeUpdate(const std::string& str) {
+ auto newSize = PlanCacheSizeParameter::parse(str);
+ if (!newSize.isOK()) {
+ return newSize.getStatus();
+ }
+
+ // The client is nullptr if the parameter is supplied from the command line. In this case, we
+ // ignore the update event, the parameter will be processed when initializing the service
+ // context.
+ if (auto client = Client::getCurrent()) {
+ auto&& [serviceCtx, updater] = getUpdater(*client);
+ updater->updateCacheSize(serviceCtx, newSize.getValue());
+ }
+
+ return Status::OK();
+}
+
+Status validatePlanCacheSize(const std::string& str) {
+ return PlanCacheSizeParameter::parse(str).getStatus();
+}
+
+const Decorable<ServiceContext>::Decoration<std::unique_ptr<OnParamChangeUpdater>>
+ sbePlanCacheOnParamChangeUpdater =
+ ServiceContext::declareDecoration<std::unique_ptr<OnParamChangeUpdater>>();
+
+} // namespace mongo::plan_cache_util
diff --git a/src/mongo/db/query/sbe_plan_cache_on_parameter_change.h b/src/mongo/db/query/sbe_plan_cache_on_parameter_change.h
new file mode 100644
index 00000000000..e126dda9f48
--- /dev/null
+++ b/src/mongo/db/query/sbe_plan_cache_on_parameter_change.h
@@ -0,0 +1,91 @@
+/**
+ * Copyright (C) 2022-present MongoDB, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the Server Side Public License, version 1,
+ * as published by MongoDB, Inc.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * Server Side Public License for more details.
+ *
+ * You should have received a copy of the Server Side Public License
+ * along with this program. If not, see
+ * <http://www.mongodb.com/licensing/server-side-public-license>.
+ *
+ * As a special exception, the copyright holders give permission to link the
+ * code of portions of this program with the OpenSSL library under certain
+ * conditions as described in each individual source file and distribute
+ * linked combinations including the program with the OpenSSL library. You
+ * must comply with the Server Side Public License in all respects for
+ * all of the code used other than as permitted herein. If you modify file(s)
+ * with this exception, you may extend this exception to your version of the
+ * file(s), but you are not obligated to do so. If you do not wish to do so,
+ * delete this exception statement from your version. If you delete this
+ * exception statement from all source files in the program, then also delete
+ * it in the license file.
+ */
+
+#pragma once
+
+#include <string>
+
+#include "mongo/base/status.h"
+#include "mongo/db/query/plan_cache_size_parameter.h"
+#include "mongo/db/service_context.h"
+
+namespace mongo::plan_cache_util {
+
+/**
+ * Callback called on a change of planCacheSize parameter.
+ */
+Status onPlanCacheSizeUpdate(const std::string& str);
+
+/**
+ * Callback called on validation of planCacheSize parameter.
+ */
+Status validatePlanCacheSize(const std::string& str);
+
+/**
+ * Clears the SBE plan cache. Used to implement 'clearSbeCacheOnParameterChange()' below.
+ */
+Status clearSbeCacheOnParameterChangeHelper();
+
+/**
+ * Hook to delete all SBE plan cache entries when query-related setParameter values are updated at
+ * runtime.
+ */
+constexpr inline auto clearSbeCacheOnParameterChange = [](auto&&) {
+ return clearSbeCacheOnParameterChangeHelper();
+};
+
+/**
+ * An interface used to modify the SBE plan cache when query setParameters are modified. This is
+ * done via an interface decorating the 'ServiceContext' in order to avoid a link-time dependency
+ * of the query knobs library on the SBE plan cache code.
+ */
+class OnParamChangeUpdater {
+public:
+ virtual ~OnParamChangeUpdater() = default;
+
+ /**
+ * Resizes the SBE plan cache decorating 'serviceCtx' to the new size given by 'parameter'. If
+ * the new cache size is smaller than the old, cache entries are evicted in order to ensure the
+ * cache fits within the new size bound.
+ */
+ virtual void updateCacheSize(ServiceContext* serviceCtx, PlanCacheSizeParameter parameter) = 0;
+
+ /**
+ * Deletes all plans from the SBE plan cache decorating 'serviceCtx'.
+ */
+ virtual void clearCache(ServiceContext* serviceCtx) = 0;
+};
+
+/**
+ * Decorated accessor to the 'OnParamChangeUpdater' stored in 'ServiceContext'.
+ */
+extern const Decorable<ServiceContext>::Decoration<std::unique_ptr<OnParamChangeUpdater>>
+ sbePlanCacheOnParamChangeUpdater;
+
+} // namespace mongo::plan_cache_util