Skip to content

Commit e2f1594

Browse files
committed
Ignore case compare
1 parent 115de28 commit e2f1594

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

sql/core/src/main/scala/org/apache/spark/sql/streaming/ui/StreamingQueryStatisticsPage.scala

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ package org.apache.spark.sql.streaming.ui
1919

2020
import java.{util => ju}
2121
import java.lang.{Long => JLong}
22-
import java.util.UUID
22+
import java.util.{Locale, UUID}
2323
import javax.servlet.http.HttpServletRequest
2424

2525
import scala.collection.JavaConverters._
@@ -41,7 +41,8 @@ private[ui] class StreamingQueryStatisticsPage(parent: StreamingQueryTab)
4141
parent.parent.conf.get(STATE_STORE_PROVIDER_CLASS)).supportedCustomMetrics
4242
logDebug(s"Supported custom metrics: $supportedCustomMetrics")
4343

44-
private val enabledCustomMetrics = parent.parent.conf.get(ENABLED_STREAMING_UI_CUSTOM_METRIC_LIST)
44+
private val enabledCustomMetrics =
45+
parent.parent.conf.get(ENABLED_STREAMING_UI_CUSTOM_METRIC_LIST).map(_.toLowerCase(Locale.ROOT))
4546
logDebug(s"Enabled custom metrics: $enabledCustomMetrics")
4647

4748
def generateLoadResources(request: HttpServletRequest): Seq[Node] = {
@@ -271,11 +272,11 @@ private[ui] class StreamingQueryStatisticsPage(parent: StreamingQueryTab)
271272
// This is made sure on caller side but put it here to be defensive
272273
require(query.lastProgress.stateOperators.nonEmpty)
273274
query.lastProgress.stateOperators.head.customMetrics.keySet().asScala
274-
.filter(enabledCustomMetrics.contains(_)).map { metricName =>
275+
.filter(m => enabledCustomMetrics.contains(m.toLowerCase(Locale.ROOT))).map { metricName =>
275276
val data = query.recentProgress.map(p => (parseProgressTimestamp(p.timestamp),
276277
p.stateOperators.map(_.customMetrics.get(metricName).toDouble).sum))
277278
val max = data.maxBy(_._2)._2
278-
val metric = supportedCustomMetrics.find(_.name == metricName).get
279+
val metric = supportedCustomMetrics.find(_.name.equalsIgnoreCase(metricName)).get
279280

280281
val graphUIData =
281282
new GraphUIData(

0 commit comments

Comments
 (0)