[doris] branch branch-1.2-lts updated: [vectorized](function) suppoort date_trunc function truncate week mode (#18334) (#20530)

[lide]

This is an automated email from the ASF dual-hosted git repository.

lide pushed a commit to branch branch-1.2-lts
in repository https://gitbox.apache.org/repos/asf/doris.git


The following commit(s) were added to refs/heads/branch-1.2-lts by this push:
     new 385f79b98c [vectorized](function) suppoort date_trunc function 
truncate week mode (#18334) (#20530)
385f79b98c is described below

commit 385f79b98c003414a4e631446a064fc97097545b
Author: yuxuan-luo <119841515+yuxuan-...@users.noreply.github.com>
AuthorDate: Wed Jun 7 14:44:36 2023 +0800

    [vectorized](function) suppoort date_trunc function truncate week mode 
(#18334) (#20530)
    
    support date_trunc could truncate week eg:
    select date_trunc('2023-4-3 19:28:30', 'week');
    
    (cherry picked from commit 50e6c4216a26ec6a667060bf1bae6fc6dee217df)
    
    Co-authored-by: zhangstar333 
<87313068+zhangstar...@users.noreply.github.com>
---
 be/src/vec/functions/function_timestamp.cpp |  2 ++
 be/src/vec/runtime/vdatetime_value.cpp      | 17 +++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/be/src/vec/functions/function_timestamp.cpp 
b/be/src/vec/functions/function_timestamp.cpp
index 704fc8f3ff..ad8af1d676 100644
--- a/be/src/vec/functions/function_timestamp.cpp
+++ b/be/src/vec/functions/function_timestamp.cpp
@@ -298,6 +298,8 @@ struct DateTrunc {
                 null_map[i] = !dt.template datetime_trunc<MINUTE>();
             } else if (std::strncmp("second", str_data, 6) == 0) {
                 null_map[i] = !dt.template datetime_trunc<SECOND>();
+            } else if (std::strncmp("week", str_data, 4) == 0) {
+                null_map[i] = !dt.template datetime_trunc<WEEK>();
             } else {
                 null_map[i] = 1;
             }
diff --git a/be/src/vec/runtime/vdatetime_value.cpp 
b/be/src/vec/runtime/vdatetime_value.cpp
index d5086ab6b0..a9e98b2ef8 100644
--- a/be/src/vec/runtime/vdatetime_value.cpp
+++ b/be/src/vec/runtime/vdatetime_value.cpp
@@ -1691,6 +1691,14 @@ bool VecDateTimeValue::datetime_trunc() {
         _hour = 0;
         break;
     }
+    case WEEK: {
+        _second = 0;
+        _minute = 0;
+        _hour = 0;
+        TimeInterval interval(DAY, weekday(), true);
+        date_add_interval<DAY>(interval);
+        break;
+    }
     case MONTH: {
         _second = 0;
         _minute = 0;
@@ -2704,6 +2712,15 @@ bool DateV2Value<T>::datetime_trunc() {
             date_v2_value_.hour_ = 0;
             break;
         }
+        case WEEK: {
+            date_v2_value_.microsecond_ = 0;
+            date_v2_value_.second_ = 0;
+            date_v2_value_.minute_ = 0;
+            date_v2_value_.hour_ = 0;
+            TimeInterval interval(DAY, weekday(), true);
+            date_add_interval<DAY>(interval);
+            break;
+        }
         case MONTH: {
             date_v2_value_.microsecond_ = 0;
             date_v2_value_.second_ = 0;


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org