Update PrivilegesBuilder.scala

zhaohehuhu · Nov 1, 2023 · 3cbba42 · 3cbba42
1 parent 9cda08b
commit 3cbba42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...i-spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/PrivilegesBuilder.scala b/...i-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)
         }
     }
   }