@@ -19,7 +19,7 @@ package org.apache.spark.sql.streaming.ui
1919
2020import java .{util => ju }
2121import java .lang .{Long => JLong }
22- import java .util .UUID
22+ import java .util .{ Locale , UUID }
2323import javax .servlet .http .HttpServletRequest
2424
2525import 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