trino engine add default result schema

apache · Jan 13, 2024 · dfc987e · dfc987e
1 parent bffbdbd
commit dfc987e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...ls/kyuubi-trino-engine/src/main/scala/org/apache/kyuubi/engine/trino/TrinoStatement.scala b/...ls/kyuubi-trino-engine/src/main/scala/org/apache/kyuubi/engine/trino/TrinoStatement.scala
@@ -25,6 +25,7 @@ import scala.concurrent.ExecutionContext
 
 import com.google.common.base.Verify
 import io.trino.client.ClientSession
+import io.trino.client.ClientTypeSignature
 import io.trino.client.Column
 import io.trino.client.StatementClient
 import io.trino.client.StatementClientFactory
@@ -46,6 +47,9 @@ class TrinoStatement(
     sql: String,
     operationLog: Option[OperationLog]) extends Logging {
 
+  private val defaultSchema: List[Column] =
+    List(new Column("Result", "VARCHAR", new ClientTypeSignature("VARCHAR")))
+
   private lazy val trino = StatementClientFactory
     .newStatementClient(trinoContext.httpClient, trinoContext.clientSession.get, sql)
 
@@ -68,6 +72,9 @@ class TrinoStatement(
       val columns = results.getColumns()
       if (columns != null) {
         info(s"Execute with Trino query id: ${results.getId}")
+        if (columns.isEmpty) {
+          return defaultSchema
+        }
         return columns.asScala.toList
       }
       trino.advance()