Skip to content

Commit bdc8300

Browse files
committed
address feedback
1 parent 36c0439 commit bdc8300

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

spark/src/main/scala/org/apache/comet/ExtendedExplainInfo.scala

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,20 @@ class ExtendedExplainInfo extends ExtendedExplainGenerator {
8383
// generates the extended info in a verbose manner, printing each node along with the
8484
// extended information in a tree display
8585
def generateVerboseExtendedInfo(plan: SparkPlan): String = {
86+
val planStats = new CometCoverageStats()
87+
val outString = new StringBuilder()
88+
generateTreeString(getActualPlan(plan), 0, Seq(), 0, outString, planStats)
89+
val eligible = planStats.sparkOperators + planStats.cometOperators
90+
val converted =
91+
if (eligible == 0) 0.0 else planStats.cometOperators.toDouble / eligible * 100.0
92+
val summary = s"Comet accelerated ${planStats.cometOperators} out of $eligible " +
93+
s"eligible operators (${converted.toInt}%). " +
94+
s"Final plan contains ${planStats.transitions} transitions."
95+
s"${outString.toString()}\n$summary"
96+
}
97+
98+
/** Get the coverage statistics without the full plan */
99+
def generateCoverageInfo(plan: SparkPlan): String = {
86100
val planStats = new CometCoverageStats()
87101
val outString = new StringBuilder()
88102
generateTreeString(getActualPlan(plan), 0, Seq(), 0, outString, planStats)

0 commit comments

Comments
 (0)