diff options
| author | Lucas de Castro Borges <lucas@gnuabordo.com.br> | 2025-02-11 15:07:35 -0300 |
|---|---|---|
| committer | Lucas de Castro Borges <lucas@gnuabordo.com.br> | 2025-02-11 15:07:35 -0300 |
| commit | 4cb8841196d0625dfa3825aa326f071cd27c7b8b (patch) | |
| tree | 1682a647d4463397c119183369ae6f750d5fdcff /src/mongo/db/query/query_shape/find_cmd_shape.cpp | |
| parent | aa03c6362cbaa767638e6eed9b031d86dd2643d1 (diff) | |
| parent | 8f0827553e09872941945a093b647a4211a9db7f (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/find_cmd_shape.cpp')
| -rw-r--r-- | src/mongo/db/query/query_shape/find_cmd_shape.cpp | 227 |
1 files changed, 0 insertions, 227 deletions
diff --git a/src/mongo/db/query/query_shape/find_cmd_shape.cpp b/src/mongo/db/query/query_shape/find_cmd_shape.cpp deleted file mode 100644 index 2d018de2619..00000000000 --- a/src/mongo/db/query/query_shape/find_cmd_shape.cpp +++ /dev/null @@ -1,227 +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/find_cmd_shape.h" - -#include "mongo/db/query/projection_ast_util.h" -#include "mongo/db/query/query_shape/shape_helpers.h" - -namespace mongo::query_shape { -namespace { - -BSONObj projectionShape(const boost::optional<projection_ast::Projection>& proj, - const SerializationOptions& opts = - SerializationOptions::kRepresentativeQueryShapeSerializeOptions) { - return proj ? projection_ast::serialize(*proj->root(), opts) : BSONObj(); -} - -BSONObj sortShape(const boost::optional<SortPattern>& sort, - const SerializationOptions& opts = - SerializationOptions::kRepresentativeQueryShapeSerializeOptions) { - return sort - ? sort->serialize(SortPattern::SortKeySerialization::kForPipelineSerialization, opts) - .toBson() - : BSONObj(); -} - -void maybeAddWithName(const OptionalBool& optBool, BSONObjBuilder& bob, StringData name) { - if (optBool.has_value()) { - bob.append(name, bool(optBool)); - } -} - -void addRemainingFindCommandFields(const FindCmdShapeComponents& components, BSONObjBuilder& bob) { - maybeAddWithName(components.singleBatch, bob, FindCommandRequest::kSingleBatchFieldName); - maybeAddWithName(components.allowDiskUse, bob, FindCommandRequest::kAllowDiskUseFieldName); - maybeAddWithName(components.returnKey, bob, FindCommandRequest::kReturnKeyFieldName); - maybeAddWithName(components.showRecordId, bob, FindCommandRequest::kShowRecordIdFieldName); - maybeAddWithName(components.tailable, bob, FindCommandRequest::kTailableFieldName); - maybeAddWithName(components.awaitData, bob, FindCommandRequest::kAwaitDataFieldName); - maybeAddWithName(components.oplogReplay, bob, FindCommandRequest::kOplogReplayFieldName); -} - -} // namespace - -FindCmdShapeComponents::FindCmdShapeComponents( - const ParsedFindCommand& request, - const boost::intrusive_ptr<ExpressionContext>& expCtx, - const SerializationOptions& opts) - : filter(request.filter->serialize(opts)), - projection(projectionShape(request.proj, opts)), - sort(sortShape(request.sort, opts)), - min(shape_helpers::extractMinOrMaxShape(request.findCommandRequest->getMin(), opts)), - max(shape_helpers::extractMinOrMaxShape(request.findCommandRequest->getMax(), opts)), - singleBatch(request.findCommandRequest->getSingleBatch()), - allowDiskUse(request.findCommandRequest->getAllowDiskUse().has_value() - ? boost::optional<bool>(bool(request.findCommandRequest->getAllowDiskUse())) - : boost::none), - returnKey(request.findCommandRequest->getReturnKey()), - showRecordId(request.findCommandRequest->getShowRecordId()), - tailable(request.findCommandRequest->getTailable()), - awaitData(request.findCommandRequest->getAwaitData()), - oplogReplay(request.findCommandRequest->getOplogReplay()), - hasField(), - serializationOpts(opts) { - hasField.projection = request.proj.has_value(); - hasField.sort = request.sort.has_value(); - hasField.limit = request.findCommandRequest->getLimit().has_value(); - hasField.skip = request.findCommandRequest->getSkip().has_value(); -} - -void FindCmdShapeComponents::appendTo(BSONObjBuilder& bob) const { - - bob.append("command", "find"); - - std::unique_ptr<MatchExpression> filterExpr; - // Filter. - bob.append(FindCommandRequest::kFilterFieldName, filter); - - if (hasField.projection) { - bob.append(FindCommandRequest::kProjectionFieldName, projection); - } - - if (!max.isEmpty()) { - bob.append(FindCommandRequest::kMaxFieldName, max); - } - if (!min.isEmpty()) { - bob.append(FindCommandRequest::kMinFieldName, min); - } - - // Sort. - if (hasField.sort) { - bob.append(FindCommandRequest::kSortFieldName, sort); - } - - // The values here don't matter (assuming we're not using the 'kUnchanged' policy). - tassert(7973601, - "Serialization policy not supported - original values have been discarded", - serializationOpts.literalPolicy != LiteralSerializationPolicy::kUnchanged); - if (hasField.limit) { - serializationOpts.appendLiteral(&bob, FindCommandRequest::kLimitFieldName, 1ll); - } - if (hasField.skip) { - serializationOpts.appendLiteral(&bob, FindCommandRequest::kSkipFieldName, 1ll); - } - - // Add the fields that require no transformation. - addRemainingFindCommandFields(*this, bob); -} - -void FindCmdShapeComponents::HashValue(absl::HashState state) const { - absl::HashState::combine(std::move(state), - simpleHash(filter), - simpleHash(projection), - simpleHash(sort), - simpleHash(min), - simpleHash(max), - singleBatch, - allowDiskUse, - returnKey, - showRecordId, - tailable, - awaitData, - oplogReplay, - hasField); -} - -std::unique_ptr<FindCommandRequest> FindCmdShape::toFindCommandRequest() const { - auto fcr = std::make_unique<FindCommandRequest>(nssOrUUID); - - fcr->setFilter(components.filter); - if (components.hasField.projection) - fcr->setProjection(components.projection); - if (components.hasField.sort) - fcr->setSort(components.sort); - - fcr->setMin(components.min); - fcr->setMax(components.max); - - // Doesn't matter what value to use for limit and skip in the context of a shape. - if (components.hasField.limit) - fcr->setLimit(1ll); - if (components.hasField.skip) - fcr->setSkip(1ll); - - // All the booleans. - if (components.singleBatch.has_value()) - fcr->setSingleBatch(bool(components.singleBatch)); - if (components.allowDiskUse.has_value()) - fcr->setAllowDiskUse(bool(components.allowDiskUse)); - if (components.returnKey.has_value()) - fcr->setReturnKey(bool(components.returnKey)); - if (components.showRecordId.has_value()) - fcr->setShowRecordId(bool(components.showRecordId)); - if (components.tailable.has_value()) - fcr->setTailable(bool(components.tailable)); - if (components.awaitData.has_value()) - fcr->setAwaitData(bool(components.awaitData)); - if (components.oplogReplay.has_value()) - fcr->setOplogReplay(bool(components.oplogReplay)); - - // Common shape components. - if (_let.hasLet) - fcr->setLet(_let.shapifiedLet); - if (!collation.isEmpty()) - fcr->setCollation(collation); - - - return fcr; -} - -FindCmdShape::FindCmdShape(const ParsedFindCommand& findRequest, - const boost::intrusive_ptr<ExpressionContext>& expCtx) - : CmdWithLetShape(findRequest.findCommandRequest->getLet(), - expCtx, - components, - findRequest.findCommandRequest->getNamespaceOrUUID(), - findRequest.findCommandRequest->getCollation()), - components(findRequest, expCtx) {} - -void FindCmdShape::appendLetCmdSpecificShapeComponents( - BSONObjBuilder& bob, - const boost::intrusive_ptr<ExpressionContext>& expCtx, - const SerializationOptions& opts) const { - if (opts == SerializationOptions::kRepresentativeQueryShapeSerializeOptions) { - // Fast path: we already have this. - return components.appendTo(bob); - } else { - // Slow path: we need to re-parse from our representative shapes. - auto request = uassertStatusOKWithContext( - parsed_find_command::parse(expCtx, - toFindCommandRequest(), - ExtensionsCallbackNoop(), - MatchExpressionParser::kAllowAllSpecialFeatures), - "Could not re-parse a representative query shape"); - - // This constructor will shapify according to the options. - FindCmdShapeComponents{*request, expCtx, opts}.appendTo(bob); - } -} - -} // namespace mongo::query_shape |
