[HUDI-908] Add some data types to HoodieTestDataGenerator and fix some some bugs. (#1690)

2020-06-22 23:13:28 +08:00
parent 68a656b016
commit 89e37d5273
8 changed files with 127 additions and 16 deletions
--- a/hudi-spark/src/main/scala/org/apache/hudi/AvroConversionHelper.scala
+++ b/hudi-spark/src/main/scala/org/apache/hudi/AvroConversionHelper.scala
@@ -110,7 +110,11 @@ object AvroConversionHelper {
            if (item == null) {
              null
            } else {
-              new Date(item.asInstanceOf[Long])
+              if (item.isInstanceOf[Integer]) {
+                new Date(item.asInstanceOf[Integer].longValue())
+              } else {
+                new Date(item.asInstanceOf[Long])
+              }
            }
        case (TimestampType, LONG) =>
          (item: AnyRef) =>