summaryrefslogtreecommitdiff
path: root/src/mongo/db/query/query_shape/agg_cmd_shape.cpp
diff options
context:
space:
mode:
authorLucas de Castro Borges <lucas@gnuabordo.com.br>2025-02-11 15:07:35 -0300
committerLucas de Castro Borges <lucas@gnuabordo.com.br>2025-02-11 15:07:35 -0300
commit4cb8841196d0625dfa3825aa326f071cd27c7b8b (patch)
tree1682a647d4463397c119183369ae6f750d5fdcff /src/mongo/db/query/query_shape/agg_cmd_shape.cpp
parentaa03c6362cbaa767638e6eed9b031d86dd2643d1 (diff)
parent8f0827553e09872941945a093b647a4211a9db7f (diff)
Update upstream source from tag 'upstream/6.0.0'master
Update to upstream version '6.0.0' with Debian dir 5604a80ec1c96ca76f25f40d78e6ef855abec322
Diffstat (limited to 'src/mongo/db/query/query_shape/agg_cmd_shape.cpp')
-rw-r--r--src/mongo/db/query/query_shape/agg_cmd_shape.cpp125
1 files changed, 0 insertions, 125 deletions
diff --git a/src/mongo/db/query/query_shape/agg_cmd_shape.cpp b/src/mongo/db/query/query_shape/agg_cmd_shape.cpp
deleted file mode 100644
index e997150ecc6..00000000000
--- a/src/mongo/db/query/query_shape/agg_cmd_shape.cpp
+++ /dev/null
@@ -1,125 +0,0 @@
-/**
- * Copyright (C) 2023-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/query_shape/agg_cmd_shape.h"
-
-#include "mongo/db/query/query_shape/shape_helpers.h"
-
-namespace mongo::query_shape {
-
-AggCmdShapeComponents::AggCmdShapeComponents(
- const AggregateCommandRequest& aggRequest,
- stdx::unordered_set<NamespaceString> involvedNamespaces_,
- std::vector<BSONObj> pipeline)
- : allowDiskUse(aggRequest.getAllowDiskUse()),
- involvedNamespaces(std::move(involvedNamespaces_)),
- representativePipeline(std::move(pipeline)) {}
-
-AggCmdShapeComponents::AggCmdShapeComponents(
- OptionalBool allowDiskUse,
- stdx::unordered_set<NamespaceString> involvedNamespaces_,
- std::vector<BSONObj> pipeline)
- : allowDiskUse(allowDiskUse),
- involvedNamespaces(std::move(involvedNamespaces_)),
- representativePipeline(std::move(pipeline)) {}
-
-void AggCmdShapeComponents::HashValue(absl::HashState state) const {
- state = absl::HashState::combine(std::move(state), allowDiskUse);
- for (auto&& shapifiedStage : representativePipeline) {
- state = absl::HashState::combine(std::move(state), simpleHash(shapifiedStage));
- }
-}
-
-void AggCmdShape::appendLetCmdSpecificShapeComponents(
- BSONObjBuilder& bob,
- const boost::intrusive_ptr<ExpressionContext>& expCtx,
- const SerializationOptions& opts) const {
- tassert(7633000,
- "We don't support serializing to the unmodified shape here, since we have already "
- "shapified and stored the representative query - we've lost the original literals",
- opts.literalPolicy != LiteralSerializationPolicy::kUnchanged);
-
- if (opts == SerializationOptions::kRepresentativeQueryShapeSerializeOptions) {
- // We have this copy stored already!
- return _components.appendTo(bob);
- } else {
- // The cached pipeline shape doesn't match the requested options, so we have to
- // re-parse the pipeline from the initial request.
- expCtx->inMongos = _inMongos;
- expCtx->addResolvedNamespaces(_components.involvedNamespaces);
- auto reparsed = Pipeline::parse(_components.representativePipeline, expCtx);
- auto serializedPipeline = reparsed->serializeToBson(opts);
- AggCmdShapeComponents{
- _components.allowDiskUse, _components.involvedNamespaces, serializedPipeline}
- .appendTo(bob);
- }
-}
-
-void AggCmdShapeComponents::appendTo(BSONObjBuilder& bob) const {
- bob.append("command", "aggregate");
-
- // pipeline
- bob.append(AggregateCommandRequest::kPipelineFieldName, representativePipeline);
-
- // allowDiskUse
- if (allowDiskUse.has_value()) {
- bob.append(AggregateCommandRequest::kAllowDiskUseFieldName, bool(allowDiskUse));
- }
-}
-
-// As part of the size, we must track the allocation of elements in the representative
-// pipeline, as well as the elements in the unordered set of involved namespaces.
-size_t AggCmdShapeComponents::size() const {
- return sizeof(AggCmdShapeComponents) + shape_helpers::containerSize(representativePipeline) +
- shape_helpers::containerSize(involvedNamespaces);
-}
-
-AggCmdShape::AggCmdShape(const AggregateCommandRequest& aggregateCommand,
- NamespaceString origNss,
- stdx::unordered_set<NamespaceString> involvedNamespaces_,
- const Pipeline& pipeline,
- const boost::intrusive_ptr<ExpressionContext>& expCtx)
- : CmdWithLetShape(aggregateCommand.getLet(),
- expCtx,
- _components,
- std::move(origNss),
- aggregateCommand.getCollation().value_or(BSONObj())),
- _components(aggregateCommand,
- std::move(involvedNamespaces_),
- pipeline.serializeToBson(
- SerializationOptions::kRepresentativeQueryShapeSerializeOptions)),
- _inMongos(expCtx->inMongos) {}
-
-size_t AggCmdShape::extraSize() const {
- // To account for possible padding, we calculate the extra space with the difference instead of
- // using sizeof(bool);
- return sizeof(AggCmdShape) - sizeof(CmdWithLetShape) - sizeof(AggCmdShapeComponents);
-}
-
-} // namespace mongo::query_shape