Skip to content

Commit fdf1a1b

Browse files
committed
chore: experiment to obtain rule output plan
1 parent 6d9d3d8 commit fdf1a1b

3 files changed

Lines changed: 7 additions & 2 deletions

File tree

spark/src/main/scala/org/apache/comet/rules/CometExecRule.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,10 +357,12 @@ case class CometExecRule(session: SparkSession) extends Rule[SparkPlan] {
357357
override def apply(plan: SparkPlan): SparkPlan = {
358358
val newPlan = _apply(plan)
359359
if (showTransformations && !newPlan.fastEquals(plan)) {
360-
logInfo(s"""
360+
// scalastyle:off println
361+
println(s"""
361362
|=== Applying Rule $ruleName ===
362363
|${sideBySide(plan.treeString, newPlan.treeString).mkString("\n")}
363364
|""".stripMargin)
365+
// scalastyle:on println
364366
}
365367
newPlan
366368
}

spark/src/main/scala/org/apache/comet/rules/CometScanRule.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,12 @@ case class CometScanRule(session: SparkSession)
6565
override def apply(plan: SparkPlan): SparkPlan = {
6666
val newPlan = _apply(plan)
6767
if (showTransformations && !newPlan.fastEquals(plan)) {
68-
logInfo(s"""
68+
// scalastyle:off println
69+
println(s"""
6970
|=== Applying Rule $ruleName ===
7071
|${sideBySide(plan.treeString, newPlan.treeString).mkString("\n")}
7172
|""".stripMargin)
73+
// scalastyle:on println
7274
}
7375
newPlan
7476
}

spark/src/test/scala/org/apache/comet/parquet/CometParquetWriterSuite.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ class CometParquetWriterSuite extends CometTestBase {
6969

7070
withSQLConf(
7171
CometConf.COMET_NATIVE_PARQUET_WRITE_ENABLED.key -> "true",
72+
CometConf.COMET_EXPLAIN_TRANSFORMATIONS.key -> "true",
7273
SQLConf.SESSION_LOCAL_TIMEZONE.key -> "America/Halifax",
7374
CometConf.COMET_OPERATOR_DATA_WRITING_COMMAND_ALLOW_INCOMPAT.key -> "true",
7475
CometConf.COMET_EXEC_ENABLED.key -> "true") {

0 commit comments

Comments
 (0)