[KYUUBI apache#5995][Improvement] Avoid unnecessary loop for injected…

… analyzer rule
AngersZhuuuu · Jan 19, 2024 · 82db80e · 82db80e
1 parent 162e72b
commit 82db80e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/...thz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/ranger/RangerSparkExtension.scala b/...thz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/ranger/RangerSparkExtension.scala
@@ -45,12 +45,12 @@ class RangerSparkExtension extends (SparkSessionExtensions => Unit) {
 
   override def apply(v1: SparkSessionExtensions): Unit = {
     v1.injectCheckRule(AuthzConfigurationChecker)
-    v1.injectResolutionRule(_ => RuleReplaceShowObjectCommands)
-    v1.injectResolutionRule(_ => RuleApplyPermanentViewMarker)
-    v1.injectResolutionRule(_ => RuleApplyTypeOfMarker)
-    v1.injectResolutionRule(RuleApplyRowFilter)
-    v1.injectResolutionRule(RuleApplyDataMaskingStage0)
-    v1.injectResolutionRule(RuleApplyDataMaskingStage1)
+    v1.injectPostHocResolutionRule(_ => RuleReplaceShowObjectCommands)
+    v1.injectPostHocResolutionRule(_ => RuleApplyPermanentViewMarker)
+    v1.injectPostHocResolutionRule(_ => RuleApplyTypeOfMarker)
+    v1.injectPostHocResolutionRule(RuleApplyRowFilter)
+    v1.injectPostHocResolutionRule(RuleApplyDataMaskingStage0)
+    v1.injectPostHocResolutionRule(RuleApplyDataMaskingStage1)
     v1.injectOptimizerRule(_ => RuleEliminateMarker)
     v1.injectOptimizerRule(RuleAuthorization)
     v1.injectOptimizerRule(RuleEliminatePermanentViewMarker)