apache/kyuubi
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

FrontendServiceSuite get info

Browse Source
This commit is contained in:
Kent Yao 2020-10-22 18:30:55 +08:00
parent fb1b806fdf
commit 767a296594
2 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
kyuubi-common/src/main/scala/org/apache/kyuubi/service/FrontendService.scala
View File

@ -217,6 +217,7 @@ class FrontendService private (name: String, be: BackendService, oomHook: Runnab
} catch {
case e: Exception =>
warn("Error getting type info: ", e)
resp.setInfoValue(TGetInfoValue.lenValue(0))
resp.setStatus(KyuubiSQLException.toTStatus(e))
}
resp

32
kyuubi-common/src/test/scala/org/apache/kyuubi/service/FrontendServiceSuite.scala
View File

@ -115,6 +115,38 @@ class FrontendServiceSuite extends KyuubiFunSuite {
}
}
test("get info") {
withSessionHandle { (client, handle) =>
val req = new TGetInfoReq()
req.setSessionHandle(handle)
req.setInfoType(TGetInfoType.CLI_DBMS_VER)
val resp = client.GetInfo(req)
assert(resp.getInfoValue.getStringValue === org.apache.kyuubi.KYUUBI_VERSION)
req.setInfoType(TGetInfoType.CLI_SERVER_NAME)
assert(client.GetInfo(req).getInfoValue.getStringValue === "Kyuubi")
req.setInfoType(TGetInfoType.CLI_DBMS_NAME)
assert(client.GetInfo(req).getInfoValue.getStringValue === "Spark SQL")
req.setInfoType(TGetInfoType.CLI_MAX_COLUMN_NAME_LEN)
assert(client.GetInfo(req).getInfoValue.getLenValue === 128)
req.setInfoType(TGetInfoType.CLI_MAX_SCHEMA_NAME_LEN)
assert(client.GetInfo(req).getInfoValue.getLenValue === 128)
req.setInfoType(TGetInfoType.CLI_MAX_TABLE_NAME_LEN)
assert(client.GetInfo(req).getInfoValue.getLenValue === 128)
req.setInfoType(TGetInfoType.CLI_ACCESSIBLE_PROCEDURES)
val resp1 = client.GetInfo(req)
assert(resp1.getInfoValue.getLenValue === 0)
assert(resp1.getStatus.getStatusCode === TStatusCode.ERROR_STATUS)
assert(resp1.getStatus.getErrorMessage ===
"Unrecognized GetInfoType value: CLI_ACCESSIBLE_PROCEDURES")
req.setInfoType(TGetInfoType.CLI_USER_NAME)
val resp2 = client.GetInfo(req)
assert(resp2.getInfoValue.getLenValue === 0)
assert(resp2.getStatus.getStatusCode === TStatusCode.ERROR_STATUS)
assert(resp2.getStatus.getErrorMessage ===
"Unrecognized GetInfoType value: CLI_USER_NAME")
}
}
test("get catalogs") {
withSessionHandle { (client, handle) =>
val req = new TGetCatalogsReq(handle)