1

liugddx · Jul 26, 2024 · 87d468e · 87d468e
1 parent d2024fd
commit 87d468e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/be/src/vec/data_types/serde/data_type_datetimev2_serde.cpp b/be/src/vec/data_types/serde/data_type_datetimev2_serde.cpp
@@ -18,6 +18,8 @@
 #include "data_type_datetimev2_serde.h"
 
 #include <arrow/builder.h>
+#include <cctz/civil_time.h>
+#include <cctz/time_zone.h>
 
 #include <chrono> // IWYU pragma: keep
 
@@ -115,10 +117,24 @@ void DataTypeDateTimeV2SerDe::write_column_to_arrow(const IColumn& column, const
             if (scale > 3) {
                 uint32_t microsecond = datetime_val.microsecond();
                 timestamp = (timestamp * 1000000) + microsecond;
+                timestamp -= std::chrono::duration_cast<std::chrono::microseconds>(
+                                     cctz::convert(cctz::civil_second(1970, 1, 1, 0, 0, 0), ctz)
+                                             .time_since_epoch())
+                                     .count();
             } else if (scale > 0) {
                 uint32_t millisecond = datetime_val.microsecond() / 1000;
                 timestamp = (timestamp * 1000) + millisecond;
+                timestamp -= std::chrono::duration_cast<std::chrono::milliseconds>(
+                                     cctz::convert(cctz::civil_second(1970, 1, 1, 0, 0, 0), ctz)
+                                             .time_since_epoch())
+                                     .count();
+            } else if (scale == 0) {
+                timestamp -= std::chrono::duration_cast<std::chrono::seconds>(
+                                     cctz::convert(cctz::civil_second(1970, 1, 1, 0, 0, 0), ctz)
+                                             .time_since_epoch())
+                                     .count();
             }
+
             checkArrowStatus(timestamp_builder.Append(timestamp), column.get_name(),
                              array_builder->type()->name());
         }