Skip to content

Commit fab5faa

Browse files
committed
remove unnecessary collection operation in valid constraints generation
1 parent 80813e1 commit fab5faa

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/plans/logical/LogicalPlan.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -152,10 +152,10 @@ abstract class UnaryNode extends LogicalPlan {
152152
override final def children: Seq[LogicalPlan] = child :: Nil
153153

154154
/**
155-
* Generates an additional set of aliased constraints by replacing the original constraint
156-
* expressions with the corresponding alias
155+
* Generates all valid constraints including an set of aliased constraints by replacing the
156+
* original constraint expressions with the corresponding alias
157157
*/
158-
protected def getAliasedConstraints(projectList: Seq[NamedExpression]): Set[Expression] = {
158+
protected def getAllValidConstraints(projectList: Seq[NamedExpression]): Set[Expression] = {
159159
var allConstraints = child.constraints.asInstanceOf[Set[Expression]]
160160
projectList.foreach {
161161
case a @ Alias(l: Literal, _) =>
@@ -170,7 +170,7 @@ abstract class UnaryNode extends LogicalPlan {
170170
case _ => // Don't change.
171171
}
172172

173-
allConstraints -- child.constraints
173+
allConstraints
174174
}
175175

176176
override protected def validConstraints: Set[Expression] = child.constraints

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/plans/logical/basicLogicalOperators.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ case class Project(projectList: Seq[NamedExpression], child: LogicalPlan)
6464
}
6565

6666
override def validConstraints: Set[Expression] =
67-
child.constraints.union(getAliasedConstraints(projectList))
67+
getAllValidConstraints(projectList)
6868
}
6969

7070
/**
@@ -595,7 +595,7 @@ case class Aggregate(
595595

596596
override def validConstraints: Set[Expression] = {
597597
val nonAgg = aggregateExpressions.filter(_.find(_.isInstanceOf[AggregateExpression]).isEmpty)
598-
child.constraints.union(getAliasedConstraints(nonAgg))
598+
getAllValidConstraints(nonAgg)
599599
}
600600
}
601601

0 commit comments

Comments
 (0)