477 lines
16 KiB
C++
477 lines
16 KiB
C++
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
#include "exprs/timestamp_functions.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "runtime/exec_env.h"
|
|
#include "runtime/test_env.h"
|
|
#include "testutil/function_utils.h"
|
|
#include "udf/udf.h"
|
|
#include "udf/udf_internal.h"
|
|
|
|
namespace doris {
|
|
class FunctionContextImpl;
|
|
|
|
doris_udf::DateTimeVal datetime_val(int64_t value) {
|
|
DateTimeValue dt(value);
|
|
doris_udf::DateTimeVal tv;
|
|
dt.to_datetime_val(&tv);
|
|
return tv;
|
|
}
|
|
|
|
class TimestampFunctionsTest : public testing::Test {
|
|
public:
|
|
TimestampFunctionsTest() {}
|
|
|
|
void SetUp() override {
|
|
utils = new FunctionUtils();
|
|
ctx = utils->get_fn_ctx();
|
|
}
|
|
|
|
void TearDown() override { delete utils; }
|
|
|
|
FunctionUtils* utils = nullptr;
|
|
FunctionContext* ctx = nullptr;
|
|
};
|
|
|
|
TEST_F(TimestampFunctionsTest, day_of_week_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
doris_udf::DateTimeVal tv;
|
|
//2001-02-03 12:34:56
|
|
tv.packed_time = 1830650338932162560L;
|
|
tv.type = TIME_DATETIME;
|
|
|
|
EXPECT_EQ(7, TimestampFunctions::day_of_week(context, tv).val);
|
|
|
|
// 2020-00-01 00:00:00
|
|
DateTimeValue dtv2(20200001000000);
|
|
dtv2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_week(context, tv2).is_null);
|
|
|
|
// 2020-01-00 00:00:00
|
|
DateTimeValue dtv3(20200100000000);
|
|
dtv3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_week(context, tv3).is_null);
|
|
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, week_day_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
doris_udf::DateTimeVal tv;
|
|
// 2001-02-03 12:34:56
|
|
tv.packed_time = 1830650338932162560L;
|
|
tv.type = TIME_DATETIME;
|
|
|
|
EXPECT_EQ(5, TimestampFunctions::week_day(context, tv).val);
|
|
|
|
// 2020-00-01 00:00:00
|
|
DateTimeValue dtv2(20200001000000);
|
|
dtv2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(true, TimestampFunctions::week_day(context, tv2).is_null);
|
|
|
|
// 2020-01-00 00:00:00
|
|
DateTimeValue dtv3(20200100000000);
|
|
dtv3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(true, TimestampFunctions::week_day(context, tv3).is_null);
|
|
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, day_of_month_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
// 2020-00-01 00:00:00
|
|
DateTimeValue dtv1(20200001000000);
|
|
dtv1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_month(context, tv1).is_null);
|
|
|
|
// 2020-01-00 00:00:00
|
|
DateTimeValue dtv2(20200100000000);
|
|
dtv2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_month(context, tv2).is_null);
|
|
|
|
// 2020-02-29 00:00:00
|
|
DateTimeValue dtv3(20200229000000);
|
|
dtv3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(false, TimestampFunctions::day_of_month(context, tv3).is_null);
|
|
EXPECT_EQ(29, TimestampFunctions::day_of_month(context, tv3).val);
|
|
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, day_of_year_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
// 2020-00-01 00:00:00
|
|
DateTimeValue dtv1(20200001000000);
|
|
dtv1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_year(context, tv1).is_null);
|
|
|
|
// 2020-01-00 00:00:00
|
|
DateTimeValue dtv2(20200100000000);
|
|
dtv2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(true, TimestampFunctions::day_of_year(context, tv2).is_null);
|
|
|
|
// 2020-02-29 00:00:00
|
|
DateTimeValue dtv3(20200229000000);
|
|
dtv3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(false, TimestampFunctions::day_of_year(context, tv3).is_null);
|
|
EXPECT_EQ(60, TimestampFunctions::day_of_year(context, tv3).val);
|
|
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, week_of_year_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
// 2020-00-01 00:00:00
|
|
DateTimeValue dtv1(20200001000000);
|
|
dtv1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(true, TimestampFunctions::week_of_year(context, tv1).is_null);
|
|
|
|
// 2020-01-00 00:00:00
|
|
DateTimeValue dtv2(20200100000000);
|
|
dtv2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(true, TimestampFunctions::week_of_year(context, tv2).is_null);
|
|
|
|
// 2020-02-29 00:00:00
|
|
DateTimeValue dtv3(20200229000000);
|
|
dtv3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(false, TimestampFunctions::week_of_year(context, tv3).is_null);
|
|
EXPECT_EQ(9, TimestampFunctions::week_of_year(context, tv3).val);
|
|
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, year_week_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
DateTimeValue dtv1(20210101000000);
|
|
dtv1.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(202052, TimestampFunctions::year_week(context, tv1).val);
|
|
|
|
DateTimeValue dtv2(20210103000000);
|
|
dtv2.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(202101, TimestampFunctions::year_week(context, tv2).val);
|
|
|
|
DateTimeValue dtv3(20210501000000);
|
|
dtv3.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(202117, TimestampFunctions::year_week(context, tv3).val);
|
|
|
|
DateTimeValue dtv4(20241230000000);
|
|
dtv4.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv4;
|
|
dtv4.to_datetime_val(&tv4);
|
|
EXPECT_EQ(202501, TimestampFunctions::year_week(context, tv4, 1).val);
|
|
|
|
DateTimeValue dtv5(20261229121030);
|
|
dtv5.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv5;
|
|
dtv5.to_datetime_val(&tv5);
|
|
EXPECT_EQ(202653, TimestampFunctions::year_week(context, tv5, 3).val);
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, week_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
DateTimeValue dtv1(20210101000000);
|
|
dtv1.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(0, TimestampFunctions::week(context, tv1).val);
|
|
|
|
DateTimeValue dtv2(20210103000000);
|
|
dtv2.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(1, TimestampFunctions::week(context, tv2).val);
|
|
|
|
DateTimeValue dtv3(20210501000000);
|
|
dtv3.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(17, TimestampFunctions::week(context, tv3).val);
|
|
|
|
DateTimeValue dtv4(20210101000000);
|
|
dtv4.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv4;
|
|
dtv4.to_datetime_val(&tv4);
|
|
EXPECT_EQ(0, TimestampFunctions::week(context, tv4, {1}).val);
|
|
|
|
DateTimeValue dtv5(20211201000000);
|
|
dtv5.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv5;
|
|
dtv5.to_datetime_val(&tv5);
|
|
EXPECT_EQ(48, TimestampFunctions::week(context, tv5, 2).val);
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, make_date_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
|
|
EXPECT_EQ(true, TimestampFunctions::make_date(context, 2021, 0).is_null);
|
|
|
|
DateTimeValue dtv1(20210101000000);
|
|
dtv1.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv1;
|
|
dtv1.to_datetime_val(&tv1);
|
|
EXPECT_EQ(tv1.packed_time, TimestampFunctions::make_date(context, 2021, 1).packed_time);
|
|
|
|
DateTimeValue dtv2(20211027000000);
|
|
dtv2.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv2;
|
|
dtv2.to_datetime_val(&tv2);
|
|
EXPECT_EQ(tv2.packed_time, TimestampFunctions::make_date(context, 2021, 300).packed_time);
|
|
|
|
DateTimeValue dtv3(20220204000000);
|
|
dtv3.set_type(TIME_DATE);
|
|
doris_udf::DateTimeVal tv3;
|
|
dtv3.to_datetime_val(&tv3);
|
|
EXPECT_EQ(tv3.packed_time, TimestampFunctions::make_date(context, 2021, 400).packed_time);
|
|
delete context;
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, time_diff_test) {
|
|
DateTimeValue dt1(20190718120000);
|
|
dt1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dt1.to_datetime_val(&tv1);
|
|
|
|
DateTimeValue dt2(20190718130102);
|
|
dt2.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv2;
|
|
dt2.to_datetime_val(&tv2);
|
|
|
|
EXPECT_EQ(-3662, TimestampFunctions::time_diff(ctx, tv1, tv2).val);
|
|
|
|
// invalid
|
|
DateTimeValue dt3(20190018120000);
|
|
dt3.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv3;
|
|
dt3.to_datetime_val(&tv3);
|
|
|
|
DateTimeValue dt4(20190718130102);
|
|
dt4.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv4;
|
|
dt4.to_datetime_val(&tv4);
|
|
|
|
EXPECT_EQ(true, TimestampFunctions::time_diff(ctx, tv3, tv4).is_null);
|
|
|
|
// invalid
|
|
DateTimeValue dt5(20190718120000);
|
|
dt5.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv5;
|
|
dt5.to_datetime_val(&tv5);
|
|
|
|
DateTimeValue dt6(20190700130102);
|
|
dt6.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv6;
|
|
dt6.to_datetime_val(&tv6);
|
|
|
|
EXPECT_EQ(true, TimestampFunctions::time_diff(ctx, tv5, tv6).is_null);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, now) {
|
|
DateTimeVal now = TimestampFunctions::now(ctx);
|
|
DateTimeValue dt = DateTimeValue::from_datetime_val(now);
|
|
EXPECT_EQ(20190806013857, dt.to_int64());
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, from_unix) {
|
|
IntVal unixtimestamp(1565026737);
|
|
StringVal sval = TimestampFunctions::from_unix(ctx, unixtimestamp);
|
|
EXPECT_EQ("2019-08-06 01:38:57", std::string((char*)sval.ptr, sval.len));
|
|
|
|
IntVal unixtimestamp2(-123);
|
|
sval = TimestampFunctions::from_unix(ctx, unixtimestamp2);
|
|
EXPECT_TRUE(sval.is_null);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, to_unix) {
|
|
DateTimeVal dt_val;
|
|
dt_val.packed_time = 1847544683002068992;
|
|
dt_val.type = TIME_DATETIME;
|
|
EXPECT_EQ(1565026737, TimestampFunctions::to_unix(ctx).val);
|
|
EXPECT_EQ(1565026737, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
EXPECT_EQ(1565026737, TimestampFunctions::to_unix(ctx, StringVal("2019-08-06 01:38:57"),
|
|
"%Y-%m-%d %H:%i:%S")
|
|
.val);
|
|
|
|
DateTimeValue dt_value;
|
|
dt_value.from_date_int64(99991230);
|
|
dt_value.to_datetime_val(&dt_val);
|
|
EXPECT_EQ(0, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
|
|
dt_value.from_date_int64(10000101);
|
|
dt_value.to_datetime_val(&dt_val);
|
|
EXPECT_EQ(0, TimestampFunctions::to_unix(ctx, dt_val).val);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, curtime) {
|
|
EXPECT_EQ(3600 + 38 * 60 + 57, TimestampFunctions::curtime(ctx).val);
|
|
}
|
|
|
|
TEST_F(TimestampFunctionsTest, convert_tz_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
DateTimeValue dt1(20190806163857);
|
|
dt1.set_type(TIME_DATETIME);
|
|
doris_udf::DateTimeVal tv1;
|
|
dt1.to_datetime_val(&tv1);
|
|
DateTimeVal t = TimestampFunctions::convert_tz(context, tv1, StringVal("Asia/Shanghai"),
|
|
StringVal("America/Los_Angeles"));
|
|
DateTimeValue dt2 = DateTimeValue::from_datetime_val(t);
|
|
EXPECT_EQ(20190806013857, dt2.to_int64());
|
|
|
|
t = TimestampFunctions::convert_tz(context, tv1, StringVal("CST"),
|
|
StringVal("America/Los_Angeles"));
|
|
DateTimeValue dt3 = DateTimeValue::from_datetime_val(t);
|
|
EXPECT_EQ(20190806013857, dt3.to_int64());
|
|
delete context;
|
|
}
|
|
|
|
#define EXPECT_DIFF(unit, diff, tv1, tv2) \
|
|
EXPECT_EQ( \
|
|
diff, \
|
|
TimestampFunctions::unit##s_diff(context, datetime_val(tv1), datetime_val(tv2)).val); \
|
|
EXPECT_EQ( \
|
|
-(diff), \
|
|
TimestampFunctions::unit##s_diff(context, datetime_val(tv2), datetime_val(tv1)).val);
|
|
TEST_F(TimestampFunctionsTest, timestampdiff_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
int64_t tv1 = 20120824000001;
|
|
int64_t tv2 = 20120830000000;
|
|
|
|
//YEAR
|
|
EXPECT_DIFF(year, 0, tv2, tv1);
|
|
EXPECT_DIFF(year, 1, tv1, 20100930000000);
|
|
//MONTH
|
|
EXPECT_DIFF(month, 0, tv2, tv1);
|
|
EXPECT_DIFF(month, 0, tv2, tv1);
|
|
EXPECT_DIFF(month, 0, 20120924000000, tv1);
|
|
EXPECT_DIFF(month, 1, tv1, 20120630000000);
|
|
//WEEK
|
|
EXPECT_DIFF(week, 0, tv2, tv1);
|
|
//DAY
|
|
EXPECT_DIFF(day, 5, tv2, tv1);
|
|
EXPECT_DIFF(day, 6, 20120830000001, tv1);
|
|
EXPECT_DIFF(day, 8, 20120901000001, tv1);
|
|
EXPECT_DIFF(day, 0, 20120823000005, tv1);
|
|
EXPECT_DIFF(day, 0, 20120824000001, tv1);
|
|
|
|
//HOUR
|
|
EXPECT_DIFF(hour, 143, tv2, tv1);
|
|
//MINUTE
|
|
EXPECT_DIFF(minute, 8639, tv2, tv1);
|
|
//SECOND
|
|
EXPECT_DIFF(second, 518399, tv2, tv1);
|
|
|
|
delete context;
|
|
}
|
|
|
|
#define EXPECT_ROUND(unit, floor, ceil, ...) \
|
|
EXPECT_EQ(datetime_val(floor).packed_time, \
|
|
TimestampFunctions::unit##_floor(context, __VA_ARGS__).packed_time); \
|
|
EXPECT_EQ(datetime_val(ceil).packed_time, \
|
|
TimestampFunctions::unit##_ceil(context, __VA_ARGS__).packed_time);
|
|
TEST_F(TimestampFunctionsTest, time_round_test) {
|
|
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
|
|
doris_udf::DateTimeVal tv = datetime_val(20120824132901);
|
|
|
|
doris_udf::IntVal three(3);
|
|
doris_udf::DateTimeVal wednesday = datetime_val(20200916000000);
|
|
//YEAR
|
|
EXPECT_ROUND(year, 20120101000000, 20130101000000, tv);
|
|
EXPECT_ROUND(year, 20110916000000, 20120916000000, tv, wednesday);
|
|
EXPECT_ROUND(year, 20110916000000, 20140916000000, tv, three, wednesday);
|
|
|
|
//MONTH
|
|
EXPECT_ROUND(month, 20120801000000, 20120901000000, tv);
|
|
EXPECT_ROUND(month, 20120701000000, 20121001000000, tv, three);
|
|
|
|
//WEEK
|
|
EXPECT_ROUND(week, 20120819000000, 20120826000000, tv);
|
|
EXPECT_ROUND(week, 20120822000000, 20120829000000, tv, wednesday);
|
|
EXPECT_ROUND(week, 20120808000000, 20120829000000, tv, three, wednesday);
|
|
|
|
doris_udf::DateTimeVal tv1 = datetime_val(20200202130920);
|
|
doris_udf::DateTimeVal monday = datetime_val(20200106000000);
|
|
EXPECT_ROUND(week, 20200202000000, 20200209000000, tv1);
|
|
EXPECT_ROUND(week, 20200127000000, 20200203000000, tv1, monday);
|
|
|
|
//DAY
|
|
doris_udf::DateTimeVal noon = datetime_val(19700101120000);
|
|
EXPECT_ROUND(day, 20120824000000, 20120825000000, tv);
|
|
EXPECT_ROUND(day, 20120824120000, 20120825120000, tv, noon);
|
|
|
|
//HOUR
|
|
doris_udf::DateTimeVal random = datetime_val(29380329113953);
|
|
EXPECT_ROUND(hour, 20120824120000, 20120824150000, tv, three);
|
|
EXPECT_ROUND(hour, 20120824113953, 20120824143953, tv, three, random);
|
|
|
|
//MINUTE
|
|
doris_udf::IntVal val90(90);
|
|
EXPECT_ROUND(minute, 20120824132900, 20120824133000, tv);
|
|
EXPECT_ROUND(minute, 20120824120000, 20120824133000, tv, val90);
|
|
EXPECT_ROUND(minute, 20120824130953, 20120824143953, tv, val90, random);
|
|
|
|
//SECOND
|
|
EXPECT_ROUND(second, 20120824132900, 20120824132903, tv, three);
|
|
EXPECT_ROUND(second, 20120824132830, 20120824133000, tv, val90);
|
|
|
|
delete context;
|
|
}
|
|
|
|
} // namespace doris
|