Skip to content

Commit 4286354

Browse files
committed
Support flink engine under the select statement, the results can be read in a stream
1 parent e5b74b0 commit 4286354

File tree

1 file changed

+3
-6
lines changed
  • externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/schema

1 file changed

+3
-6
lines changed

externals/kyuubi-spark-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/schema/SchemaHelper.scala

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -140,14 +140,11 @@ object SchemaHelper {
140140
case dt
141141
if Array(TIMESTAMP_NTZ, DAY_TIME_INTERVAL, YEAR_MONTH_INTERVAL)
142142
.contains(dt.getClass.getSimpleName) => Some(dt.defaultSize)
143+
case dt @ (_: DecimalType) =>
144+
Some(dt.precision)
143145
case dt @ (BooleanType | _: NumericType | DateType | TimestampType |
144146
CalendarIntervalType | NullType) =>
145-
// decimal type
146-
if (dt.isInstanceOf[DecimalType]) {
147-
Some(dt.asInstanceOf[DecimalType].precision)
148-
} else {
149-
Some(dt.defaultSize)
150-
}
147+
Some(dt.defaultSize)
151148
case StructType(fields) =>
152149
val sizeArr = fields.map(f => getColumnSize(f.dataType))
153150
if (sizeArr.contains(None)) {

0 commit comments

Comments
 (0)