[bug] NPE occurs for StringType when column value is null.

apache · Sep 14, 2023 · 283f930 · 283f930
1 parent 0daf6c4
commit 283f930
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/SchemaUtils.scala b/spark-doris-connector/src/main/scala/org/apache/doris/spark/sql/SchemaUtils.scala
@@ -156,7 +156,7 @@ private[spark] object SchemaUtils {
       case LongType => row.getLong(ordinal)
       case FloatType => row.getFloat(ordinal)
       case DoubleType => row.getDouble(ordinal)
-      case StringType => row.getUTF8String(ordinal).toString
+      case StringType => Option(row.getUTF8String(ordinal)).map(_.toString).getOrElse(DataUtil.NULL_VALUE)
       case TimestampType =>
         LocalDateTime.ofEpochSecond(row.getLong(ordinal) / 100000, (row.getLong(ordinal) % 1000).toInt, ZoneOffset.UTC)
         new Timestamp(row.getLong(ordinal) / 1000).toString