Lordworms commented on code in PR #10218:
URL: https://github.com/apache/datafusion/pull/10218#discussion_r1578376374
##########
datafusion/optimizer/src/eliminate_duplicated_expr.rs:
##########
@@ -35,78 +36,111 @@ impl EliminateDuplicatedExpr {
Self {}
}
}
-
+// use this structure to avoid initial clone
+#[derive(Eq, Clone, Debug)]
+struct SortExprWrapper {
+ expr: Expr,
+}
+impl PartialEq for SortExprWrapper {
+ fn eq(&self, other: &Self) -> bool {
+ match (&self.expr, &other.expr) {
+ (Expr::Sort(own_sort), Expr::Sort(other_sort)) => {
+ own_sort.expr == other_sort.expr
+ }
+ _ => self.expr == other.expr,
+ }
+ }
+}
+impl Hash for SortExprWrapper {
+ fn hash<H: Hasher>(&self, state: &mut H) {
+ match &self.expr {
+ Expr::Sort(sort) => {
+ sort.expr.hash(state);
+ }
+ _ => {
+ self.expr.hash(state);
+ }
+ }
+ }
+}
impl OptimizerRule for EliminateDuplicatedExpr {
fn try_optimize(
&self,
- plan: &LogicalPlan,
+ _plan: &LogicalPlan,
_config: &dyn OptimizerConfig,
) -> Result<Option<LogicalPlan>> {
+ internal_err!("Should have called EliminateDuplicatedExpr::rewrite")
+ }
+
+ fn apply_order(&self) -> Option<ApplyOrder> {
+ Some(ApplyOrder::TopDown)
+ }
+
+ fn supports_rewrite(&self) -> bool {
+ true
+ }
+
+ fn rewrite(
+ &self,
+ plan: LogicalPlan,
+ _config: &dyn OptimizerConfig,
+ ) -> Result<Transformed<LogicalPlan>> {
match plan {
LogicalPlan::Sort(sort) => {
- let normalized_sort_keys = sort
+ let len = sort.expr.len();
+ let normalized_sort_keys: Vec<_> = sort
.expr
- .iter()
- .map(|e| match e {
- Expr::Sort(ExprSort { expr, .. }) => {
- Expr::Sort(ExprSort::new(expr.clone(), true,
false))
Review Comment:
avoid the normalized clone here
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
