fix

apache · Feb 23, 2024 · 97db3c9 · 97db3c9
1 parent 5442296
commit 97db3c9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kyuubi-common/src/main/scala/org/apache/kyuubi/engine/result/TRowSetGenerator.scala b/kyuubi-common/src/main/scala/org/apache/kyuubi/engine/result/TRowSetGenerator.scala
@@ -61,11 +61,13 @@ trait TRowSetGenerator[SchemaT, RowT, ColumnT]
     val tRowSet = new TRowSet(0, new JArrayList[TRow](rowSize))
     var i = 0
     val columnSize = getColumnSizeFromSchemaType(schema)
+    val tColumns = new JArrayList[TColumn](columnSize)
     while (i < columnSize) {
       val tColumn = toTColumn(rows, i, getColumnType(schema, i))
-      tRowSet.addToColumns(tColumn)
+      tColumns.add(tColumn)
       i += 1
     }
+    tRowSet.setColumns(tColumns)
     tRowSet
   }
 }