ClickHouse/src/Functions/parseDateTime.cpp

1900 lines
78 KiB
C++
Raw Normal View History

#include <Columns/ColumnNullable.h>
#include <Columns/ColumnsNumber.h>
#include <Columns/ColumnString.h>
#include <Columns/ColumnsDateTime.h>
2023-02-22 10:05:53 +00:00
#include <DataTypes/DataTypeDateTime.h>
#include <DataTypes/DataTypeString.h>
2023-02-22 10:05:53 +00:00
#include <Functions/FunctionFactory.h>
#include <Functions/FunctionHelpers.h>
#include <Functions/FunctionsConversion.h>
#include <Functions/IFunction.h>
#include <Functions/castTypeToEither.h>
2023-03-05 08:24:05 +00:00
#include <Functions/numLiteralChars.h>
2023-02-22 10:05:53 +00:00
#include <IO/WriteHelpers.h>
2023-02-24 09:07:27 +00:00
#include <base/types.h>
2023-03-05 08:24:05 +00:00
#include <boost/algorithm/string/case_conv.hpp>
2023-02-22 10:05:53 +00:00
namespace DB
{
2023-03-01 12:24:50 +00:00
namespace ErrorCodes
{
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
extern const int ILLEGAL_COLUMN;
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
extern const int NOT_IMPLEMENTED;
extern const int BAD_ARGUMENTS;
2023-03-05 08:24:05 +00:00
extern const int VALUE_IS_OUT_OF_RANGE_OF_DATA_TYPE;
2023-03-07 06:35:50 +00:00
extern const int CANNOT_PARSE_DATETIME;
2023-03-06 09:08:55 +00:00
extern const int NOT_ENOUGH_SPACE;
2023-03-01 12:24:50 +00:00
}
2023-02-22 10:05:53 +00:00
namespace
{
2023-02-27 13:41:38 +00:00
using Pos = const char *;
2023-03-01 03:02:41 +00:00
constexpr Int32 minYear = 1970;
constexpr Int32 maxYear = 2106;
2023-02-27 13:41:38 +00:00
const std::unordered_map<String, std::pair<String, Int32>> dayOfWeekMap{
{"mon", {"day", 1}},
{"tue", {"sday", 2}},
{"wed", {"nesday", 3}},
{"thu", {"rsday", 4}},
{"fri", {"day", 5}},
{"sat", {"urday", 6}},
{"sun", {"day", 7}},
};
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
const std::unordered_map<String, std::pair<String, Int32>> monthMap{
{"jan", {"uary", 1}},
{"feb", {"ruary", 2}},
{"mar", {"ch", 3}},
2023-02-27 13:41:38 +00:00
{"apr", {"il", 4}},
{"may", {"", 5}},
{"jun", {"e", 6}},
{"jul", {"y", 7}},
{"aug", {"ust", 8}},
{"sep", {"tember", 9}},
{"oct", {"ober", 10}},
{"nov", {"ember", 11}},
{"dec", {"ember", 12}},
};
2023-02-22 10:05:53 +00:00
2023-03-01 03:02:41 +00:00
/// key: month, value: total days of current month if current year is leap year.
2023-02-27 13:41:38 +00:00
constexpr Int32 leapDays[] = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
2023-03-01 03:02:41 +00:00
/// key: month, value: total days of current month if current year is not leap year.
2023-02-27 13:41:38 +00:00
constexpr Int32 normalDays[] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
2023-03-01 12:24:50 +00:00
/// key: month, value: cumulative days from January to current month(inclusive) if current year is leap year.
2023-02-27 13:41:38 +00:00
constexpr Int32 cumulativeLeapDays[] = {0, 31, 60, 91, 121, 152, 182, 213, 244, 274, 305, 335, 366};
2023-03-01 03:02:41 +00:00
2023-03-01 12:24:50 +00:00
/// key: month, value: cumulative days from January to current month(inclusive) if current year is not leap year.
2023-02-27 13:41:38 +00:00
constexpr Int32 cumulativeDays[] = {0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365};
2023-03-01 03:02:41 +00:00
/// key: year, value: cumulative days from epoch(1970-01-01) to the first day of current year(exclusive).
2023-02-27 13:41:38 +00:00
constexpr Int32 cumulativeYearDays[]
= {0, 365, 730, 1096, 1461, 1826, 2191, 2557, 2922, 3287, 3652, 4018, 4383, 4748, 5113, 5479, 5844, 6209,
6574, 6940, 7305, 7670, 8035, 8401, 8766, 9131, 9496, 9862, 10227, 10592, 10957, 11323, 11688, 12053, 12418, 12784,
13149, 13514, 13879, 14245, 14610, 14975, 15340, 15706, 16071, 16436, 16801, 17167, 17532, 17897, 18262, 18628, 18993, 19358,
19723, 20089, 20454, 20819, 21184, 21550, 21915, 22280, 22645, 23011, 23376, 23741, 24106, 24472, 24837, 25202, 25567, 25933,
26298, 26663, 27028, 27394, 27759, 28124, 28489, 28855, 29220, 29585, 29950, 30316, 30681, 31046, 31411, 31777, 32142, 32507,
32872, 33238, 33603, 33968, 34333, 34699, 35064, 35429, 35794, 36160, 36525, 36890, 37255, 37621, 37986, 38351, 38716, 39082,
39447, 39812, 40177, 40543, 40908, 41273, 41638, 42004, 42369, 42734, 43099, 43465, 43830, 44195, 44560, 44926, 45291, 45656,
46021, 46387, 46752, 47117, 47482, 47847, 48212, 48577, 48942, 49308, 49673};
2023-03-03 02:42:35 +00:00
struct DateTime
2023-02-27 13:41:38 +00:00
{
2023-03-07 03:45:32 +00:00
/// If both week_date_format and week_date_format is false, date is composed of year, month and day
Int32 year = 1970; /// year, range [1970, 2106]
Int32 month = 1; /// month of year, range [1, 12]
Int32 day = 1; /// day of month, range [1, 31]
2023-02-22 10:05:53 +00:00
2023-03-07 03:45:32 +00:00
Int32 week = 1; /// ISO week of year, range [1, 53]
Int32 day_of_week = 1; /// day of week, range [1, 7], 1 represents Monday, 2 represents Tuesday...
bool week_date_format
= false; /// If true, date is composed of week year(reuse year), week of year(use week) and day of week(use day_of_week)
2023-02-22 10:05:53 +00:00
2023-03-07 03:45:32 +00:00
Int32 day_of_year = 1; /// day of year, range [1, 366]
bool day_of_year_format = false; /// If true, date is composed of year(reuse year), day of year(use day_of_year)
2023-02-22 10:05:53 +00:00
2023-03-07 03:45:32 +00:00
bool is_year_of_era = false; /// If true, year is calculated from era and year of era, the latter cannot be zero or negative.
bool has_year = false; /// Whether year was explicitly specified.
2023-02-22 10:05:53 +00:00
2023-03-23 21:41:01 +00:00
/// If hour_starts_at_1 = true, is_hour_of_half_day = true, hour's range is [1, 12]
/// If hour_starts_at_1 = true, is_hour_of_half_day = false, hour's range is [1, 24]
/// If hour_starts_at_1 = false, is_hour_of_half_day = true, hour's range is [0, 11]
/// If hour_starts_at_1 = false, is_hour_of_half_day = false, hour's range is [0, 23]
2023-02-27 13:41:38 +00:00
Int32 hour = 0;
2023-03-07 03:45:32 +00:00
Int32 minute = 0; /// range [0, 59]
Int32 second = 0; /// range [0, 59]
2023-02-22 10:05:53 +00:00
2023-03-20 15:20:57 +00:00
bool is_am = true; /// If is_hour_of_half_day = true and is_am = false (i.e. pm) then add 12 hours to the result DateTime
2023-03-23 21:41:01 +00:00
bool hour_starts_at_1 = false; /// Whether the hour is clockhour
2023-03-07 03:45:32 +00:00
bool is_hour_of_half_day = false; /// Whether the hour is of half day
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
bool has_time_zone_offset = false; /// If true, time zone offset is explicitly specified.
2023-03-07 06:35:50 +00:00
Int64 time_zone_offset = 0; /// Offset in seconds between current timezone to UTC.
2023-03-01 09:15:29 +00:00
void reset()
{
year = 1970;
month = 1;
day = 1;
week = 1;
day_of_week = 1;
week_date_format = false;
day_of_year = 1;
day_of_year_format = false;
2023-03-07 03:45:32 +00:00
is_year_of_era = false;
has_year = false;
2023-03-01 09:15:29 +00:00
hour = 0;
minute = 0;
second = 0;
2023-03-03 02:42:35 +00:00
2023-03-07 03:45:32 +00:00
is_am = true;
2023-03-23 21:41:01 +00:00
hour_starts_at_1 = false;
2023-03-07 03:45:32 +00:00
is_hour_of_half_day = false;
2023-03-01 09:15:29 +00:00
2023-03-07 03:45:32 +00:00
has_time_zone_offset = false;
time_zone_offset = 0;
2023-03-01 09:15:29 +00:00
}
2023-02-22 10:05:53 +00:00
2023-03-20 15:58:37 +00:00
/// Input text is expected to be lowered by caller
void setEra(const String & text) // NOLINT
2023-03-20 15:20:57 +00:00
{
if (text == "bc")
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Era BC exceeds the range of DateTime");
else if (text != "ad")
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Unknown era {}", text);
}
2023-03-03 02:42:35 +00:00
void setCentury(Int32 century)
2023-02-28 15:02:02 +00:00
{
if (century < 19 || century > 21)
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for century must be in the range [19, 21]", century);
2023-02-28 15:02:02 +00:00
year = 100 * century;
has_year = true;
}
2023-03-20 15:20:57 +00:00
void setYear(Int32 year_, bool is_year_of_era_ = false, bool is_week_year = false)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
if (year_ < minYear || year_ > maxYear)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for year must be in the range [{}, {}]", year_, minYear, maxYear);
2023-02-28 15:02:02 +00:00
2023-03-20 15:20:57 +00:00
year = year_;
has_year = true;
is_year_of_era = is_year_of_era_;
if (is_week_year)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
week_date_format = true;
day_of_year_format = false;
2023-02-28 15:02:02 +00:00
}
}
2023-03-20 15:58:37 +00:00
void setYear2(Int32 year_)
2023-03-20 15:20:57 +00:00
{
if (year_ >= 70 && year_ < 100)
year_ += 1900;
else if (year_ >= 0 && year_ < 70)
year_ += 2000;
else
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for year2 must be in the range [0, 99]", year_);
2023-03-20 15:58:37 +00:00
setYear(year_, false, false);
2023-03-20 15:20:57 +00:00
}
2023-02-28 15:02:02 +00:00
void setMonth(Int32 month_)
{
if (month_ < 1 || month_ > 12)
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for month of year must be in the range [1, 12]", month_);
2023-02-28 15:02:02 +00:00
month = month_;
week_date_format = false;
day_of_year_format = false;
if (!has_year)
{
has_year = true;
year = 2000;
}
}
2023-03-20 15:20:57 +00:00
void setWeek(Int32 week_)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
if (week_ < 1 || week_ > 53)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for week of week year must be in the range [1, 53]", week_);
2023-02-28 15:02:02 +00:00
2023-03-20 15:20:57 +00:00
week = week_;
week_date_format = true;
2023-02-28 15:02:02 +00:00
day_of_year_format = false;
if (!has_year)
{
has_year = true;
year = 2000;
}
}
2023-03-20 15:20:57 +00:00
void setDayOfYear(Int32 day_of_year_)
2023-02-28 15:02:02 +00:00
{
if (day_of_year_ < 1 || day_of_year_ > 366)
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for day of year must be in the range [1, 366]", day_of_year_);
2023-02-28 15:02:02 +00:00
day_of_year = day_of_year_;
day_of_year_format = true;
week_date_format = false;
if (!has_year)
{
has_year = true;
year = 2000;
}
}
2023-03-20 15:20:57 +00:00
void setDayOfMonth(Int32 day_of_month)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
if (day_of_month < 1 || day_of_month > 31)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for day of month must be in the range [1, 31]", day_of_month);
2023-02-28 15:02:02 +00:00
2023-03-20 15:20:57 +00:00
day = day_of_month;
week_date_format = false;
day_of_year_format = false;
if (!has_year)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
has_year = true;
year = 2000;
2023-02-28 15:02:02 +00:00
}
}
2023-03-20 15:20:57 +00:00
void setDayOfWeek(Int32 day_of_week_)
2023-02-28 15:02:02 +00:00
{
2023-03-20 15:20:57 +00:00
if (day_of_week_ < 1 || day_of_week_ > 7)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for day of week must be in the range [1, 7]", day_of_week_);
2023-02-28 15:02:02 +00:00
2023-03-20 15:20:57 +00:00
day_of_week = day_of_week_;
2023-02-28 15:02:02 +00:00
week_date_format = true;
day_of_year_format = false;
if (!has_year)
{
has_year = true;
year = 2000;
}
}
2023-03-20 15:58:37 +00:00
/// Input text is expected to be lowered by caller
void setAMPM(const String & text)
2023-02-28 15:02:02 +00:00
{
if (text == "am")
is_am = true;
else if (text == "pm")
is_am = false;
else
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Unknown half day of day: {}", text);
2023-02-28 15:02:02 +00:00
}
2023-03-23 21:41:01 +00:00
void setHour(Int32 hour_, bool is_hour_of_half_day_ = false, bool hour_starts_at_1_ = false)
2023-02-28 15:02:02 +00:00
{
Int32 max_hour;
Int32 min_hour;
2023-03-01 03:02:41 +00:00
Int32 new_hour = hour_;
2023-03-23 21:41:01 +00:00
if (!is_hour_of_half_day_ && !hour_starts_at_1_)
2023-02-28 15:02:02 +00:00
{
max_hour = 23;
min_hour = 0;
}
2023-03-23 21:41:01 +00:00
else if (!is_hour_of_half_day_ && hour_starts_at_1_)
2023-02-28 15:02:02 +00:00
{
max_hour = 24;
min_hour = 1;
new_hour = hour_ % 24;
}
2023-03-23 21:41:01 +00:00
else if (is_hour_of_half_day_ && !hour_starts_at_1_)
2023-02-28 15:02:02 +00:00
{
max_hour = 11;
min_hour = 0;
}
else
{
max_hour = 12;
min_hour = 1;
new_hour = hour_ % 12;
}
if (hour_ < min_hour || hour_ > max_hour)
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-23 21:41:01 +00:00
"Value {} for hour must be in the range [{}, {}] if_hour_of_half_day={} and hour_starts_at_1={}",
2023-02-28 15:02:02 +00:00
hour,
max_hour,
min_hour,
is_hour_of_half_day_,
2023-03-23 21:41:01 +00:00
hour_starts_at_1_);
2023-02-28 15:02:02 +00:00
hour = new_hour;
is_hour_of_half_day = is_hour_of_half_day_;
2023-03-23 21:41:01 +00:00
hour_starts_at_1 = hour_starts_at_1_;
2023-02-28 15:02:02 +00:00
}
2023-03-20 15:20:57 +00:00
void setMinute(Int32 minute_)
{
if (minute_ < 0 || minute_ > 59)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for minute must be in the range [0, 59]", minute_);
minute = minute_;
}
void setSecond(Int32 second_)
{
if (second_ < 0 || second_ > 59)
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Value {} for second must be in the range [0, 59]", second_);
second = second_;
}
2023-02-27 13:41:38 +00:00
/// For debug
[[maybe_unused]] String toString() const
{
String res;
res += "year:" + std::to_string(year);
res += ",";
res += "month:" + std::to_string(month);
res += ",";
res += "day:" + std::to_string(day);
res += ",";
res += "hour:" + std::to_string(hour);
res += ",";
res += "minute:" + std::to_string(minute);
res += ",";
res += "second:" + std::to_string(second);
2023-02-28 06:52:13 +00:00
res += ",";
res += "AM:" + std::to_string(is_am);
2023-02-27 13:41:38 +00:00
return res;
}
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
static bool isLeapYear(Int32 year_) { return year_ % 4 == 0 && (year_ % 100 != 0 || year_ % 400 == 0); }
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
static bool isDateValid(Int32 year_, Int32 month_, Int32 day_)
{
2023-03-01 09:15:29 +00:00
/// The range of month[1, 12] and day[1, 31] already checked before
2023-02-27 13:41:38 +00:00
bool leap = isLeapYear(year_);
2023-03-01 09:15:29 +00:00
return (year_ >= minYear && year_ <= maxYear) && ((leap && day_ <= leapDays[month_]) || (!leap && day_ <= normalDays[month_]));
2023-02-27 13:41:38 +00:00
}
2023-02-27 13:41:38 +00:00
static bool isDayOfYearValid(Int32 year_, Int32 day_of_year_)
{
2023-03-01 09:15:29 +00:00
/// The range of day_of_year[1, 366] already checked before
bool leap = isLeapYear(year_);
return (year_ >= minYear && year_ <= maxYear) && (day_of_year_ <= 365 + (leap ? 1 : 0));
2023-02-27 13:41:38 +00:00
}
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
static Int32 extractISODayOfTheWeek(Int32 days_since_epoch)
{
if (days_since_epoch < 0)
{
// negative date: start off at 4 and cycle downwards
2023-03-01 09:15:29 +00:00
return (7 - ((-days_since_epoch + 3) % 7));
2023-02-27 13:41:38 +00:00
}
else
{
// positive date: start off at 4 and cycle upwards
2023-03-01 09:15:29 +00:00
return ((days_since_epoch + 3) % 7) + 1;
2023-02-27 13:41:38 +00:00
}
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Int32 daysSinceEpochFromWeekDate(int32_t week_year_, int32_t week_of_year_, int32_t day_of_week_)
2023-02-27 13:41:38 +00:00
{
2023-03-01 09:15:29 +00:00
/// The range of week_of_year[1, 53], day_of_week[1, 7] already checked before
if (week_year_ < minYear || week_year_ > maxYear)
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Invalid week year {}", week_year_);
2023-02-27 13:41:38 +00:00
Int32 days_since_epoch_of_jan_fourth = daysSinceEpochFromDate(week_year_, 1, 4);
Int32 first_day_of_week_year = extractISODayOfTheWeek(days_since_epoch_of_jan_fourth);
return days_since_epoch_of_jan_fourth - (first_day_of_week_year - 1) + 7 * (week_of_year_ - 1) + day_of_week_ - 1;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Int32 daysSinceEpochFromDayOfYear(Int32 year_, Int32 day_of_year_)
{
2023-02-27 13:41:38 +00:00
if (!isDayOfYearValid(year_, day_of_year_))
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Invalid day of year, year:{} day of year:{}", year_, day_of_year_);
2023-02-27 13:41:38 +00:00
Int32 res = daysSinceEpochFromDate(year_, 1, 1);
res += day_of_year_ - 1;
return res;
}
2023-02-27 13:41:38 +00:00
2023-03-20 16:24:06 +00:00
static Int32 daysSinceEpochFromDate(Int32 year_, Int32 month_, Int32 day_)
{
2023-02-27 13:41:38 +00:00
if (!isDateValid(year_, month_, day_))
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::CANNOT_PARSE_DATETIME, "Invalid date, year:{} month:{} day:{}", year_, month_, day_);
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
Int32 res = cumulativeYearDays[year_ - 1970];
res += isLeapYear(year_) ? cumulativeLeapDays[month_ - 1] : cumulativeDays[month_ - 1];
res += day_ - 1;
return res;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
Int64 buildDateTime(const DateLUTImpl & time_zone)
2023-02-27 13:41:38 +00:00
{
if (is_hour_of_half_day && !is_am)
hour += 12;
2023-02-27 13:41:38 +00:00
// Convert the parsed date/time into a timestamp.
Int32 days_since_epoch;
if (week_date_format)
days_since_epoch = daysSinceEpochFromWeekDate(year, week, day_of_week);
else if (day_of_year_format)
days_since_epoch = daysSinceEpochFromDayOfYear(year, day_of_year);
else
days_since_epoch = daysSinceEpochFromDate(year, month, day);
2023-02-22 10:05:53 +00:00
2023-03-22 08:53:47 +00:00
Int64 seconds_since_epoch = days_since_epoch * 86400UZ + hour * 3600UZ + minute * 60UZ + second;
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
/// Time zone is not specified, use local time zone
2023-03-07 06:35:50 +00:00
if (!has_time_zone_offset)
time_zone_offset = time_zone.timezoneOffset(seconds_since_epoch);
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
/// Time zone is specified in format string.
2023-03-07 03:45:32 +00:00
if (seconds_since_epoch >= time_zone_offset)
seconds_since_epoch -= time_zone_offset;
2023-02-28 06:08:03 +00:00
else
2023-03-07 06:35:50 +00:00
throw Exception(ErrorCodes::VALUE_IS_OUT_OF_RANGE_OF_DATA_TYPE, "Seconds since epoch is negative");
2023-02-28 06:08:03 +00:00
2023-02-27 13:41:38 +00:00
return seconds_since_epoch;
}
2023-02-27 13:41:38 +00:00
};
2023-03-20 15:58:37 +00:00
enum class ParseSyntax
2023-02-27 13:41:38 +00:00
{
2023-03-20 15:58:37 +00:00
MySQL,
Joda
2023-02-27 13:41:38 +00:00
};
2023-02-22 10:05:53 +00:00
enum class ErrorHandling
{
Exception,
Zero,
Null
};
2023-02-27 13:41:38 +00:00
/// _FUNC_(str[, format, timezone])
template <typename Name, ParseSyntax parse_syntax, ErrorHandling error_handling>
2023-02-27 13:41:38 +00:00
class FunctionParseDateTimeImpl : public IFunction
{
public:
static constexpr auto name = Name::name;
static FunctionPtr create(ContextPtr) { return std::make_shared<FunctionParseDateTimeImpl>(); }
2023-02-27 13:41:38 +00:00
String getName() const override { return name; }
2023-02-27 13:41:38 +00:00
bool useDefaultImplementationForConstants() const override { return true; }
bool isSuitableForShortCircuitArgumentsExecution(const DataTypesWithConstInfo & /*arguments*/) const override { return false; }
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
ColumnNumbers getArgumentsThatAreAlwaysConstant() const override { return {1, 2}; }
bool isVariadic() const override { return true; }
size_t getNumberOfArguments() const override { return 0; }
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
DataTypePtr getReturnTypeImpl(const ColumnsWithTypeAndName & arguments) const override
{
2023-02-27 13:41:38 +00:00
if (arguments.size() != 1 && arguments.size() != 2 && arguments.size() != 3)
2023-02-24 09:07:27 +00:00
throw Exception(
2023-02-27 13:41:38 +00:00
ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH,
"Number of arguments for function {} doesn't match: passed {}, should be 1, 2 or 3",
getName(),
arguments.size());
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
if (!isString(arguments[0].type))
throw Exception(
ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
2023-03-20 15:58:37 +00:00
"Illegal type {} of first argument of function {}. Should be String",
2023-02-27 13:41:38 +00:00
arguments[0].type->getName(),
getName());
2023-02-24 09:07:27 +00:00
2023-02-27 13:41:38 +00:00
if (arguments.size() > 1 && !isString(arguments[1].type))
throw Exception(
ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
2023-03-20 15:58:37 +00:00
"Illegal type {} of second argument of function {}. Should be String",
2023-02-27 13:41:38 +00:00
arguments[0].type->getName(),
getName());
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
if (arguments.size() > 2 && !isString(arguments[2].type))
throw Exception(
ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
2023-03-20 15:58:37 +00:00
"Illegal type {} of third argument of function {}. Should be String",
2023-02-27 13:41:38 +00:00
arguments[0].type->getName(),
getName());
2023-03-06 09:08:55 +00:00
String time_zone_name = getTimeZone(arguments).getTimeZone();
DataTypePtr date_type = std::make_shared<DataTypeDateTime>(time_zone_name);
if (error_handling == ErrorHandling::Null)
return std::make_shared<DataTypeNullable>(date_type);
else
return date_type;
2023-02-27 13:41:38 +00:00
}
2023-02-22 10:05:53 +00:00
ColumnPtr executeImpl(const ColumnsWithTypeAndName & arguments, const DataTypePtr & /*result_type*/, size_t input_rows_count) const override
2023-02-27 13:41:38 +00:00
{
const auto * col_str = checkAndGetColumn<ColumnString>(arguments[0].column.get());
2023-03-22 08:45:46 +00:00
if (!col_str)
throw Exception(
ErrorCodes::ILLEGAL_COLUMN,
"Illegal column {} of first ('str') argument of function {}. Must be string.",
arguments[0].column->getName(),
getName());
2023-02-27 13:41:38 +00:00
String format = getFormat(arguments);
2023-03-06 09:08:55 +00:00
const auto & time_zone = getTimeZone(arguments);
std::vector<Instruction> instructions = parseFormat(format);
2023-02-27 13:41:38 +00:00
auto col_res = ColumnDateTime::create(input_rows_count);
ColumnUInt8::MutablePtr col_null_map;
if constexpr (error_handling == ErrorHandling::Null)
col_null_map = ColumnUInt8::create(input_rows_count, 0);
2023-03-20 16:24:06 +00:00
auto & res_data = col_res->getData();
2023-03-07 07:50:11 +00:00
/// Make datetime fit in a cache line.
alignas(64) DateTime datetime;
2023-02-27 13:41:38 +00:00
for (size_t i = 0; i < input_rows_count; ++i)
{
2023-03-07 07:50:11 +00:00
datetime.reset();
2023-02-27 13:41:38 +00:00
StringRef str_ref = col_str->getDataAt(i);
Pos cur = str_ref.data;
Pos end = str_ref.data + str_ref.size;
bool error = false;
2023-02-27 13:41:38 +00:00
for (const auto & instruction : instructions)
2023-03-02 08:58:34 +00:00
{
try
{
cur = instruction.perform(cur, end, datetime);
}
catch (...)
{
if constexpr (error_handling == ErrorHandling::Zero)
{
res_data[i] = 0;
error = true;
break;
}
else if constexpr (error_handling == ErrorHandling::Null)
{
res_data[i] = 0;
col_null_map->getData()[i] = 1;
error = true;
break;
}
else
{
static_assert(error_handling == ErrorHandling::Exception);
throw;
}
}
2023-03-02 08:58:34 +00:00
}
if (error)
continue;
try
{
/// Ensure all input was consumed
if (cur < end)
throw Exception(
ErrorCodes::CANNOT_PARSE_DATETIME,
"Invalid format input {} is malformed at {}",
str_ref.toView(),
std::string_view(cur, end - cur));
Int64 time = datetime.buildDateTime(time_zone);
res_data[i] = static_cast<UInt32>(time);
}
catch (...)
{
if constexpr (error_handling == ErrorHandling::Zero)
res_data[i] = 0;
else if constexpr (error_handling == ErrorHandling::Null)
{
res_data[i] = 0;
col_null_map->getData()[i] = 1;
}
else
{
static_assert(error_handling == ErrorHandling::Exception);
throw;
}
}
2023-02-27 13:41:38 +00:00
}
2023-02-24 09:07:27 +00:00
if constexpr (error_handling == ErrorHandling::Null)
return ColumnNullable::create(std::move(col_res), std::move(col_null_map));
else
return col_res;
}
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
private:
2023-03-05 08:24:05 +00:00
class Instruction
2023-02-22 10:05:53 +00:00
{
2023-02-27 13:41:38 +00:00
private:
2023-03-03 02:42:35 +00:00
enum class NeedCheckSpace
{
Yes,
No
};
2023-02-27 13:41:38 +00:00
using Func = std::conditional_t<
2023-03-20 15:58:37 +00:00
parse_syntax == ParseSyntax::MySQL,
2023-03-06 09:08:55 +00:00
Pos (*)(Pos, Pos, const String &, DateTime &),
std::function<Pos(Pos, Pos, const String &, DateTime &)>>;
2023-03-20 16:24:06 +00:00
const Func func{};
const String func_name;
const String literal; /// Only used when current instruction parses literal
const String fragment; /// Parsed fragments in MySQL or Joda format string
2023-02-22 12:42:59 +00:00
2023-02-27 13:41:38 +00:00
public:
2023-03-20 16:24:06 +00:00
explicit Instruction(Func && func_, const char * func_name_, const std::string_view & fragment_)
: func(std::move(func_)), func_name(func_name_), fragment(fragment_)
2023-03-06 09:08:55 +00:00
{
}
2023-02-22 12:42:59 +00:00
2023-03-20 16:24:06 +00:00
explicit Instruction(const String & literal_) : literal(literal_), fragment("LITERAL") { }
explicit Instruction(String && literal_) : literal(std::move(literal_)), fragment("LITERAL") { }
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
/// For debug
[[maybe_unused]] String toString() const
{
if (func)
2023-03-20 16:24:06 +00:00
return "func:" + func_name + ",fragment:" + fragment;
2023-02-27 13:41:38 +00:00
else
2023-03-20 16:24:06 +00:00
return "literal:" + literal + ",fragment:" + fragment;
2023-02-27 13:41:38 +00:00
}
2023-02-22 12:42:59 +00:00
2023-03-03 02:42:35 +00:00
Pos perform(Pos cur, Pos end, DateTime & date) const
2023-02-27 13:41:38 +00:00
{
if (func)
2023-03-20 16:24:06 +00:00
return func(cur, end, fragment, date);
2023-02-27 13:41:38 +00:00
else
{
2023-03-20 16:24:06 +00:00
/// literal:
checkSpace(cur, end, literal.size(), "insufficient space to parse literal", fragment);
2023-02-27 13:41:38 +00:00
if (std::string_view(cur, literal.size()) != literal)
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because literal {} is expected but {} provided",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
literal,
std::string_view(cur, literal.size()));
2023-02-27 13:41:38 +00:00
cur += literal.size();
return cur;
}
}
2023-02-22 10:05:53 +00:00
2023-03-03 02:42:35 +00:00
template <typename T, NeedCheckSpace need_check_space>
2023-03-20 16:24:06 +00:00
static Pos readNumber2(Pos cur, Pos end, [[maybe_unused]] const String & fragment, T & res)
2023-02-27 13:41:38 +00:00
{
2023-03-03 02:42:35 +00:00
if constexpr (need_check_space == NeedCheckSpace::Yes)
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 2, "readNumber2 requires size >= 2", fragment);
2023-03-01 09:15:29 +00:00
res = (*cur - '0');
2023-02-27 13:41:38 +00:00
++cur;
2023-03-01 09:15:29 +00:00
res = res * 10 + (*cur - '0');
2023-02-27 13:41:38 +00:00
++cur;
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-03 02:42:35 +00:00
template <typename T, NeedCheckSpace need_check_space>
2023-03-20 16:24:06 +00:00
static Pos readNumber3(Pos cur, Pos end, [[maybe_unused]] const String & fragment, T & res)
2023-02-27 13:41:38 +00:00
{
2023-03-03 02:42:35 +00:00
if constexpr (need_check_space == NeedCheckSpace::Yes)
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 3, "readNumber4 requires size >= 3", fragment);
2023-02-24 09:07:27 +00:00
2023-03-01 09:15:29 +00:00
res = (*cur - '0');
++cur;
res = res * 10 + (*cur - '0');
++cur;
2023-02-27 13:41:38 +00:00
res = res * 10 + (*cur - '0');
++cur;
return cur;
}
2023-03-03 02:42:35 +00:00
template <typename T, NeedCheckSpace need_check_space>
2023-03-20 16:24:06 +00:00
static Pos readNumber4(Pos cur, Pos end, [[maybe_unused]] const String & fragment, T & res)
2023-02-27 13:41:38 +00:00
{
2023-03-03 02:42:35 +00:00
if constexpr (need_check_space == NeedCheckSpace::Yes)
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 4, "readNumber4 requires size >= 4", fragment);
2023-03-01 09:15:29 +00:00
res = (*cur - '0');
++cur;
res = res * 10 + (*cur - '0');
++cur;
res = res * 10 + (*cur - '0');
++cur;
res = res * 10 + (*cur - '0');
++cur;
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static void checkSpace(Pos cur, Pos end, size_t len, const String & msg, const String & fragment)
2023-02-27 13:41:38 +00:00
{
2023-03-05 08:24:05 +00:00
if (cur > end || cur + len > end) [[unlikely]]
2023-03-06 09:08:55 +00:00
throw Exception(
ErrorCodes::NOT_ENOUGH_SPACE,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
msg);
2023-02-27 13:41:38 +00:00
}
2023-02-22 10:05:53 +00:00
2023-03-03 02:42:35 +00:00
template <NeedCheckSpace need_check_space>
2023-03-20 16:24:06 +00:00
static Pos assertChar(Pos cur, Pos end, char expected, const String & fragment)
2023-02-27 13:41:38 +00:00
{
2023-03-03 02:42:35 +00:00
if constexpr (need_check_space == NeedCheckSpace::Yes)
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 1, "assertChar requires size >= 1", fragment);
2023-02-22 10:05:53 +00:00
2023-03-20 15:20:57 +00:00
if (*cur != expected)
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because char {} is expected but {} provided",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
2023-03-20 15:20:57 +00:00
String(expected, 1),
2023-03-06 09:08:55 +00:00
String(*cur, 1));
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
++cur;
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfWeekTextShort(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 3, "mysqlDayOfWeekTextShort requires size >= 3", fragment);
2023-02-22 10:05:53 +00:00
2023-02-28 15:02:02 +00:00
String text(cur, 3);
2023-03-05 08:24:05 +00:00
boost::to_lower(text);
2023-03-01 03:02:41 +00:00
auto it = dayOfWeekMap.find(text);
if (it == dayOfWeekMap.end())
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because of unknown day of week short text {} ",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text);
2023-03-01 03:02:41 +00:00
date.setDayOfWeek(it->second.second);
2023-02-27 13:41:38 +00:00
cur += 3;
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlMonthOfYearTextShort(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 3, "mysqlMonthOfYearTextShort requires size >= 3", fragment);
2023-02-24 09:07:27 +00:00
2023-03-01 03:02:41 +00:00
String text(cur, 3);
2023-03-05 08:24:05 +00:00
boost::to_lower(text);
2023-03-01 03:02:41 +00:00
auto it = monthMap.find(text);
if (it == monthMap.end())
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because of unknown month of year short text {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text);
2023-02-22 10:05:53 +00:00
2023-03-01 03:02:41 +00:00
date.setMonth(it->second.second);
2023-02-27 13:41:38 +00:00
cur += 3;
return cur;
}
2023-02-22 10:05:53 +00:00
static Pos mysqlMonthOfYearTextLong(Pos cur, Pos end, const String & fragment, DateTime & date)
{
checkSpace(cur, end, 3, "mysqlMonthOfYearTextLong requires size >= 3", fragment);
String text1(cur, 3);
boost::to_lower(text1);
auto it = monthMap.find(text1);
if (it == monthMap.end())
throw Exception(
ErrorCodes::CANNOT_PARSE_DATETIME,
"Unable to parse first part of fragment {} from {} because of unknown month of year text: {}",
fragment,
std::string_view(cur, end - cur),
text1);
cur += 3;
size_t expected_remaining_size = it->second.first.size();
checkSpace(cur, end, expected_remaining_size, "mysqlMonthOfYearTextLong requires the second parg size >= " + std::to_string(expected_remaining_size), fragment);
String text2(cur, expected_remaining_size);
boost::to_lower(text2);
if (text2 != it->second.first)
throw Exception(
ErrorCodes::CANNOT_PARSE_DATETIME,
"Unable to parse second part of fragment {} from {} because of unknown month of year text: {}",
fragment,
std::string_view(cur, end - cur),
text1 + text2);
cur += expected_remaining_size;
date.setMonth(it->second.second);
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlMonth(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-28 06:08:03 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 month;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, month);
2023-02-28 15:02:02 +00:00
date.setMonth(month);
2023-02-28 06:08:03 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlCentury(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 century;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, century);
2023-03-03 02:42:35 +00:00
date.setCentury(century);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfMonth(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 day_of_month;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, day_of_month);
2023-03-06 09:35:08 +00:00
date.setDayOfMonth(day_of_month);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlAmericanDate(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 8, "mysqlAmericanDate requires size >= 8", fragment);
2023-03-01 09:15:29 +00:00
2023-02-28 15:02:02 +00:00
Int32 month;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, month);
cur = assertChar<NeedCheckSpace::No>(cur, end, '/', fragment);
2023-02-28 15:02:02 +00:00
date.setMonth(month);
2023-02-22 10:05:53 +00:00
2023-02-28 15:02:02 +00:00
Int32 day;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, day);
cur = assertChar<NeedCheckSpace::No>(cur, end, '/', fragment);
2023-03-06 09:35:08 +00:00
date.setDayOfMonth(day);
2023-02-22 10:05:53 +00:00
2023-02-28 15:02:02 +00:00
Int32 year;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, year);
2023-02-28 15:02:02 +00:00
date.setYear(year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfMonthSpacePadded(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 2, "mysqlDayOfMonthSpacePadded requires size >= 2", fragment);
2023-02-24 09:07:27 +00:00
2023-02-28 15:02:02 +00:00
Int32 day_of_month = *cur == ' ' ? 0 : (*cur - '0');
2023-02-27 13:41:38 +00:00
++cur;
2023-02-22 10:05:53 +00:00
2023-02-28 15:02:02 +00:00
day_of_month = 10 * day_of_month + (*cur - '0');
2023-02-27 13:41:38 +00:00
++cur;
2023-02-22 10:05:53 +00:00
2023-03-06 09:35:08 +00:00
date.setDayOfMonth(day_of_month);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlISO8601Date(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 10, "mysqlISO8601Date requires size >= 10", fragment);
2023-02-27 13:41:38 +00:00
2023-03-01 09:15:29 +00:00
Int32 year;
2023-02-28 15:02:02 +00:00
Int32 month;
Int32 day;
2023-03-20 16:24:06 +00:00
cur = readNumber4<Int32, NeedCheckSpace::No>(cur, end, fragment, year);
cur = assertChar<NeedCheckSpace::No>(cur, end, '-', fragment);
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, month);
cur = assertChar<NeedCheckSpace::No>(cur, end, '-', fragment);
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, day);
2023-03-01 09:15:29 +00:00
date.setYear(year);
date.setMonth(month);
2023-03-06 09:35:08 +00:00
date.setDayOfMonth(day);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlISO8601Year2(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year2;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, year2);
2023-02-28 15:02:02 +00:00
date.setYear2(year2);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlISO8601Year4(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year;
2023-03-20 16:24:06 +00:00
cur = readNumber4<Int32, NeedCheckSpace::Yes>(cur, end, fragment, year);
2023-02-28 15:02:02 +00:00
date.setYear(year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfYear(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 day_of_year;
2023-03-20 16:24:06 +00:00
cur = readNumber3<Int32, NeedCheckSpace::Yes>(cur, end, fragment, day_of_year);
2023-03-07 06:35:50 +00:00
date.setDayOfYear(day_of_year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfWeek(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 1, "mysqlDayOfWeek requires size >= 1", fragment);
2023-02-28 15:02:02 +00:00
date.setDayOfWeek(*cur - '0');
++cur;
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlISO8601Week(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 week;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, week);
2023-02-28 15:02:02 +00:00
date.setWeek(week);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfWeek0To6(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 1, "mysqlDayOfWeek requires size >= 1", fragment);
2023-02-28 15:02:02 +00:00
Int32 day_of_week = *cur - '0';
if (day_of_week == 0)
day_of_week = 7;
2023-02-28 15:02:02 +00:00
date.setDayOfWeek(day_of_week);
++cur;
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlDayOfWeekTextLong(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
checkSpace(cur, end, 6, "mysqlDayOfWeekTextLong requires size >= 6", fragment);
2023-02-28 15:02:02 +00:00
String text1(cur, 3);
2023-03-05 08:24:05 +00:00
boost::to_lower(text1);
2023-02-28 15:02:02 +00:00
auto it = dayOfWeekMap.find(text1);
if (it == dayOfWeekMap.end())
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse first part of fragment {} from {} because of unknown day of week text: {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text1);
2023-02-28 15:02:02 +00:00
cur += 3;
2023-03-20 15:20:57 +00:00
size_t expected_remaining_size = it->second.first.size();
checkSpace(cur, end, expected_remaining_size, "mysqlDayOfWeekTextLong requires the second parg size >= " + std::to_string(expected_remaining_size), fragment);
2023-03-20 15:20:57 +00:00
String text2(cur, expected_remaining_size);
2023-03-05 08:24:05 +00:00
boost::to_lower(text2);
2023-02-28 15:02:02 +00:00
if (text2 != it->second.first)
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse second part of fragment {} from {} because of unknown day of week text: {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text1 + text2);
2023-03-20 15:20:57 +00:00
cur += expected_remaining_size;
2023-02-28 15:02:02 +00:00
date.setDayOfWeek(it->second.second);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlYear2(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year2;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, year2);
2023-02-28 15:02:02 +00:00
date.setYear2(year2);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlYear4(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year;
2023-03-20 16:24:06 +00:00
cur = readNumber4<Int32, NeedCheckSpace::Yes>(cur, end, fragment, year);
2023-02-28 15:02:02 +00:00
date.setYear(year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 12:42:59 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlTimezoneOffset(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 5, "mysqlTimezoneOffset requires size >= 5", fragment);
2023-03-01 09:15:29 +00:00
Int32 sign;
2023-02-27 13:41:38 +00:00
if (*cur == '-')
sign = -1;
2023-03-01 09:15:29 +00:00
else if (*cur == '+')
sign = 1;
else
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because of unknown sign time zone offset: {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
std::string_view(cur, 1));
2023-02-27 13:41:38 +00:00
++cur;
2023-02-22 12:42:59 +00:00
2023-02-27 13:41:38 +00:00
Int32 hour;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, hour);
2023-02-22 12:42:59 +00:00
2023-02-27 13:41:38 +00:00
Int32 minute;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, minute);
2023-02-22 12:42:59 +00:00
2023-03-07 03:45:32 +00:00
date.has_time_zone_offset = true;
date.time_zone_offset = sign * (hour * 3600 + minute * 60);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlMinute(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-28 06:08:03 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 minute;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, minute);
2023-02-28 15:02:02 +00:00
date.setMinute(minute);
2023-02-28 06:08:03 +00:00
return cur;
}
2023-02-27 13:41:38 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlAMPM(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-24 10:27:21 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 2, "mysqlAMPM requires size >= 2", fragment);
2023-02-27 13:41:38 +00:00
2023-02-28 15:02:02 +00:00
String text(cur, 2);
2023-03-20 15:58:37 +00:00
boost::to_lower(text);
2023-02-28 15:02:02 +00:00
date.setAMPM(text);
2023-02-27 13:41:38 +00:00
cur += 2;
return cur;
2023-02-24 10:27:21 +00:00
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlHHMM12(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 8, "mysqlHHMM12 requires size >= 8", fragment);
2023-03-01 09:15:29 +00:00
2023-02-28 15:02:02 +00:00
Int32 hour;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, hour);
cur = assertChar<NeedCheckSpace::No>(cur, end, ':', fragment);
2023-02-28 15:02:02 +00:00
date.setHour(hour, true, true);
Int32 minute;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, minute);
cur = assertChar<NeedCheckSpace::No>(cur, end, ' ', fragment);
2023-02-28 15:02:02 +00:00
date.setMinute(minute);
2023-03-20 16:24:06 +00:00
cur = mysqlAMPM(cur, end, fragment, date);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 12:42:59 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlHHMM24(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 5, "mysqlHHMM24 requires size >= 5", fragment);
2023-03-01 09:15:29 +00:00
2023-02-28 15:02:02 +00:00
Int32 hour;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, hour);
cur = assertChar<NeedCheckSpace::No>(cur, end, ':', fragment);
2023-02-28 15:02:02 +00:00
date.setHour(hour, false, false);
Int32 minute;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, minute);
2023-02-28 15:02:02 +00:00
date.setMinute(minute);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlSecond(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-28 06:08:03 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 second;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, second);
2023-02-28 15:02:02 +00:00
date.setSecond(second);
2023-02-28 06:08:03 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlISO8601Time(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 8, "mysqlISO8601Time requires size >= 8", fragment);
2023-02-28 15:02:02 +00:00
2023-03-01 09:15:29 +00:00
Int32 hour;
2023-02-28 15:02:02 +00:00
Int32 minute;
Int32 second;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, hour);
cur = assertChar<NeedCheckSpace::No>(cur, end, ':', fragment);
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, minute);
cur = assertChar<NeedCheckSpace::No>(cur, end, ':', fragment);
cur = readNumber2<Int32, NeedCheckSpace::No>(cur, end, fragment, second);
2023-03-01 09:15:29 +00:00
date.setHour(hour, false, false);
date.setMinute(minute);
2023-02-28 15:02:02 +00:00
date.setSecond(second);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos mysqlHour12(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-22 12:42:59 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, true, true);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-03-20 16:24:06 +00:00
static Pos mysqlHour24(Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
2023-03-20 16:24:06 +00:00
cur = readNumber2<Int32, NeedCheckSpace::Yes>(cur, end, fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, false, false);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
static Pos readNumberWithVariableLength(
Pos cur,
Pos end,
bool allow_negative,
bool allow_plus_sign,
bool is_year,
size_t repetitions,
size_t max_digits_to_read,
2023-03-20 16:24:06 +00:00
const String & fragment,
Int32 & result)
2023-02-27 13:41:38 +00:00
{
2023-02-27 13:41:38 +00:00
bool negative = false;
if (allow_negative && cur < end && *cur == '-')
2023-02-24 09:07:27 +00:00
{
2023-02-27 13:41:38 +00:00
negative = true;
++cur;
}
else if (allow_plus_sign && cur < end && *cur == '+')
{
negative = false;
++cur;
}
2023-02-24 09:07:27 +00:00
Int64 number = 0;
2023-03-20 15:20:57 +00:00
const Pos start = cur;
/// Avoid integer overflow in (*)
if (max_digits_to_read >= std::numeric_limits<decltype(number)>::digits10) [[unlikely]]
throw Exception(
ErrorCodes::CANNOT_PARSE_DATETIME,
"Unable to parse fragment {} from {} because max_digits_to_read is too big",
fragment,
std::string_view(start, cur - start));
2023-02-27 13:41:38 +00:00
if (is_year && repetitions == 2)
{
// If abbreviated two year digit is provided in format string, try to read
// in two digits of year and convert to appropriate full length year The
// two-digit mapping is as follows: [00, 69] -> [2000, 2069]
// [70, 99] -> [1970, 1999]
// If more than two digits are provided, then simply read in full year
// normally without conversion
size_t count = 0;
2023-03-20 15:20:57 +00:00
while (cur < end && cur < start + max_digits_to_read && *cur >= '0' && *cur <= '9')
2023-02-27 13:41:38 +00:00
{
number = number * 10 + (*cur - '0'); /// (*)
2023-02-27 13:41:38 +00:00
++cur;
++count;
}
if (count == 2)
{
if (number >= 70)
number += 1900;
else if (number >= 0 && number < 70)
number += 2000;
}
else
{
2023-03-20 15:20:57 +00:00
while (cur < end && cur < start + max_digits_to_read && *cur >= '0' && *cur <= '9')
2023-02-27 13:41:38 +00:00
{
number = number * 10 + (*cur - '0'); /// (*)
2023-02-27 13:41:38 +00:00
++cur;
}
}
}
else
{
2023-03-20 15:20:57 +00:00
while (cur < end && cur < start + max_digits_to_read && *cur >= '0' && *cur <= '9')
2023-02-27 13:41:38 +00:00
{
number = number * 10 + (*cur - '0');
++cur;
}
}
2023-02-24 09:07:27 +00:00
if (negative)
number *= -1;
2023-02-27 13:41:38 +00:00
/// Need to have read at least one digit.
if (cur == start) [[unlikely]]
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because read number failed",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur));
2023-02-24 09:07:27 +00:00
/// Check if number exceeds the range of Int32
if (number < std::numeric_limits<Int32>::min() || number > std::numeric_limits<Int32>::max()) [[unlikely]]
throw Exception(
ErrorCodes::CANNOT_PARSE_DATETIME,
"Unable to parse fragment {} from {} because number is out of range of Int32",
fragment,
std::string_view(start, cur - start));
result = static_cast<Int32>(number);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos jodaEra(int, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 2, "jodaEra requires size >= 2", fragment);
2023-02-28 12:42:16 +00:00
2023-02-28 15:02:02 +00:00
String era(cur, 2);
2023-03-20 15:58:37 +00:00
boost::to_lower(era);
2023-02-28 15:02:02 +00:00
date.setEra(era);
2023-02-27 13:41:38 +00:00
cur += 2;
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaCenturyOfEra(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 century;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, repetitions, fragment, century);
2023-03-03 02:42:35 +00:00
date.setCentury(century);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaYearOfEra(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year_of_era;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, false, false, true, repetitions, repetitions, fragment, year_of_era);
2023-02-28 15:02:02 +00:00
date.setYear(year_of_era, true);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaWeekYear(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 week_year;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, true, true, true, repetitions, repetitions, fragment, week_year);
2023-02-28 15:02:02 +00:00
date.setYear(week_year, false, true);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaWeekOfWeekYear(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 week;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, week);
2023-02-28 15:02:02 +00:00
date.setWeek(week);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaDayOfWeek1Based(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 day_of_week;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, repetitions, fragment, day_of_week);
2023-02-28 15:02:02 +00:00
date.setDayOfWeek(day_of_week);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-06 09:08:55 +00:00
static Pos
2023-03-20 16:24:06 +00:00
jodaDayOfWeekText(size_t /*min_represent_digits*/, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 3, "jodaDayOfWeekText requires size >= 3", fragment);
2023-02-24 09:07:27 +00:00
2023-02-27 13:41:38 +00:00
String text1(cur, 3);
2023-03-05 08:24:05 +00:00
boost::to_lower(text1);
2023-02-27 13:41:38 +00:00
auto it = dayOfWeekMap.find(text1);
if (it == dayOfWeekMap.end())
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because of unknown day of week text: {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text1);
2023-02-27 13:41:38 +00:00
cur += 3;
2023-02-28 15:02:02 +00:00
date.setDayOfWeek(it->second.second);
2023-03-20 15:20:57 +00:00
size_t expected_remaining_size = it->second.first.size();
if (cur + expected_remaining_size <= end)
2023-02-27 13:41:38 +00:00
{
2023-03-20 15:20:57 +00:00
String text2(cur, expected_remaining_size);
2023-03-05 08:24:05 +00:00
boost::to_lower(text2);
2023-02-27 13:41:38 +00:00
if (text2 == it->second.first)
{
2023-03-20 15:20:57 +00:00
cur += expected_remaining_size;
2023-02-27 13:41:38 +00:00
return cur;
}
}
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaYear(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 year;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, true, true, true, repetitions, repetitions, fragment, year);
2023-02-28 15:02:02 +00:00
date.setYear(year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaDayOfYear(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 day_of_year;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 3uz), fragment, day_of_year);
2023-03-07 06:35:50 +00:00
date.setDayOfYear(day_of_year);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaMonthOfYear(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 month;
2023-03-20 16:24:06 +00:00
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, 2, fragment, month);
2023-02-28 15:02:02 +00:00
date.setMonth(month);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos jodaMonthOfYearText(int, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 3, "jodaMonthOfYearText requires size >= 3", fragment);
2023-02-27 13:41:38 +00:00
String text1(cur, 3);
2023-03-05 08:24:05 +00:00
boost::to_lower(text1);
2023-02-27 13:41:38 +00:00
auto it = monthMap.find(text1);
if (it == monthMap.end())
2023-03-06 09:08:55 +00:00
throw Exception(
2023-03-07 06:35:50 +00:00
ErrorCodes::CANNOT_PARSE_DATETIME,
2023-03-20 16:24:06 +00:00
"Unable to parse fragment {} from {} because of unknown month of year text: {}",
fragment,
2023-03-06 09:08:55 +00:00
std::string_view(cur, end - cur),
text1);
2023-02-27 13:41:38 +00:00
cur += 3;
2023-02-28 15:02:02 +00:00
date.setMonth(it->second.second);
2023-03-20 15:20:57 +00:00
size_t expected_remaining_size = it->second.first.size();
if (cur + expected_remaining_size <= end)
2023-02-27 13:41:38 +00:00
{
2023-03-20 15:20:57 +00:00
String text2(cur, expected_remaining_size);
2023-03-05 08:24:05 +00:00
boost::to_lower(text2);
2023-02-27 13:41:38 +00:00
if (text2 == it->second.first)
{
2023-03-20 15:20:57 +00:00
cur += expected_remaining_size;
2023-02-27 13:41:38 +00:00
return cur;
}
}
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaDayOfMonth(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 day_of_month;
2023-03-06 09:08:55 +00:00
cur = readNumberWithVariableLength(
cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, day_of_month);
2023-03-06 09:35:08 +00:00
date.setDayOfMonth(day_of_month);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
2023-03-20 16:24:06 +00:00
static Pos jodaHalfDayOfDay(int, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-03-20 16:24:06 +00:00
checkSpace(cur, end, 2, "jodaHalfDayOfDay requires size >= 2", fragment);
2023-02-27 13:41:38 +00:00
String text(cur, 2);
2023-03-20 15:58:37 +00:00
boost::to_lower(text);
2023-02-28 15:02:02 +00:00
date.setAMPM(text);
2023-02-27 13:41:38 +00:00
cur += 2;
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaHourOfHalfDay(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, true, false);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaClockHourOfHalfDay(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, true, true);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaHourOfDay(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, false, false);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaClockHourOfDay(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 hour;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, hour);
2023-02-28 15:02:02 +00:00
date.setHour(hour, false, true);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaMinuteOfHour(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 minute;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, minute);
2023-02-28 15:02:02 +00:00
date.setMinute(minute);
2023-02-27 13:41:38 +00:00
return cur;
}
2023-02-24 09:07:27 +00:00
static Pos jodaSecondOfMinute(size_t repetitions, Pos cur, Pos end, const String & fragment, DateTime & date)
2023-02-27 13:41:38 +00:00
{
2023-02-28 15:02:02 +00:00
Int32 second;
cur = readNumberWithVariableLength(cur, end, false, false, false, repetitions, std::max(repetitions, 2uz), fragment, second);
2023-02-28 15:02:02 +00:00
date.setSecond(second);
2023-02-27 13:41:38 +00:00
return cur;
}
};
2023-02-24 09:07:27 +00:00
2023-03-06 09:08:55 +00:00
std::vector<Instruction> parseFormat(const String & format) const
2023-02-27 13:41:38 +00:00
{
2023-03-06 09:08:55 +00:00
static_assert(
2023-03-20 15:58:37 +00:00
parse_syntax == ParseSyntax::MySQL || parse_syntax == ParseSyntax::Joda,
2023-03-06 09:08:55 +00:00
"parse syntax must be one of MySQL or Joda");
2023-03-20 15:58:37 +00:00
if constexpr (parse_syntax == ParseSyntax::MySQL)
2023-03-06 09:08:55 +00:00
return parseMysqlFormat(format);
2023-02-27 13:41:38 +00:00
else
2023-03-06 09:08:55 +00:00
return parseJodaFormat(format);
2023-02-27 13:41:38 +00:00
}
2023-02-24 09:07:27 +00:00
2023-03-06 09:08:55 +00:00
std::vector<Instruction> parseMysqlFormat(const String & format) const
2023-02-27 13:41:38 +00:00
{
2023-03-06 09:08:55 +00:00
#define ACTION_ARGS(func) &(func), #func, std::string_view(pos - 1, 2)
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
Pos pos = format.data();
2023-03-06 09:08:55 +00:00
Pos end = format.data() + format.size();
std::vector<Instruction> instructions;
2023-02-27 13:41:38 +00:00
while (true)
{
2023-03-05 08:24:05 +00:00
Pos next_percent_pos = find_first_symbols<'%'>(pos, end);
2023-03-06 09:08:55 +00:00
2023-03-05 08:24:05 +00:00
if (next_percent_pos < end)
2023-02-27 13:41:38 +00:00
{
2023-03-05 08:24:05 +00:00
if (pos < next_percent_pos)
instructions.emplace_back(String(pos, next_percent_pos - pos));
2023-02-27 13:41:38 +00:00
2023-03-05 08:24:05 +00:00
pos = next_percent_pos + 1;
2023-02-27 13:41:38 +00:00
if (pos >= end)
2023-03-20 15:58:37 +00:00
throw Exception(
ErrorCodes::BAD_ARGUMENTS, "'%' must not be the last character in the format string, use '%%' instead");
2023-02-27 13:41:38 +00:00
switch (*pos)
{
// Abbreviated weekday [Mon...Sun]
case 'a':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfWeekTextShort));
2023-02-27 13:41:38 +00:00
break;
// Abbreviated month [Jan...Dec]
case 'b':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlMonthOfYearTextShort));
2023-02-27 13:41:38 +00:00
break;
// Month as a decimal number (01-12)
case 'c':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlMonth));
2023-02-27 13:41:38 +00:00
break;
// Year, divided by 100, zero-padded
case 'C':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlCentury));
2023-02-27 13:41:38 +00:00
break;
// Day of month, zero-padded (01-31)
case 'd':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfMonth));
2023-02-27 13:41:38 +00:00
break;
// Short MM/DD/YY date, equivalent to %m/%d/%y
case 'D':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlAmericanDate));
2023-02-27 13:41:38 +00:00
break;
// Day of month, space-padded ( 1-31) 23
case 'e':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfMonthSpacePadded));
2023-02-27 13:41:38 +00:00
break;
// Short YYYY-MM-DD date, equivalent to %Y-%m-%d 2001-08-23
case 'F':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlISO8601Date));
2023-02-27 13:41:38 +00:00
break;
// Last two digits of year of ISO 8601 week number (see %G)
case 'g':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlISO8601Year2));
2023-02-27 13:41:38 +00:00
break;
// Year of ISO 8601 week number (see %V)
case 'G':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlISO8601Year4));
2023-02-27 13:41:38 +00:00
break;
// Day of the year (001-366) 235
case 'j':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfYear));
2023-02-27 13:41:38 +00:00
break;
// Month as a decimal number (01-12)
case 'm':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlMonth));
2023-02-27 13:41:38 +00:00
break;
// ISO 8601 weekday as number with Monday as 1 (1-7)
case 'u':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfWeek));
2023-02-27 13:41:38 +00:00
break;
// ISO 8601 week number (01-53)
case 'V':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlISO8601Week));
2023-02-27 13:41:38 +00:00
break;
2023-03-20 15:58:37 +00:00
// Weekday as a integer number with Sunday as 0 (0-6) 4
2023-02-27 13:41:38 +00:00
case 'w':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfWeek0To6));
2023-02-27 13:41:38 +00:00
break;
// Full weekday [Monday...Sunday]
case 'W':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlDayOfWeekTextLong));
2023-02-27 13:41:38 +00:00
break;
// Two digits year
case 'y':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlYear2));
2023-02-27 13:41:38 +00:00
break;
// Four digits year
case 'Y':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlYear4));
2023-02-27 13:41:38 +00:00
break;
// Quarter (1-4)
case 'Q':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for quarter");
break;
// Offset from UTC timezone as +hhmm or -hhmm
case 'z':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlTimezoneOffset));
2023-02-27 13:41:38 +00:00
break;
// Minute (00-59)
case 'M':
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlMonthOfYearTextLong));
2023-02-27 13:41:38 +00:00
break;
// AM or PM
case 'p':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlAMPM));
2023-02-27 13:41:38 +00:00
break;
// 12-hour HH:MM time, equivalent to %h:%i %p 2:55 PM
case 'r':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHHMM12));
2023-02-27 13:41:38 +00:00
break;
// 24-hour HH:MM time, equivalent to %H:%i 14:55
case 'R':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHHMM24));
2023-02-27 13:41:38 +00:00
break;
// Seconds
case 's':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlSecond));
2023-02-27 13:41:38 +00:00
break;
// Seconds
case 'S':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlSecond));
2023-02-27 13:41:38 +00:00
break;
// ISO 8601 time format (HH:MM:SS), equivalent to %H:%i:%S 14:55:02
case 'T':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlISO8601Time));
2023-02-27 13:41:38 +00:00
break;
// Hour in 12h format (01-12)
case 'h':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHour12));
2023-02-27 13:41:38 +00:00
break;
// Hour in 24h format (00-23)
case 'H':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHour24));
2023-02-27 13:41:38 +00:00
break;
// Minute of hour range [0, 59]
case 'i':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlMinute));
2023-02-27 13:41:38 +00:00
break;
// Hour in 12h format (01-12)
case 'I':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHour12));
2023-02-27 13:41:38 +00:00
break;
// Hour in 24h format (00-23)
case 'k':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHour24));
2023-02-27 13:41:38 +00:00
break;
// Hour in 12h format (01-12)
case 'l':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS(Instruction::mysqlHour12));
2023-02-27 13:41:38 +00:00
break;
case 't':
instructions.emplace_back("\t");
break;
case 'n':
instructions.emplace_back("\n");
break;
// Escaped literal characters.
case '%':
2023-03-20 15:20:57 +00:00
instructions.emplace_back("%");
2023-02-27 13:41:38 +00:00
break;
2023-03-20 15:20:57 +00:00
/// Unimplemented
/// Fractional seconds
case 'f':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for fractional seconds");
2023-02-27 13:41:38 +00:00
case 'U':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for WEEK (Sun-Sat)");
case 'v':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for WEEK (Mon-Sun)");
case 'x':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for YEAR for week (Mon-Sun)");
case 'X':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for YEAR for week (Sun-Sat)");
default:
throw Exception(
ErrorCodes::BAD_ARGUMENTS,
"Incorrect syntax '{}', symbol is not supported '{}' for function {}",
format,
*pos,
getName());
}
++pos;
}
else
{
2023-03-20 15:58:37 +00:00
/// Handle characters after last %
2023-02-27 13:41:38 +00:00
if (pos < end)
instructions.emplace_back(String(pos, end - pos));
break;
}
2023-02-24 09:07:27 +00:00
}
2023-03-06 09:08:55 +00:00
return instructions;
2023-02-27 13:41:38 +00:00
#undef ACTION_ARGS
}
2023-03-06 09:08:55 +00:00
std::vector<Instruction> parseJodaFormat(const String & format) const
2023-02-27 13:41:38 +00:00
{
2023-03-06 09:08:55 +00:00
#define ACTION_ARGS_WITH_BIND(func, arg) std::bind_front(&(func), (arg)), #func, std::string_view(cur_token, repetitions)
2023-02-27 13:41:38 +00:00
2023-03-20 15:20:57 +00:00
Pos pos = format.data();
Pos end = format.data() + format.size();
2023-02-27 13:41:38 +00:00
2023-03-06 09:08:55 +00:00
std::vector<Instruction> instructions;
2023-02-27 13:41:38 +00:00
while (pos < end)
2023-02-22 10:05:53 +00:00
{
2023-03-20 15:20:57 +00:00
Pos cur_token = pos;
2023-02-27 13:41:38 +00:00
// Literal case
if (*cur_token == '\'')
{
// Case 1: 2 consecutive single quote
if (pos + 1 < end && *(pos + 1) == '\'')
{
instructions.emplace_back(String(cur_token, 1));
pos += 2;
}
else
{
// Case 2: find closing single quote
Int64 count = numLiteralChars(cur_token + 1, end);
if (count == -1)
throw Exception(ErrorCodes::BAD_ARGUMENTS, "No closing single quote for literal");
else
{
for (Int64 i = 1; i <= count; i++)
{
instructions.emplace_back(String(cur_token + i, 1));
if (*(cur_token + i) == '\'')
i += 1;
}
pos += count + 2;
}
}
}
else
{
size_t repetitions = 1;
2023-02-27 13:41:38 +00:00
++pos;
while (pos < end && *cur_token == *pos)
{
++repetitions;
++pos;
}
switch (*cur_token)
{
case 'G':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaEra, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'C':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaCenturyOfEra, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'Y':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaYearOfEra, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'x':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaWeekYear, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'w':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaWeekOfWeekYear, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'e':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaDayOfWeek1Based, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'E':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaDayOfWeekText, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'y':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaYear, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'D':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaDayOfYear, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'M':
if (repetitions <= 2)
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaMonthOfYear, repetitions));
2023-02-27 13:41:38 +00:00
else
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaMonthOfYearText, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'd':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaDayOfMonth, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'a':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaHalfDayOfDay, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'K':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaHourOfHalfDay, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'h':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaClockHourOfHalfDay, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'H':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaHourOfDay, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'k':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaClockHourOfDay, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'm':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaMinuteOfHour, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 's':
2023-03-05 08:24:05 +00:00
instructions.emplace_back(ACTION_ARGS_WITH_BIND(Instruction::jodaSecondOfMinute, repetitions));
2023-02-27 13:41:38 +00:00
break;
case 'S':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for fractional seconds");
case 'z':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for timezone");
case 'Z':
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for timezone offset id");
default:
if (isalpha(*cur_token))
throw Exception(
ErrorCodes::NOT_IMPLEMENTED, "format is not supported for {}", String(cur_token, repetitions));
instructions.emplace_back(String(cur_token, pos - cur_token));
break;
}
}
2023-02-22 10:05:53 +00:00
}
2023-03-06 09:08:55 +00:00
return instructions;
2023-02-28 06:08:03 +00:00
#undef ACTION_ARGS_WITH_BIND
2023-02-22 10:05:53 +00:00
}
2023-03-03 02:42:35 +00:00
String getFormat(const ColumnsWithTypeAndName & arguments) const
2023-02-27 13:41:38 +00:00
{
if (arguments.size() < 2)
{
2023-03-20 15:58:37 +00:00
if constexpr (parse_syntax == ParseSyntax::Joda)
2023-02-27 13:41:38 +00:00
return "yyyy-MM-dd HH:mm:ss";
else
return "%Y-%m-%d %H:%M:%S";
}
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
const auto * format_column = checkAndGetColumnConst<ColumnString>(arguments[1].column.get());
2023-03-22 08:45:46 +00:00
if (!format_column)
throw Exception(
ErrorCodes::ILLEGAL_COLUMN,
"Illegal column {} of second ('format') argument of function {}. Must be constant string.",
arguments[1].column->getName(),
getName());
2023-02-27 13:41:38 +00:00
return format_column->getValue<String>();
}
2023-03-06 09:08:55 +00:00
const DateLUTImpl & getTimeZone(const ColumnsWithTypeAndName & arguments) const
2023-02-22 10:05:53 +00:00
{
2023-02-27 13:41:38 +00:00
if (arguments.size() < 3)
2023-03-06 09:08:55 +00:00
return DateLUT::instance();
2023-02-27 13:41:38 +00:00
const auto * col = checkAndGetColumnConst<ColumnString>(arguments[2].column.get());
if (!col)
throw Exception(
ErrorCodes::ILLEGAL_COLUMN,
2023-03-20 15:58:37 +00:00
"Illegal column {} of third ('timezone') argument of function {}. Must be constant String.",
2023-02-27 13:41:38 +00:00
arguments[2].column->getName(),
getName());
String time_zone = col->getValue<String>();
2023-03-06 09:08:55 +00:00
return DateLUT::instance(time_zone);
2023-02-22 10:05:53 +00:00
}
2023-02-27 13:41:38 +00:00
};
2023-02-22 10:05:53 +00:00
2023-02-27 13:41:38 +00:00
struct NameParseDateTime
{
static constexpr auto name = "parseDateTime";
};
2023-02-24 09:07:27 +00:00
struct NameParseDateTimeOrZero
{
static constexpr auto name = "parseDateTimeOrZero";
};
struct NameParseDateTimeOrNull
{
static constexpr auto name = "parseDateTimeOrNull";
};
2023-02-27 13:41:38 +00:00
struct NameParseDateTimeInJodaSyntax
2023-02-24 09:07:27 +00:00
{
2023-02-27 13:41:38 +00:00
static constexpr auto name = "parseDateTimeInJodaSyntax";
};
2023-02-24 09:07:27 +00:00
struct NameParseDateTimeInJodaSyntaxOrZero
{
static constexpr auto name = "parseDateTimeInJodaSyntaxOrZero";
};
struct NameParseDateTimeInJodaSyntaxOrNull
{
static constexpr auto name = "parseDateTimeInJodaSyntaxOrNull";
};
2023-02-22 10:05:53 +00:00
using FunctionParseDateTime = FunctionParseDateTimeImpl<NameParseDateTime, ParseSyntax::MySQL, ErrorHandling::Exception>;
using FunctionParseDateTimeOrZero = FunctionParseDateTimeImpl<NameParseDateTimeOrZero, ParseSyntax::MySQL, ErrorHandling::Zero>;
using FunctionParseDateTimeOrNull = FunctionParseDateTimeImpl<NameParseDateTimeOrNull, ParseSyntax::MySQL, ErrorHandling::Null>;
using FunctionParseDateTimeInJodaSyntax = FunctionParseDateTimeImpl<NameParseDateTimeInJodaSyntax, ParseSyntax::Joda, ErrorHandling::Exception>;
using FunctionParseDateTimeInJodaSyntaxOrZero = FunctionParseDateTimeImpl<NameParseDateTimeInJodaSyntaxOrZero, ParseSyntax::Joda, ErrorHandling::Zero>;
using FunctionParseDateTimeInJodaSyntaxOrNull = FunctionParseDateTimeImpl<NameParseDateTimeInJodaSyntaxOrNull, ParseSyntax::Joda, ErrorHandling::Null>;
2023-02-24 09:07:27 +00:00
}
2023-02-22 10:05:53 +00:00
2023-02-24 09:07:27 +00:00
REGISTER_FUNCTION(ParseDateTime)
{
factory.registerFunction<FunctionParseDateTime>();
2023-03-20 15:58:37 +00:00
factory.registerAlias("TO_UNIXTIME", FunctionParseDateTime::name);
factory.registerFunction<FunctionParseDateTimeOrZero>();
factory.registerFunction<FunctionParseDateTimeOrNull>();
factory.registerAlias("str_to_date", FunctionParseDateTimeOrNull::name);
2023-02-27 13:41:38 +00:00
factory.registerFunction<FunctionParseDateTimeInJodaSyntax>();
factory.registerFunction<FunctionParseDateTimeInJodaSyntaxOrZero>();
factory.registerFunction<FunctionParseDateTimeInJodaSyntaxOrNull>();
2023-02-22 10:05:53 +00:00
}
2023-02-24 09:07:27 +00:00
2023-02-22 10:05:53 +00:00
}