fix with bigint

apache · Aug 12, 2024 · 6515cc0 · 6515cc0
1 parent 80e601a
commit 6515cc0
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/paimon-core/src/main/java/org/apache/paimon/casting/TimestampToNumericPrimitiveCastRule.java b/paimon-core/src/main/java/org/apache/paimon/casting/TimestampToNumericPrimitiveCastRule.java
@@ -44,12 +44,15 @@ private TimestampToNumericPrimitiveCastRule() {
 
     @Override
     public CastExecutor<Timestamp, Number> create(DataType inputType, DataType targetType) {
-        if (inputType.is(DataTypeRoot.TIMESTAMP_WITHOUT_TIME_ZONE)) {
-            return value -> DateTimeUtils.unixTimestamp(value.getMillisecond());
-        } else if (inputType.is(DataTypeRoot.TIMESTAMP_WITH_LOCAL_TIME_ZONE)) {
-            return value ->
-                    DateTimeUtils.unixTimestamp(
-                            Timestamp.fromLocalDateTime(value.toLocalDateTime()).getMillisecond());
+        if (targetType.is(DataTypeRoot.BIGINT)) {
+            if (inputType.is(DataTypeRoot.TIMESTAMP_WITHOUT_TIME_ZONE)) {
+                return value -> DateTimeUtils.unixTimestamp(value.getMillisecond());
+            } else if (inputType.is(DataTypeRoot.TIMESTAMP_WITH_LOCAL_TIME_ZONE)) {
+                return value ->
+                        DateTimeUtils.unixTimestamp(
+                                Timestamp.fromLocalDateTime(value.toLocalDateTime())
+                                        .getMillisecond());
+            }
         }
         return null;
     }