1

liugddx · Jul 26, 2024 · 7f56fee · 7f56fee
1 parent 87d468e
commit 7f56fee
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 16 deletions.
diff --git a/be/src/util/arrow/row_batch.cpp b/be/src/util/arrow/row_batch.cpp
@@ -96,11 +96,11 @@ Status convert_to_arrow_type(const TypeDescriptor& type, std::shared_ptr<arrow::
         break;
     case TYPE_DATETIMEV2:
         if (type.scale > 3) {
-            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::MICRO);
+            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::MICRO, "UTC");
         } else if (type.scale > 0) {
-            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::MILLI);
+            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::MILLI, "UTC");
         } else {
-            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::SECOND);
+            *result = std::make_shared<arrow::TimestampType>(arrow::TimeUnit::SECOND, "UTC");
         }
         break;
     case TYPE_DECIMALV2:

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
@@ -117,22 +117,19 @@ 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();
+                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();
+                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(),