/** * Copyright (C) 2018-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 * . * * 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 "mongo/platform/basic.h" #include #include #include #include #include #include "mongo/base/init.h" #include "mongo/db/pipeline/dependencies.h" #include "mongo/db/pipeline/document.h" #include "mongo/db/pipeline/expression_context.h" #include "mongo/db/pipeline/field_path.h" #include "mongo/db/pipeline/value.h" #include "mongo/db/pipeline/variables.h" #include "mongo/db/query/datetime/date_time_support.h" #include "mongo/db/server_options.h" #include "mongo/stdx/functional.h" #include "mongo/util/intrusive_counter.h" #include "mongo/util/mongoutils/str.h" namespace mongo { class BSONArrayBuilder; class BSONElement; class BSONObjBuilder; class DocumentSource; /** * Registers a Parser so it can be called from parseExpression and friends. * * As an example, if your expression looks like {"$foo": [1,2,3]} you would add this line: * REGISTER_EXPRESSION(foo, ExpressionFoo::parse); * * An expression registered this way can be used in any featureCompatibilityVersion. */ #define REGISTER_EXPRESSION(key, parser) \ MONGO_INITIALIZER(addToExpressionParserMap_##key)(InitializerContext*) { \ Expression::registerExpression("$" #key, (parser), boost::none); \ return Status::OK(); \ } /** * Registers a Parser so it can be called from parseExpression and friends. Use this version if your * expression can only be persisted to a catalog data structure in a feature compatibility version * >= X. * * As an example, if your expression looks like {"$foo": [1,2,3]}, and can only be used in a feature * compatibility version >= X, you would add this line: * REGISTER_EXPRESSION_WITH_MIN_VERSION(foo, ExpressionFoo::parse, X); */ #define REGISTER_EXPRESSION_WITH_MIN_VERSION(key, parser, minVersion) \ MONGO_INITIALIZER(addToExpressionParserMap_##key)(InitializerContext*) { \ Expression::registerExpression("$" #key, (parser), (minVersion)); \ return Status::OK(); \ } class Expression : public RefCountable { public: using Parser = stdx::function( const boost::intrusive_ptr&, BSONElement, const VariablesParseState&)>; /** * Represents new paths computed by an expression. Computed paths are partitioned into renames * and non-renames. See the comments for Expression::getComputedPaths() for more information. */ struct ComputedPaths { // Non-rename computed paths. std::set paths; // Mappings from the old name of a path before applying this expression, to the new one // after applying this expression. StringMap renames; }; virtual ~Expression(){}; /** * Optimize the Expression. * * This provides an opportunity to do constant folding, or to collapse nested operators that * have the same precedence, such as $add, $and, or $or. * * The Expression will be replaced with the return value, which may or may not be the same * object. In the case of constant folding, a computed expression may be replaced by a constant. * * Returns the optimized Expression. */ virtual boost::intrusive_ptr optimize() { return this; } /** * Add the fields and variables used in this expression to 'deps'. References to variables which * are local to a particular expression will be filtered out of the tracker upon return. */ void addDependencies(DepsTracker* deps) { _doAddDependencies(deps); // Filter out references to any local variables. if (_boundaryVariableId) { deps->vars.erase(deps->vars.upper_bound(*_boundaryVariableId), deps->vars.end()); } } /** * Serialize the Expression tree recursively. * * If 'explain' is false, the returned Value must result in the same Expression when parsed by * parseOperand(). */ virtual Value serialize(bool explain) const = 0; /** * Evaluate expression with respect to the Document given by 'root', and return the result. */ virtual Value evaluate(const Document& root) const = 0; /** * Returns information about the paths computed by this expression. This only needs to be * overridden by expressions that have renaming semantics, where optimization code could take * advantage of knowledge of these renames. * * Partitions paths involved in this expression into the set of computed paths and the set of * ("new" => "old") rename mappings. Here "new" refers to the name of the path after applying * this expression, whereas "old" refers to the name of the path before applying this * expression. * * The 'exprFieldPath' is the field path at which the result of this expression will be stored. * This is used to determine the value of the "new" path created by the rename. * * The 'renamingVar' is needed for checking whether a field path is a rename. For example, at * the top level only field paths that begin with the ROOT variable, as in "$$ROOT.path", are * renames. A field path such as "$$var.path" is not a rename. * * Now consider the example of a rename expressed via a $map: * * {$map: {input: "$array", as: "iter", in: {...}}} * * In this case, only field paths inside the "in" clause beginning with "iter", such as * "$$iter.path", are renames. */ virtual ComputedPaths getComputedPaths(const std::string& exprFieldPath, Variables::Id renamingVar = Variables::kRootId) const { return {{exprFieldPath}, {}}; } /** * Parses a BSON Object that could represent an object literal or a functional expression like * $add. * * Calls parseExpression() on any sub-document (including possibly the entire document) which * consists of a single field name starting with a '$'. */ static boost::intrusive_ptr parseObject( const boost::intrusive_ptr& expCtx, BSONObj obj, const VariablesParseState& vps); /** * Parses a BSONObj which has already been determined to be a functional expression. * * Throws an error if 'obj' does not contain exactly one field, or if that field's name does not * match a registered expression name. */ static boost::intrusive_ptr parseExpression( const boost::intrusive_ptr& expCtx, BSONObj obj, const VariablesParseState& vps); /** * Parses a BSONElement which is an argument to an Expression. * * An argument is allowed to be another expression, or a literal value, so this can call * parseObject(), ExpressionFieldPath::parse(), ExpressionArray::parse(), or * ExpressionConstant::parse() as necessary. */ static boost::intrusive_ptr parseOperand( const boost::intrusive_ptr& expCtx, BSONElement exprElement, const VariablesParseState& vps); /* Produce a field path std::string with the field prefix removed. Throws an error if the field prefix is not present. @param prefixedField the prefixed field @returns the field path with the prefix removed */ static std::string removeFieldPrefix(const std::string& prefixedField); /** * Registers an Parser so it can be called from parseExpression. * * DO NOT call this method directly. Instead, use the REGISTER_EXPRESSION macro defined in this * file. */ static void registerExpression( std::string key, Parser parser, boost::optional requiredMinVersion); protected: Expression(const boost::intrusive_ptr& expCtx) : _expCtx(expCtx) { auto varIds = _expCtx->variablesParseState.getDefinedVariableIDs(); if (!varIds.empty()) { _boundaryVariableId = *std::prev(varIds.end()); } } typedef std::vector> ExpressionVector; const boost::intrusive_ptr& getExpressionContext() const { return _expCtx; } virtual void _doAddDependencies(DepsTracker* deps) const = 0; private: boost::optional _boundaryVariableId; boost::intrusive_ptr _expCtx; }; // Inherit from ExpressionVariadic or ExpressionFixedArity instead of directly from this class. class ExpressionNary : public Expression { public: boost::intrusive_ptr optimize() override; Value serialize(bool explain) const override; /* Add an operand to the n-ary expression. @param pExpression the expression to add */ virtual void addOperand(const boost::intrusive_ptr& pExpression); virtual bool isAssociative() const { return false; } virtual bool isCommutative() const { return false; } /* Get the name of the operator. @returns the name of the operator; this std::string belongs to the class implementation, and should not be deleted and should not */ virtual const char* getOpName() const = 0; /// Allow subclasses the opportunity to validate arguments at parse time. virtual void validateArguments(const ExpressionVector& args) const {} static ExpressionVector parseArguments(const boost::intrusive_ptr& expCtx, BSONElement bsonExpr, const VariablesParseState& vps); const ExpressionVector& getOperandList() const { return vpOperand; } protected: explicit ExpressionNary(const boost::intrusive_ptr& expCtx) : Expression(expCtx) {} void _doAddDependencies(DepsTracker* deps) const override; ExpressionVector vpOperand; }; /// Inherit from ExpressionVariadic or ExpressionFixedArity instead of directly from this class. template class ExpressionNaryBase : public ExpressionNary { public: static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement bsonExpr, const VariablesParseState& vps) { auto expr = make_intrusive(expCtx); ExpressionVector args = parseArguments(expCtx, bsonExpr, vps); expr->validateArguments(args); expr->vpOperand = args; return std::move(expr); } protected: explicit ExpressionNaryBase(const boost::intrusive_ptr& expCtx) : ExpressionNary(expCtx) {} }; /// Inherit from this class if your expression takes a variable number of arguments. template class ExpressionVariadic : public ExpressionNaryBase { public: explicit ExpressionVariadic(const boost::intrusive_ptr& expCtx) : ExpressionNaryBase(expCtx) {} }; /** * Inherit from this class if your expression can take a range of arguments, e.g. if it has some * optional arguments. */ template class ExpressionRangedArity : public ExpressionNaryBase { public: explicit ExpressionRangedArity(const boost::intrusive_ptr& expCtx) : ExpressionNaryBase(expCtx) {} void validateArguments(const Expression::ExpressionVector& args) const override { uassert(28667, mongoutils::str::stream() << "Expression " << this->getOpName() << " takes at least " << MinArgs << " arguments, and at most " << MaxArgs << ", but " << args.size() << " were passed in.", MinArgs <= args.size() && args.size() <= MaxArgs); } }; /// Inherit from this class if your expression takes a fixed number of arguments. template class ExpressionFixedArity : public ExpressionNaryBase { public: explicit ExpressionFixedArity(const boost::intrusive_ptr& expCtx) : ExpressionNaryBase(expCtx) {} void validateArguments(const Expression::ExpressionVector& args) const override { uassert(16020, mongoutils::str::stream() << "Expression " << this->getOpName() << " takes exactly " << NArgs << " arguments. " << args.size() << " were passed in.", args.size() == NArgs); } }; /** * Used to make Accumulators available as Expressions, e.g., to make $sum available as an Expression * use "REGISTER_EXPRESSION(sum, ExpressionAccumulator::parse);". */ template class ExpressionFromAccumulator : public ExpressionVariadic> { public: explicit ExpressionFromAccumulator(const boost::intrusive_ptr& expCtx) : ExpressionVariadic>(expCtx) {} Value evaluate(const Document& root) const final { Accumulator accum(this->getExpressionContext()); const size_t n = this->vpOperand.size(); // If a single array arg is given, loop through it passing each member to the accumulator. // If a single, non-array arg is given, pass it directly to the accumulator. if (n == 1) { Value singleVal = this->vpOperand[0]->evaluate(root); if (singleVal.getType() == Array) { for (const Value& val : singleVal.getArray()) { accum.process(val, false); } } else { accum.process(singleVal, false); } } else { // If multiple arguments are given, pass all arguments to the accumulator. for (auto&& argument : this->vpOperand) { accum.process(argument->evaluate(root), false); } } return accum.getValue(false); } bool isAssociative() const final { // Return false if a single argument is given to avoid a single array argument being treated // as an array instead of as a list of arguments. if (this->vpOperand.size() == 1) { return false; } return Accumulator(this->getExpressionContext()).isAssociative(); } bool isCommutative() const final { return Accumulator(this->getExpressionContext()).isCommutative(); } const char* getOpName() const final { return Accumulator(this->getExpressionContext()).getOpName(); } }; /** * Inherit from this class if your expression takes exactly one numeric argument. */ template class ExpressionSingleNumericArg : public ExpressionFixedArity { public: explicit ExpressionSingleNumericArg(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} virtual ~ExpressionSingleNumericArg() = default; Value evaluate(const Document& root) const final { Value arg = this->vpOperand[0]->evaluate(root); if (arg.nullish()) return Value(BSONNULL); uassert(28765, str::stream() << this->getOpName() << " only supports numeric types, not " << typeName(arg.getType()), arg.numeric()); return evaluateNumericArg(arg); } virtual Value evaluateNumericArg(const Value& numericArg) const = 0; }; /** * Inherit from this class if your expression takes exactly two numeric arguments. */ template class ExpressionTwoNumericArgs : public ExpressionFixedArity { public: explicit ExpressionTwoNumericArgs(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} virtual ~ExpressionTwoNumericArgs() = default; /** * Evaluate performs the type checking necessary to make sure that both arguments are numeric, * then calls the evaluateNumericArgs on the two numeric args: * 1. If either input is nullish, it returns null. * 2. If either input is not numeric, it throws an error. * 3. Call evaluateNumericArgs on the two numeric args. */ Value evaluate(const Document& root) const final { Value arg1 = this->vpOperand[0]->evaluate(root); if (arg1.nullish()) return Value(BSONNULL); uassert(51044, str::stream() << this->getOpName() << " only supports numeric types, not " << typeName(arg1.getType()), arg1.numeric()); Value arg2 = this->vpOperand[1]->evaluate(root); if (arg2.nullish()) return Value(BSONNULL); uassert(51045, str::stream() << this->getOpName() << " only supports numeric types, not " << typeName(arg2.getType()), arg2.numeric()); return evaluateNumericArgs(arg1, arg2); } /** * Evaluate the expression on exactly two numeric arguments. */ virtual Value evaluateNumericArgs(const Value& numericArg1, const Value& numericArg2) const = 0; }; /** * A constant expression. Repeated calls to evaluate() will always return the same thing. */ class ExpressionConstant final : public Expression { public: boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; Value serialize(bool explain) const final; const char* getOpName() const; /** * Creates a new ExpressionConstant with value 'value'. */ static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, const Value& value); static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement bsonExpr, const VariablesParseState& vps); /** * Returns true if 'expression' is nullptr or if 'expression' is an instance of an * ExpressionConstant. */ static bool isNullOrConstant(boost::intrusive_ptr expression) { return !expression || dynamic_cast(expression.get()); } /** * Returns true if every expression in 'expressions' is either a nullptr or an instance of an * ExpressionConstant. */ static bool allNullOrConstant( const std::initializer_list>& expressions) { return std::all_of(expressions.begin(), expressions.end(), [](auto exp) { return ExpressionConstant::isNullOrConstant(exp); }); } /** * Returns the constant value represented by this Expression. */ Value getValue() const { return _value; } protected: void _doAddDependencies(DepsTracker* deps) const override; private: ExpressionConstant(const boost::intrusive_ptr& expCtx, const Value& value); Value _value; }; /** * Inherit from this class if your expression works with date types, and accepts either a single * argument which is a date, or an object {date: , timezone: }. */ template class DateExpressionAcceptingTimeZone : public Expression { public: virtual ~DateExpressionAcceptingTimeZone() {} Value evaluate(const Document& root) const final { auto dateVal = _date->evaluate(root); if (dateVal.nullish()) { return Value(BSONNULL); } auto date = dateVal.coerceToDate(); if (!_timeZone) { return evaluateDate(date, TimeZoneDatabase::utcZone()); } auto timeZoneId = _timeZone->evaluate(root); if (timeZoneId.nullish()) { return Value(BSONNULL); } uassert(40533, str::stream() << _opName << " requires a string for the timezone argument, but was given a " << typeName(timeZoneId.getType()) << " (" << timeZoneId.toString() << ")", timeZoneId.getType() == BSONType::String); invariant(getExpressionContext()->timeZoneDatabase); auto timeZone = getExpressionContext()->timeZoneDatabase->getTimeZone(timeZoneId.getString()); return evaluateDate(date, timeZone); } /** * Always serializes to the full {date: , timezone: } format, leaving * off the timezone if not specified. */ Value serialize(bool explain) const final { auto timezone = _timeZone ? _timeZone->serialize(explain) : Value(); return Value(Document{ {_opName, Document{{"date", _date->serialize(explain)}, {"timezone", std::move(timezone)}}}}); } boost::intrusive_ptr optimize() final { _date = _date->optimize(); if (_timeZone) { _timeZone = _timeZone->optimize(); } if (ExpressionConstant::allNullOrConstant({_date, _timeZone})) { // Everything is a constant, so we can turn into a constant. return ExpressionConstant::create(getExpressionContext(), evaluate(Document{})); } return this; } static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement operatorElem, const VariablesParseState& variablesParseState) { boost::intrusive_ptr dateExpression{new SubClass(expCtx)}; if (operatorElem.type() == BSONType::Object) { if (operatorElem.embeddedObject().firstElementFieldName()[0] == '$') { // Assume this is an expression specification representing the date argument // like {$add: [, 1000]}. dateExpression->_date = Expression::parseObject( expCtx, operatorElem.embeddedObject(), variablesParseState); return dateExpression; } else { // It's an object specifying the date and timezone options like {date: , // timezone: }. auto opName = operatorElem.fieldNameStringData(); for (const auto& subElem : operatorElem.embeddedObject()) { auto argName = subElem.fieldNameStringData(); if (argName == "date"_sd) { dateExpression->_date = Expression::parseOperand(expCtx, subElem, variablesParseState); } else if (argName == "timezone"_sd) { dateExpression->_timeZone = Expression::parseOperand(expCtx, subElem, variablesParseState); } else { uasserted(40535, str::stream() << "unrecognized option to " << opName << ": \"" << argName << "\""); } } uassert(40539, str::stream() << "missing 'date' argument to " << opName << ", provided: " << operatorElem, dateExpression->_date); } return dateExpression; } else if (operatorElem.type() == BSONType::Array) { auto elems = operatorElem.Array(); uassert( 40536, str::stream() << dateExpression->_opName << " accepts exactly one argument if given an array, but was given " << elems.size(), elems.size() == 1); // We accept an argument wrapped in a single array. For example, either {$week: } // or {$week: []} are valid, but not {$week: [{date: }]}. operatorElem = elems[0]; } // It's some literal value, or the first element of a user-specified array. Either way it // should be treated as the date argument. dateExpression->_date = Expression::parseOperand(expCtx, operatorElem, variablesParseState); return dateExpression; } protected: explicit DateExpressionAcceptingTimeZone(StringData opName, const boost::intrusive_ptr& expCtx) : Expression(expCtx), _opName(opName) {} void _doAddDependencies(DepsTracker* deps) const final { _date->addDependencies(deps); if (_timeZone) { _timeZone->addDependencies(deps); } } /** * Subclasses should implement this to do their actual date-related logic. Uses 'timezone' to * evaluate the expression against 'data'. If the user did not specify a time zone, 'timezone' * will represent the UTC zone. */ virtual Value evaluateDate(Date_t date, const TimeZone& timezone) const = 0; private: // The name of this expression, e.g. $week or $month. StringData _opName; // The expression representing the date argument. boost::intrusive_ptr _date; // The expression representing the timezone argument, nullptr if not specified. boost::intrusive_ptr _timeZone = nullptr; }; class ExpressionAbs final : public ExpressionSingleNumericArg { public: explicit ExpressionAbs(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionAdd final : public ExpressionVariadic { public: explicit ExpressionAdd(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; class ExpressionAllElementsTrue final : public ExpressionFixedArity { public: explicit ExpressionAllElementsTrue(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionAnd final : public ExpressionVariadic { public: explicit ExpressionAnd(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; class ExpressionAnyElementTrue final : public ExpressionFixedArity { public: explicit ExpressionAnyElementTrue(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionArray final : public ExpressionVariadic { public: explicit ExpressionArray(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; Value serialize(bool explain) const final; boost::intrusive_ptr optimize() final; const char* getOpName() const final; }; class ExpressionArrayElemAt final : public ExpressionFixedArity { public: explicit ExpressionArrayElemAt(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionObjectToArray final : public ExpressionFixedArity { public: explicit ExpressionObjectToArray(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionArrayToObject final : public ExpressionFixedArity { public: explicit ExpressionArrayToObject(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionCeil final : public ExpressionSingleNumericArg { public: explicit ExpressionCeil(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionCoerceToBool final : public Expression { public: boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; Value serialize(bool explain) const final; static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, const boost::intrusive_ptr& pExpression); protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionCoerceToBool(const boost::intrusive_ptr& expCtx, const boost::intrusive_ptr& pExpression); boost::intrusive_ptr pExpression; }; class ExpressionCompare final : public ExpressionFixedArity { public: /** * Enumeration of comparison operators. Any changes to these values require adjustment of * the lookup table in the implementation. */ enum CmpOp { EQ = 0, // return true for a == b, false otherwise NE = 1, // return true for a != b, false otherwise GT = 2, // return true for a > b, false otherwise GTE = 3, // return true for a >= b, false otherwise LT = 4, // return true for a < b, false otherwise LTE = 5, // return true for a <= b, false otherwise CMP = 6, // return -1, 0, 1 for a < b, a == b, a > b }; ExpressionCompare(const boost::intrusive_ptr& expCtx, CmpOp cmpOp) : ExpressionFixedArity(expCtx), cmpOp(cmpOp) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; CmpOp getOp() const { return cmpOp; } static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement bsonExpr, const VariablesParseState& vps, CmpOp cmpOp); static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, CmpOp cmpOp, const boost::intrusive_ptr& exprLeft, const boost::intrusive_ptr& exprRight); private: CmpOp cmpOp; }; class ExpressionConcat final : public ExpressionVariadic { public: explicit ExpressionConcat(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } }; class ExpressionConcatArrays final : public ExpressionVariadic { public: explicit ExpressionConcatArrays(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } }; class ExpressionCond final : public ExpressionFixedArity { public: explicit ExpressionCond(const boost::intrusive_ptr& expCtx) : Base(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); private: typedef ExpressionFixedArity Base; }; class ExpressionDateFromString final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document&) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionDateFromString(const boost::intrusive_ptr& expCtx, boost::intrusive_ptr dateString, boost::intrusive_ptr timeZone, boost::intrusive_ptr format, boost::intrusive_ptr onNull, boost::intrusive_ptr onError); boost::intrusive_ptr _dateString; boost::intrusive_ptr _timeZone; boost::intrusive_ptr _format; boost::intrusive_ptr _onNull; boost::intrusive_ptr _onError; }; class ExpressionDateFromParts final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionDateFromParts(const boost::intrusive_ptr& expCtx, boost::intrusive_ptr year, boost::intrusive_ptr month, boost::intrusive_ptr day, boost::intrusive_ptr hour, boost::intrusive_ptr minute, boost::intrusive_ptr second, boost::intrusive_ptr millisecond, boost::intrusive_ptr isoWeekYear, boost::intrusive_ptr isoWeek, boost::intrusive_ptr isoDayOfWeek, boost::intrusive_ptr timeZone); /** * This function checks whether a field is a number. * * If 'field' is null, the default value is returned trough the 'returnValue' out * parameter and the function returns true. * * If 'field' is not null: * - if the value is "nullish", the function returns false. * - if the value can not be coerced to an integral value, a UserException is thrown. * - otherwise, the coerced integral value is returned through the 'returnValue' * out parameter, and the function returns true. */ bool evaluateNumberWithDefault(const Document& root, boost::intrusive_ptr field, StringData fieldName, long long defaultValue, long long* returnValue) const; boost::intrusive_ptr _year; boost::intrusive_ptr _month; boost::intrusive_ptr _day; boost::intrusive_ptr _hour; boost::intrusive_ptr _minute; boost::intrusive_ptr _second; boost::intrusive_ptr _millisecond; boost::intrusive_ptr _isoWeekYear; boost::intrusive_ptr _isoWeek; boost::intrusive_ptr _isoDayOfWeek; boost::intrusive_ptr _timeZone; }; class ExpressionDateToParts final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: /** * The iso8601 argument controls whether to output ISO8601 elements or natural calendar. */ ExpressionDateToParts(const boost::intrusive_ptr& expCtx, boost::intrusive_ptr date, boost::intrusive_ptr timeZone, boost::intrusive_ptr iso8601); boost::optional evaluateIso8601Flag(const Document& root) const; boost::intrusive_ptr _date; boost::intrusive_ptr _timeZone; boost::intrusive_ptr _iso8601; }; class ExpressionDateToString final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionDateToString(const boost::intrusive_ptr& expCtx, boost::intrusive_ptr format, boost::intrusive_ptr date, boost::intrusive_ptr timeZone, boost::intrusive_ptr onNull); boost::intrusive_ptr _format; boost::intrusive_ptr _date; boost::intrusive_ptr _timeZone; boost::intrusive_ptr _onNull; }; class ExpressionDayOfMonth final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionDayOfMonth(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$dayOfMonth", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).dayOfMonth); } }; class ExpressionDayOfWeek final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionDayOfWeek(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$dayOfWeek", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dayOfWeek(date)); } }; class ExpressionDayOfYear final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionDayOfYear(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$dayOfYear", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dayOfYear(date)); } }; class ExpressionDivide final : public ExpressionFixedArity { public: explicit ExpressionDivide(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionExp final : public ExpressionSingleNumericArg { public: explicit ExpressionExp(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionFieldPath final : public Expression { public: bool isRootFieldPath() const { return _variable == Variables::kRootId; } boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; Value serialize(bool explain) const final; /* Create a field path expression using old semantics (rooted off of CURRENT). // NOTE: this method is deprecated and only used by tests // TODO remove this method in favor of parse() Evaluation will extract the value associated with the given field path from the source document. @param fieldPath the field path string, without any leading document indicator @returns the newly created field path expression */ static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, const std::string& fieldPath); /// Like create(), but works with the raw std::string from the user with the "$" prefixes. static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, const std::string& raw, const VariablesParseState& vps); /** * Returns true if this expression logically represents the path 'dottedPath'. For example, if * 'dottedPath' is 'a.b' and this FieldPath is '$$CURRENT.a.b', returns true. */ bool representsPath(const std::string& dottedPath) const; const FieldPath& getFieldPath() const { return _fieldPath; } ComputedPaths getComputedPaths(const std::string& exprFieldPath, Variables::Id renamingVar) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionFieldPath(const boost::intrusive_ptr& expCtx, const std::string& fieldPath, Variables::Id variable); /* Internal implementation of evaluate(), used recursively. The internal implementation doesn't just use a loop because of the possibility that we need to skip over an array. If the path is "a.b.c", and a is an array, then we fan out from there, and traverse "b.c" for each element of a:[...]. This requires that a be an array of objects in order to navigate more deeply. @param index current path field index to extract @param input current document traversed to (not the top-level one) @returns the field found; could be an array */ Value evaluatePath(size_t index, const Document& input) const; // Helper for evaluatePath to handle Array case Value evaluatePathArray(size_t index, const Value& input) const; const FieldPath _fieldPath; const Variables::Id _variable; }; class ExpressionFilter final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionFilter(const boost::intrusive_ptr& expCtx, std::string varName, Variables::Id varId, boost::intrusive_ptr input, boost::intrusive_ptr filter); // The name of the variable to set to each element in the array. std::string _varName; // The id of the variable to set. Variables::Id _varId; // The array to iterate over. boost::intrusive_ptr _input; // The expression determining whether each element should be present in the result array. boost::intrusive_ptr _filter; }; class ExpressionFloor final : public ExpressionSingleNumericArg { public: explicit ExpressionFloor(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionHour final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionHour(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$hour", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).hour); } }; class ExpressionIfNull final : public ExpressionFixedArity { public: explicit ExpressionIfNull(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionIn final : public ExpressionFixedArity { public: explicit ExpressionIn(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionIndexOfArray : public ExpressionRangedArity { public: explicit ExpressionIndexOfArray(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const; boost::intrusive_ptr optimize() final; const char* getOpName() const final; protected: struct Arguments { Arguments(Value targetOfSearch, int startIndex, int endIndex) : targetOfSearch(targetOfSearch), startIndex(startIndex), endIndex(endIndex) {} Value targetOfSearch; int startIndex; int endIndex; }; /** * When given 'operands' which correspond to the arguments to $indexOfArray, evaluates and * validates the target value, starting index, and ending index arguments and returns their * values as a Arguments struct. The starting index and ending index are optional, so as default * 'startIndex' will be 0 and 'endIndex' will be the length of the input array. Throws a * UserException if the values are found to be invalid in some way, e.g. if the indexes are not * numbers. */ Arguments evaluateAndValidateArguments(const Document& root, const ExpressionVector& operands, size_t arrayLength) const; private: class Optimized; }; class ExpressionIndexOfBytes final : public ExpressionRangedArity { public: explicit ExpressionIndexOfBytes(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; /** * Implements indexOf behavior for strings with UTF-8 encoding. */ class ExpressionIndexOfCP final : public ExpressionRangedArity { public: explicit ExpressionIndexOfCP(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionLet final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); struct NameAndExpression { NameAndExpression() {} NameAndExpression(std::string name, boost::intrusive_ptr expression) : name(name), expression(expression) {} std::string name; boost::intrusive_ptr expression; }; typedef std::map VariableMap; protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionLet(const boost::intrusive_ptr& expCtx, const VariableMap& vars, boost::intrusive_ptr subExpression); VariableMap _variables; boost::intrusive_ptr _subExpression; }; class ExpressionLn final : public ExpressionSingleNumericArg { public: explicit ExpressionLn(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionLog final : public ExpressionFixedArity { public: explicit ExpressionLog(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionLog10 final : public ExpressionSingleNumericArg { public: explicit ExpressionLog10(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionMap final : public Expression { public: boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); ComputedPaths getComputedPaths(const std::string& exprFieldPath, Variables::Id renamingVar) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionMap( const boost::intrusive_ptr& expCtx, const std::string& varName, // name of variable to set Variables::Id varId, // id of variable to set boost::intrusive_ptr input, // yields array to iterate boost::intrusive_ptr each); // yields results to be added to output array std::string _varName; Variables::Id _varId; boost::intrusive_ptr _input; boost::intrusive_ptr _each; }; class ExpressionMeta final : public Expression { public: Value serialize(bool explain) const final; Value evaluate(const Document& root) const final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vps); protected: void _doAddDependencies(DepsTracker* deps) const final; private: enum MetaType { TEXT_SCORE, RAND_VAL, }; ExpressionMeta(const boost::intrusive_ptr& expCtx, MetaType metaType); MetaType _metaType; }; class ExpressionMillisecond final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionMillisecond(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$millisecond", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).millisecond); } }; class ExpressionMinute final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionMinute(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$minute", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).minute); } }; class ExpressionMod final : public ExpressionFixedArity { public: explicit ExpressionMod(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionMultiply final : public ExpressionVariadic { public: explicit ExpressionMultiply(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; class ExpressionMonth final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionMonth(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$month", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).month); } }; class ExpressionNot final : public ExpressionFixedArity { public: explicit ExpressionNot(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; /** * This class is used to represent expressions that create object literals, such as the value of * '_id' in this group stage: * {$group: { * _id: {b: "$a", c: {$add: [4, "$c"]}} <- This is represented as an ExpressionObject. * ... * }} */ class ExpressionObject final : public Expression { public: boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; Value serialize(bool explain) const final; static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, std::vector>>&& expressions); /** * Parses and constructs an ExpressionObject from 'obj'. */ static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONObj obj, const VariablesParseState& vps); /** * This ExpressionObject must outlive the returned vector. */ const std::vector>>& getChildExpressions() const { return _expressions; } ComputedPaths getComputedPaths(const std::string& exprFieldPath, Variables::Id renamingVar) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionObject( const boost::intrusive_ptr& expCtx, std::vector>>&& expressions); // The mapping from field name to expression within this object. This needs to respect the order // in which the fields were specified in the input BSON. std::vector>> _expressions; }; class ExpressionOr final : public ExpressionVariadic { public: explicit ExpressionOr(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} boost::intrusive_ptr optimize() final; Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; class ExpressionPow final : public ExpressionFixedArity { public: explicit ExpressionPow(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} static boost::intrusive_ptr create( const boost::intrusive_ptr& expCtx, Value base, Value exp); private: Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionRange final : public ExpressionRangedArity { public: explicit ExpressionRange(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionReduce final : public Expression { public: explicit ExpressionReduce(const boost::intrusive_ptr& expCtx) : Expression(expCtx) {} Value evaluate(const Document& root) const final; boost::intrusive_ptr optimize() final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vpsIn); Value serialize(bool explain) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: boost::intrusive_ptr _input; boost::intrusive_ptr _initial; boost::intrusive_ptr _in; Variables::Id _valueVar; Variables::Id _thisVar; }; class ExpressionSecond final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionSecond(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$second", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).second); } }; class ExpressionSetDifference final : public ExpressionFixedArity { public: explicit ExpressionSetDifference(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSetEquals final : public ExpressionVariadic { public: explicit ExpressionSetEquals(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; void validateArguments(const ExpressionVector& args) const final; }; class ExpressionSetIntersection final : public ExpressionVariadic { public: explicit ExpressionSetIntersection(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; // Not final, inherited from for optimizations. class ExpressionSetIsSubset : public ExpressionFixedArity { public: explicit ExpressionSetIsSubset(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} boost::intrusive_ptr optimize() override; Value evaluate(const Document& root) const override; const char* getOpName() const final; private: class Optimized; }; class ExpressionSetUnion final : public ExpressionVariadic { public: explicit ExpressionSetUnion(const boost::intrusive_ptr& expCtx) : ExpressionVariadic(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; bool isAssociative() const final { return true; } bool isCommutative() const final { return true; } }; class ExpressionSize final : public ExpressionFixedArity { public: explicit ExpressionSize(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionReverseArray final : public ExpressionFixedArity { public: explicit ExpressionReverseArray(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSlice final : public ExpressionRangedArity { public: explicit ExpressionSlice(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionIsArray final : public ExpressionFixedArity { public: explicit ExpressionIsArray(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionRound final : public ExpressionRangedArity { public: explicit ExpressionRound(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSplit final : public ExpressionFixedArity { public: explicit ExpressionSplit(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSqrt final : public ExpressionSingleNumericArg { public: explicit ExpressionSqrt(const boost::intrusive_ptr& expCtx) : ExpressionSingleNumericArg(expCtx) {} Value evaluateNumericArg(const Value& numericArg) const final; const char* getOpName() const final; }; class ExpressionStrcasecmp final : public ExpressionFixedArity { public: explicit ExpressionStrcasecmp(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSubstrBytes : public ExpressionFixedArity { public: explicit ExpressionSubstrBytes(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const; }; class ExpressionSubstrCP final : public ExpressionFixedArity { public: explicit ExpressionSubstrCP(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionStrLenBytes final : public ExpressionFixedArity { public: explicit ExpressionStrLenBytes(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionStrLenCP final : public ExpressionFixedArity { public: explicit ExpressionStrLenCP(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSubtract final : public ExpressionFixedArity { public: explicit ExpressionSubtract(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionSwitch final : public Expression { public: explicit ExpressionSwitch(const boost::intrusive_ptr& expCtx) : Expression(expCtx) {} Value evaluate(const Document& root) const final; boost::intrusive_ptr optimize() final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vpsIn); Value serialize(bool explain) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: using ExpressionPair = std::pair, boost::intrusive_ptr>; boost::intrusive_ptr _default; std::vector _branches; }; class ExpressionToLower final : public ExpressionFixedArity { public: explicit ExpressionToLower(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionToUpper final : public ExpressionFixedArity { public: explicit ExpressionToUpper(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; /** * This class is used to implement all three trim expressions: $trim, $ltrim, and $rtrim. */ class ExpressionTrim final : public Expression { private: enum class TrimType { kBoth, kLeft, kRight, }; public: ExpressionTrim(const boost::intrusive_ptr& expCtx, TrimType trimType, StringData name, const boost::intrusive_ptr& input, const boost::intrusive_ptr& charactersToTrim) : Expression(expCtx), _trimType(trimType), _name(name.toString()), _input(input), _characters(charactersToTrim) {} Value evaluate(const Document& root) const final; boost::intrusive_ptr optimize() final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vpsIn); Value serialize(bool explain) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: /** * Returns true if the unicode character found at index 'indexIntoInput' of 'input' is equal to * 'testCP'. */ static bool codePointMatchesAtIndex(const StringData& input, std::size_t indexIntoInput, const StringData& testCP); /** * Given the input string and the code points to trim from that string, returns a substring of * 'input' with any code point from 'trimCPs' trimmed from the left. */ static StringData trimFromLeft(StringData input, const std::vector& trimCPs); /** * Given the input string and the code points to trim from that string, returns a substring of * 'input' with any code point from 'trimCPs' trimmed from the right. */ static StringData trimFromRight(StringData input, const std::vector& trimCPs); /** * Returns the trimmed version of 'input', with all code points in 'trimCPs' removed from the * front, back, or both - depending on _trimType. */ StringData doTrim(StringData input, const std::vector& trimCPs) const; TrimType _trimType; std::string _name; // "$trim", "$ltrim", or "$rtrim". boost::intrusive_ptr _input; boost::intrusive_ptr _characters; // Optional, null if not specified. }; class ExpressionTrunc final : public ExpressionRangedArity { public: explicit ExpressionTrunc(const boost::intrusive_ptr& expCtx) : ExpressionRangedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionType final : public ExpressionFixedArity { public: explicit ExpressionType(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionWeek final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionWeek(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$week"_sd, expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.week(date)); } }; class ExpressionIsoWeekYear final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionIsoWeekYear(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$isoWeekYear"_sd, expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.isoYear(date)); } }; class ExpressionIsoDayOfWeek final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionIsoDayOfWeek(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$isoDayOfWeek"_sd, expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.isoDayOfWeek(date)); } }; class ExpressionIsoWeek final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionIsoWeek(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$isoWeek", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.isoWeek(date)); } }; class ExpressionYear final : public DateExpressionAcceptingTimeZone { public: explicit ExpressionYear(const boost::intrusive_ptr& expCtx) : DateExpressionAcceptingTimeZone("$year", expCtx) {} Value evaluateDate(Date_t date, const TimeZone& timeZone) const final { return Value(timeZone.dateParts(date).year); } }; class ExpressionZip final : public Expression { public: explicit ExpressionZip(const boost::intrusive_ptr& expCtx) : Expression(expCtx) {} Value evaluate(const Document& root) const final; boost::intrusive_ptr optimize() final; static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vpsIn); Value serialize(bool explain) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: bool _useLongestLength = false; ExpressionVector _inputs; ExpressionVector _defaults; }; class ExpressionConvert final : public Expression { public: /** * Creates a $convert expression converting from 'input' to the type given by 'toType'. Leaves * 'onNull' and 'onError' unspecified. */ static boost::intrusive_ptr create(const boost::intrusive_ptr&, const boost::intrusive_ptr& input, BSONType toType); static boost::intrusive_ptr parse( const boost::intrusive_ptr& expCtx, BSONElement expr, const VariablesParseState& vpsIn); explicit ExpressionConvert(const boost::intrusive_ptr& expCtx) : Expression(expCtx) {} Value evaluate(const Document& root) const final; boost::intrusive_ptr optimize() final; Value serialize(bool explain) const final; protected: void _doAddDependencies(DepsTracker* deps) const final; private: ExpressionConvert(const boost::intrusive_ptr&, const boost::intrusive_ptr& input, BSONType toType); BSONType computeTargetType(Value typeName) const; Value performConversion(BSONType targetType, Value inputValue) const; boost::intrusive_ptr _input; boost::intrusive_ptr _to; boost::intrusive_ptr _onError; boost::intrusive_ptr _onNull; }; class ExpressionRegexFind final : public ExpressionFixedArity { public: explicit ExpressionRegexFind(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionRegexFindAll final : public ExpressionFixedArity { public: explicit ExpressionRegexFindAll(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; class ExpressionRegexMatch final : public ExpressionFixedArity { public: explicit ExpressionRegexMatch(const boost::intrusive_ptr& expCtx) : ExpressionFixedArity(expCtx) {} Value evaluate(const Document& root) const final; const char* getOpName() const final; }; }