update

zhaohehuhu · Nov 6, 2023 · 05a9480 · 05a9480
1 parent 5acc919
commit 05a9480
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/.../main/resources/META-INF/services/org.apache.kyuubi.plugin.spark.authz.serde.URIExtractor b/.../main/resources/META-INF/services/org.apache.kyuubi.plugin.spark.authz.serde.URIExtractor
@@ -16,6 +16,6 @@
 #
 
 org.apache.kyuubi.plugin.spark.authz.serde.CatalogStorageFormatURIExtractor
-org.apache.kyuubi.plugin.spark.authz.serde.HadoopFsRelationFileIndexURIExtractor
+org.apache.kyuubi.plugin.spark.authz.serde.BaseRelationFileIndexURIExtractor
 org.apache.kyuubi.plugin.spark.authz.serde.OptionsUriExtractor
 org.apache.kyuubi.plugin.spark.authz.serde.StringURIExtractor
diff --git a/extensions/spark/kyuubi-spark-authz/src/main/resources/scan_command_spec.json b/extensions/spark/kyuubi-spark-authz/src/main/resources/scan_command_spec.json
@@ -26,7 +26,8 @@
   "functionDescs" : [ ],
   "uriDescs" : [ {
     "fieldName" : "relation",
-    "fieldExtractor" : "HadoopFsRelationFileIndexURIExtractor",
+    "fieldExtractor" : "BaseRelationFileIndexURIExtractor",
+    "actionTypeDesc" : null,
     "isInput" : false
   } ]
 }, {

diff --git a/...spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/serde/uriExtractors.scala b/...spark-authz/src/main/scala/org/apache/kyuubi/plugin/spark/authz/serde/uriExtractors.scala
@@ -49,7 +49,7 @@ class OptionsUriExtractor extends URIExtractor {
   }
 }
 
-class HadoopFsRelationFileIndexURIExtractor extends URIExtractor {
+class BaseRelationFileIndexURIExtractor extends URIExtractor {
   override def apply(v1: AnyRef): Seq[Uri] = {
     v1.asInstanceOf[HadoopFsRelation].location.rootPaths.map(_.toString).map(Uri)
   }

diff --git a/...rk/kyuubi-spark-authz/src/test/scala/org/apache/kyuubi/plugin/spark/authz/gen/Scans.scala b/...rk/kyuubi-spark-authz/src/test/scala/org/apache/kyuubi/plugin/spark/authz/gen/Scans.scala
@@ -37,7 +37,7 @@ object Scans extends CommandSpecs[ScanSpec] {
       ScanDesc(
         "catalogTable",
         classOf[CatalogTableOptionTableExtractor])
-    val uriDesc = UriDesc("relation", classOf[HadoopFsRelationFileIndexURIExtractor])
+    val uriDesc = UriDesc("relation", classOf[BaseRelationFileIndexURIExtractor])
     ScanSpec(r, Seq(tableDesc), uriDescs = Seq(uriDesc))
   }