[KYUUBI apache#758] [KYUUBI 661] Add UDF system_user

zhaomin1423 · yaooqinn · commit cd308f38adf4 · 2021-07-19T18:16:38.000+08:00
### _Why are the changes needed?_  add udf system_user. ### _How was this patch tested?_ - [ ] Add some test cases that check the changes thoroughly including negative and positive cases if possible - [ ] Add screenshots for manual tests if appropriate - [ ] [Run test](https://kyuubi.readthedocs.io/en/latest/tools/testing.html#running-tests) locally before make a pull request Closes apache#758 from zhaomin1423/661_add-system_user. Closes apache#758 70eba56 [Min Zhao] update system user name f5edc62 [Min Zhao] [KYUUBI 661] Add UDF system_user Authored-by: Min Zhao <zhaomin1423@163.com> Signed-off-by: Kent Yao <yao@apache.org>
diff --git a/docs/sql/functions.md b/docs/sql/functions.md
@@ -13,4 +13,5 @@ Kyuubi provides several auxiliary SQL functions as supplement to Spark's [Built-
 Name | Description | Return Type | Since
 --- | --- | --- | ---
 kyuubi_version | Return the version of Kyuubi Server | string | 1.3.0
+system_user | Return the system user name for the associated query engine | string | 1.3.0
 
diff --git a/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/udf/KDFRegistry.scala b/externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/udf/KDFRegistry.scala
@@ -36,6 +36,13 @@ object KDFRegistry {
     "string",
     "1.3.0")
 
+  val system_user: KyuubiDefinedFunction = create(
+    "system_user",
+    udf(() => System.getProperty("user.name")).asNonNullable(),
+    "Return the system user name for the associated query engine",
+    "string",
+    "1.3.0")
+
   def create(
     name: String,
     udf: UserDefinedFunction,
diff --git a/kyuubi-common/src/test/scala/org/apache/kyuubi/operation/JDBCTests.scala b/kyuubi-common/src/test/scala/org/apache/kyuubi/operation/JDBCTests.scala
@@ -359,4 +359,12 @@ trait JDBCTests extends BasicJDBCTests {
       assert(rs.getString(1) == KYUUBI_VERSION)
     }
   }
+
+  test("kyuubi defined function - system_user") {
+    withJdbcStatement() { statement =>
+      val rs = statement.executeQuery("SELECT system_user()")
+      assert(rs.next())
+      assert(rs.getString(1) == System.getProperty("user.name"))
+    }
+  }
 }

Original file line number	Diff line number	Diff line change
`@@ -359,4 +359,12 @@ trait JDBCTests extends BasicJDBCTests {`
`359`	`359`	`assert(rs.getString(1) == KYUUBI_VERSION)`
`360`	`360`	`}`
`361`	`361`	`}`
	`362`	`+`
	`363`	`+ test("kyuubi defined function - system_user") {`
	`364`	`+ withJdbcStatement() { statement =>`
	`365`	`+ val rs = statement.executeQuery("SELECT system_user()")`
	`366`	`+ assert(rs.next())`
	`367`	`+ assert(rs.getString(1) == System.getProperty("user.name"))`
	`368`	`+ }`
	`369`	`+ }`
`362`	`370`	`}`