| 1 | #include "duckdb/planner/expression/bound_aggregate_expression.hpp" |
| 2 | #include "duckdb/parser/expression/function_expression.hpp" |
| 3 | |
| 4 | #include "duckdb/catalog/catalog_entry/aggregate_function_catalog_entry.hpp" |
| 5 | #include "duckdb/common/types/hash.hpp" |
| 6 | #include "duckdb/common/field_writer.hpp" |
| 7 | #include "duckdb/planner/expression/bound_cast_expression.hpp" |
| 8 | #include "duckdb/function/function_serialization.hpp" |
| 9 | |
| 10 | namespace duckdb { |
| 11 | |
| 12 | BoundAggregateExpression::BoundAggregateExpression(AggregateFunction function, vector<unique_ptr<Expression>> children, |
| 13 | unique_ptr<Expression> filter, unique_ptr<FunctionData> bind_info, |
| 14 | AggregateType aggr_type) |
| 15 | : Expression(ExpressionType::BOUND_AGGREGATE, ExpressionClass::BOUND_AGGREGATE, function.return_type), |
| 16 | function(std::move(function)), children(std::move(children)), bind_info(std::move(bind_info)), |
| 17 | aggr_type(aggr_type), filter(std::move(filter)) { |
| 18 | D_ASSERT(!function.name.empty()); |
| 19 | } |
| 20 | |
| 21 | string BoundAggregateExpression::ToString() const { |
| 22 | return FunctionExpression::ToString<BoundAggregateExpression, Expression, BoundOrderModifier>( |
| 23 | entry: *this, schema: string(), function_name: function.name, is_operator: false, distinct: IsDistinct(), filter: filter.get(), order_bys: order_bys.get()); |
| 24 | } |
| 25 | |
| 26 | hash_t BoundAggregateExpression::Hash() const { |
| 27 | hash_t result = Expression::Hash(); |
| 28 | result = CombineHash(left: result, right: function.Hash()); |
| 29 | result = CombineHash(left: result, right: duckdb::Hash(value: IsDistinct())); |
| 30 | return result; |
| 31 | } |
| 32 | |
| 33 | bool BoundAggregateExpression::Equals(const BaseExpression &other_p) const { |
| 34 | if (!Expression::Equals(other: other_p)) { |
| 35 | return false; |
| 36 | } |
| 37 | auto &other = other_p.Cast<BoundAggregateExpression>(); |
| 38 | if (other.aggr_type != aggr_type) { |
| 39 | return false; |
| 40 | } |
| 41 | if (other.function != function) { |
| 42 | return false; |
| 43 | } |
| 44 | if (children.size() != other.children.size()) { |
| 45 | return false; |
| 46 | } |
| 47 | if (!Expression::Equals(left: other.filter, right: filter)) { |
| 48 | return false; |
| 49 | } |
| 50 | for (idx_t i = 0; i < children.size(); i++) { |
| 51 | if (!Expression::Equals(left: *children[i], right: *other.children[i])) { |
| 52 | return false; |
| 53 | } |
| 54 | } |
| 55 | if (!FunctionData::Equals(left: bind_info.get(), right: other.bind_info.get())) { |
| 56 | return false; |
| 57 | } |
| 58 | if (!BoundOrderModifier::Equals(left: order_bys, right: other.order_bys)) { |
| 59 | return false; |
| 60 | } |
| 61 | return true; |
| 62 | } |
| 63 | |
| 64 | bool BoundAggregateExpression::PropagatesNullValues() const { |
| 65 | return function.null_handling == FunctionNullHandling::SPECIAL_HANDLING ? false |
| 66 | : Expression::PropagatesNullValues(); |
| 67 | } |
| 68 | |
| 69 | unique_ptr<Expression> BoundAggregateExpression::Copy() { |
| 70 | vector<unique_ptr<Expression>> new_children; |
| 71 | new_children.reserve(n: children.size()); |
| 72 | for (auto &child : children) { |
| 73 | new_children.push_back(x: child->Copy()); |
| 74 | } |
| 75 | auto new_bind_info = bind_info ? bind_info->Copy() : nullptr; |
| 76 | auto new_filter = filter ? filter->Copy() : nullptr; |
| 77 | auto copy = make_uniq<BoundAggregateExpression>(args&: function, args: std::move(new_children), args: std::move(new_filter), |
| 78 | args: std::move(new_bind_info), args&: aggr_type); |
| 79 | copy->CopyProperties(other&: *this); |
| 80 | copy->order_bys = order_bys ? order_bys->Copy() : nullptr; |
| 81 | return std::move(copy); |
| 82 | } |
| 83 | |
| 84 | void BoundAggregateExpression::Serialize(FieldWriter &writer) const { |
| 85 | writer.WriteField(element: IsDistinct()); |
| 86 | writer.WriteOptional(element: filter); |
| 87 | writer.WriteOptional(element: order_bys); |
| 88 | FunctionSerializer::Serialize<AggregateFunction>(writer, function, return_type, children, bind_info: bind_info.get()); |
| 89 | } |
| 90 | |
| 91 | unique_ptr<Expression> BoundAggregateExpression::Deserialize(ExpressionDeserializationState &state, |
| 92 | FieldReader &reader) { |
| 93 | auto distinct = reader.ReadRequired<bool>(); |
| 94 | auto filter = reader.ReadOptional<Expression>(default_value: nullptr, args&: state.gstate); |
| 95 | auto order_bys = reader.ReadOptional<BoundOrderModifier>(default_value: nullptr, args&: state.gstate); |
| 96 | vector<unique_ptr<Expression>> children; |
| 97 | unique_ptr<FunctionData> bind_info; |
| 98 | auto function = FunctionSerializer::Deserialize<AggregateFunction, AggregateFunctionCatalogEntry>( |
| 99 | reader, state, type: CatalogType::AGGREGATE_FUNCTION_ENTRY, children, bind_info); |
| 100 | |
| 101 | auto x = make_uniq<BoundAggregateExpression>(args&: function, args: std::move(children), args: std::move(filter), args: std::move(bind_info), |
| 102 | args: distinct ? AggregateType::DISTINCT : AggregateType::NON_DISTINCT); |
| 103 | x->order_bys = std::move(order_bys); |
| 104 | return std::move(x); |
| 105 | } |
| 106 | |
| 107 | } // namespace duckdb |
| 108 | |