spark sql 写入 smallint 需要什么类型

在 Spark SQL 中写入 smallint，可以使用 ShortType 类型。例如：

import org.apache.spark.sql.types.ShortType

val schema = StructType(Seq(
  StructField("id", ShortType, true),
  StructField("name", StringType, true)
))

val data = Seq(
  Row(1.toShort, "John"),
  Row(2.toShort, "Jane")
)

val df = spark.createDataFrame(
  spark.sparkContext.parallelize(data),
  schema
)

df.show()

这里使用 ShortType 定义了 id 字段的类型。在数据中使用 toShort 将整型转换为 smallint 类型。