database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](year) fix year() results are not as expected (#13426)

Browse Source 
fix `year()` results are not as expected
This commit is contained in:
luozenglin
2022-10-19 11:28:00 +08:00
committed by GitHub
parent 8a068c8c92
commit c449028a5f
4 changed files with 43 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
be/test/vec/function/function_test_util.cpp
View File

@ -172,12 +172,37 @@ bool parse_ut_data_type(const std::vector<std::any>& input_types, ut_type::UTDat
}
return true;
}
template <typename Date, TypeIndex type_index = TypeIndex::Nothing>
bool insert_date_cell(MutableColumnPtr& column, const std::string& format, const std::any& cell) {
auto datetime_str = std::any_cast<std::string>(cell);
Date v;
auto result = v.from_date_format_str(format.c_str(), format.size(), datetime_str.c_str(),
datetime_str.size());
if constexpr (type_index == TypeIndex::Date) {
v.cast_to_date();
} else if constexpr (type_index == TypeIndex::DateTime) {
v.to_datetime();
}
if (result) {
column->insert_data(reinterpret_cast<char*>(&v), 0);
} else if (column->is_nullable()) {
column->insert_data(nullptr, 0);
} else {
return false;
}
return true;
}
bool insert_cell(MutableColumnPtr& column, DataTypePtr type_ptr, const std::any& cell) {
if (cell.type() == typeid(Null)) {
column->insert_data(nullptr, 0);
return true;
}
#define RETURN_IF_FALSE(x) \
if (UNLIKELY(!(x))) return false
WhichDataType type(type_ptr);
if (type.is_string()) {
auto str = std::any_cast<ut_type::STRING>(cell);
@ -215,34 +240,20 @@ bool insert_cell(MutableColumnPtr& column, DataTypePtr type_ptr, const std::any&
column->insert_data(reinterpret_cast<char*>(&value), 0);
} else if (type.is_date_time()) {
static std::string date_time_format("%Y-%m-%d %H:%i:%s");
auto datetime_str = std::any_cast<std::string>(cell);
VecDateTimeValue v;
v.from_date_format_str(date_time_format.c_str(), date_time_format.size(),
datetime_str.c_str(), datetime_str.size());
v.to_datetime();
column->insert_data(reinterpret_cast<char*>(&v), 0);
RETURN_IF_FALSE((insert_date_cell<VecDateTimeValue, TypeIndex::DateTime>(
column, date_time_format, cell)));
} else if (type.is_date()) {
static std::string date_time_format("%Y-%m-%d");
auto datetime_str = std::any_cast<std::string>(cell);
VecDateTimeValue v;
v.from_date_format_str(date_time_format.c_str(), date_time_format.size(),
datetime_str.c_str(), datetime_str.size());
v.cast_to_date();
column->insert_data(reinterpret_cast<char*>(&v), 0);
RETURN_IF_FALSE((insert_date_cell<VecDateTimeValue, TypeIndex::Date>(
column, date_time_format, cell)));
} else if (type.is_date_v2()) {
static std::string date_time_format("%Y-%m-%d");
auto datetime_str = std::any_cast<std::string>(cell);
DateV2Value<DateV2ValueType> v;
v.from_date_format_str(date_time_format.c_str(), date_time_format.size(),
datetime_str.c_str(), datetime_str.size());
column->insert_data(reinterpret_cast<char*>(&v), 0);
RETURN_IF_FALSE(
(insert_date_cell<DateV2Value<DateV2ValueType>>(column, date_time_format, cell)));
} else if (type.is_date_time_v2()) {
static std::string date_time_format("%Y-%m-%d %H:%i:%s.%f");
auto datetime_str = std::any_cast<std::string>(cell);
DateV2Value<DateTimeV2ValueType> v;
v.from_date_format_str(date_time_format.c_str(), date_time_format.size(),
datetime_str.c_str(), datetime_str.size());
column->insert_data(reinterpret_cast<char*>(&v), 0);
RETURN_IF_FALSE((insert_date_cell<DateV2Value<DateTimeV2ValueType>>(
column, date_time_format, cell)));
} else if (type.is_array()) {
auto v = std::any_cast<Array>(cell);
column->insert(v);