improve MetricNode: fix metric missing in Union children (#832)

Co-authored-by: zhangli20 <zhangli20@kuaishou.com>
kwai · Feb 18, 2025 · 6cf2894 · 6cf2894
1 parent 3e14226
commit 6cf2894
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/spark-extension/src/main/scala/org/apache/spark/sql/blaze/MetricNode.scala b/spark-extension/src/main/scala/org/apache/spark/sql/blaze/MetricNode.scala
@@ -24,8 +24,13 @@ case class MetricNode(
     metricValueHandler: Option[(String, Long) => Unit] = None)
     extends Logging {
 
-  def getChild(i: Int): MetricNode =
-    children(i)
+  def getChild(i: Int): MetricNode = {
+    if (i < children.length) {
+      children(i)
+    } else {
+      null
+    }
+  }
 
   def add(metricName: String, v: Long): Unit = {
     metricValueHandler.foreach(_.apply(metricName, v))