Modify RowWriteSupport to use the exact types to cast.

ueshin · ueshin · commit 323d1d208940 · 2014-07-07T19:23:40.000+09:00
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableSupport.scala b/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableSupport.scala
@@ -192,9 +192,9 @@ private[parquet] class RowWriteSupport extends WriteSupport[Row] with Logging {
           )
         )
         case IntegerType => writer.addInteger(value.asInstanceOf[Int])
-        case ShortType => writer.addInteger(value.asInstanceOf[Int])
+        case ShortType => writer.addInteger(value.asInstanceOf[Short])
         case LongType => writer.addLong(value.asInstanceOf[Long])
-        case ByteType => writer.addInteger(value.asInstanceOf[Int])
+        case ByteType => writer.addInteger(value.asInstanceOf[Byte])
         case DoubleType => writer.addDouble(value.asInstanceOf[Double])
         case FloatType => writer.addFloat(value.asInstanceOf[Float])
         case BooleanType => writer.addBoolean(value.asInstanceOf[Boolean])

Original file line number	Diff line number	Diff line change
`@@ -192,9 +192,9 @@ private[parquet] class RowWriteSupport extends WriteSupport[Row] with Logging {`
`192`	`192`	`)`
`193`	`193`	`)`
`194`	`194`	`case IntegerType => writer.addInteger(value.asInstanceOf[Int])`
`195`		`- case ShortType => writer.addInteger(value.asInstanceOf[Int])`
	`195`	`+ case ShortType => writer.addInteger(value.asInstanceOf[Short])`
`196`	`196`	`case LongType => writer.addLong(value.asInstanceOf[Long])`
`197`		`- case ByteType => writer.addInteger(value.asInstanceOf[Int])`
	`197`	`+ case ByteType => writer.addInteger(value.asInstanceOf[Byte])`
`198`	`198`	`case DoubleType => writer.addDouble(value.asInstanceOf[Double])`
`199`	`199`	`case FloatType => writer.addFloat(value.asInstanceOf[Float])`
`200`	`200`	`case BooleanType => writer.addBoolean(value.asInstanceOf[Boolean])`