Update PrivilegesBuilder.scala

AngersZhuuuu · AngersZhuuuu · commit 3cbba422a6fa · 2023-11-01T13:40:56.000+08:00
diff --git a/extensions/spark/kyuubi-spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/PrivilegesBuilder.scala b/extensions/spark/kyuubi-spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/PrivilegesBuilder.scala
@@ -122,8 +122,9 @@ object PrivilegesBuilder {
         privilegeObjects += PrivilegeObject(table)
 
       case p =>
+        val newProjection = projectionList ++ p.inputSet.map(_.toAttribute)
         for (child <- p.children) {
-          buildQuery(child, privilegeObjects, projectionList, conditionList, spark)
+          buildQuery(child, privilegeObjects, newProjection, conditionList, spark)
         }
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -122,8 +122,9 @@ object PrivilegesBuilder {`
`122`	`122`	`privilegeObjects += PrivilegeObject(table)`
`123`	`123`
`124`	`124`	`case p =>`
	`125`	`+ val newProjection = projectionList ++ p.inputSet.map(_.toAttribute)`
`125`	`126`	`for (child <- p.children) {`
`126`		`- buildQuery(child, privilegeObjects, projectionList, conditionList, spark)`
	`127`	`+ buildQuery(child, privilegeObjects, newProjection, conditionList, spark)`
`127`	`128`	`}`
`128`	`129`	`}`
`129`	`130`	`}`