[MINOR] Define gluten ui enabled key in GlutenConfig (#4164)

minor fix, define gluten ui enabled key in GlutenConfig
apache · Dec 23, 2023 · 7fff4e5 · 7fff4e5
1 parent f956945
commit 7fff4e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/gluten-core/src/main/scala/io/glutenproject/GlutenPlugin.scala b/gluten-core/src/main/scala/io/glutenproject/GlutenPlugin.scala
@@ -81,7 +81,7 @@ private[glutenproject] class GlutenDriverPlugin extends DriverPlugin with Loggin
   }
 
   override def registerMetrics(appId: String, pluginContext: PluginContext): Unit = {
-    if (pluginContext.conf().getBoolean("spark.gluten.ui.enabled", true)) {
+    if (pluginContext.conf().getBoolean(GlutenConfig.GLUTEN_UI_ENABLED, true)) {
       _sc.foreach {
         sc =>
           GlutenEventUtils.attachUI(sc)

diff --git a/shims/common/src/main/scala/io/glutenproject/GlutenConfig.scala b/shims/common/src/main/scala/io/glutenproject/GlutenConfig.scala
@@ -439,6 +439,8 @@ object GlutenConfig {
   // Tokens of current user, split by `\0`
   val GLUTEN_UGI_TOKENS = "spark.gluten.ugi.tokens"
 
+  val GLUTEN_UI_ENABLED = "spark.gluten.ui.enabled"
+
   var ins: GlutenConfig = _
 
   def getConf: GlutenConfig = {