Skip to content

Commit 6afed7a

Browse files
committed
update RuleFunctionAuthorization
1 parent 85d19d2 commit 6afed7a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

extensions/spark/kyuubi-spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/ranger/RuleFunctionAuthorization.scala

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ package org.apache.kyuubi.plugin.spark.authz.ranger
1919

2020
import scala.collection.mutable
2121

22+
import org.apache.ranger.plugin.policyengine.RangerAccessRequest
2223
import org.apache.spark.sql.SparkSession
2324
import org.apache.spark.sql.catalyst.plans.logical.LogicalPlan
2425

@@ -51,11 +52,13 @@ case class RuleFunctionAuthorization(spark: SparkSession) extends (LogicalPlan =
5152

5253
addAccessRequest(inputs, isInput = true)
5354

54-
val requestArrays = requests.map(Seq(_))
55+
val requestSeq: Seq[RangerAccessRequest] =
56+
requests.map(_.asInstanceOf[RangerAccessRequest]).toSeq
57+
5558
if (authorizeInSingleCall) {
56-
verify(requestArrays.flatten, auditHandler)
59+
verify(requestSeq, auditHandler)
5760
} else {
58-
requestArrays.flatten.foreach { req =>
61+
requestSeq.foreach { req =>
5962
verify(Seq(req), auditHandler)
6063
}
6164
}

0 commit comments

Comments
 (0)