2018-09-26 00:31:40 +00:00
|
|
|
#include <DataTypes/DataTypeString.h>
|
2019-11-06 12:01:39 +00:00
|
|
|
#include <DataTypes/DataTypeDate.h>
|
2022-10-26 08:00:12 +00:00
|
|
|
#include <DataTypes/DataTypeDate32.h>
|
2019-11-06 12:01:39 +00:00
|
|
|
#include <DataTypes/DataTypeDateTime.h>
|
|
|
|
#include <DataTypes/DataTypeDateTime64.h>
|
2022-12-14 06:28:43 +00:00
|
|
|
#include <DataTypes/NumberTraits.h>
|
2018-09-26 00:31:40 +00:00
|
|
|
#include <Columns/ColumnString.h>
|
|
|
|
|
|
|
|
#include <Functions/DateTimeTransforms.h>
|
2020-07-14 06:04:37 +00:00
|
|
|
#include <Functions/FunctionFactory.h>
|
|
|
|
#include <Functions/FunctionHelpers.h>
|
|
|
|
#include <Functions/FunctionsConversion.h>
|
2021-05-17 07:30:42 +00:00
|
|
|
#include <Functions/IFunction.h>
|
2020-07-14 06:04:37 +00:00
|
|
|
#include <Functions/castTypeToEither.h>
|
2018-09-26 00:31:40 +00:00
|
|
|
#include <Functions/extractTimeZoneFromFunctionArguments.h>
|
|
|
|
|
|
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
|
2021-12-21 13:41:53 +00:00
|
|
|
#include <Common/DateLUTImpl.h>
|
2021-10-02 07:13:14 +00:00
|
|
|
#include <base/find_symbols.h>
|
2019-11-06 12:01:39 +00:00
|
|
|
#include <Core/DecimalFunctions.h>
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
#include <type_traits>
|
2022-12-14 06:28:43 +00:00
|
|
|
#include <concepts>
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
|
|
|
|
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
|
|
|
|
extern const int NOT_IMPLEMENTED;
|
|
|
|
extern const int ILLEGAL_COLUMN;
|
|
|
|
extern const int BAD_ARGUMENTS;
|
|
|
|
}
|
|
|
|
|
2019-11-08 11:52:01 +00:00
|
|
|
namespace
|
|
|
|
{
|
2020-09-07 18:00:37 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
struct FormatDateTimeTraits
|
|
|
|
{
|
|
|
|
enum class SupportInteger
|
|
|
|
{
|
|
|
|
Yes,
|
|
|
|
No
|
|
|
|
};
|
|
|
|
|
|
|
|
enum class FormatSyntax
|
|
|
|
{
|
|
|
|
MySQL,
|
|
|
|
Joda
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2020-07-23 10:23:55 +00:00
|
|
|
template <typename DataType> struct ActionValueTypeMap {};
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeInt8> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeUInt8> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeInt16> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeUInt16> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeInt32> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeUInt32> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeInt64> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeUInt64> { using ActionValueType = UInt32; };
|
|
|
|
template <> struct ActionValueTypeMap<DataTypeDate> { using ActionValueType = UInt16; };
|
2022-10-26 08:00:12 +00:00
|
|
|
template <> struct ActionValueTypeMap<DataTypeDate32> { using ActionValueType = Int32; };
|
2020-07-23 10:23:55 +00:00
|
|
|
template <> struct ActionValueTypeMap<DataTypeDateTime> { using ActionValueType = UInt32; };
|
2020-04-17 13:26:44 +00:00
|
|
|
template <> struct ActionValueTypeMap<DataTypeDateTime64> { using ActionValueType = Int64; };
|
2020-09-07 18:00:37 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
/// Counts the number of literal characters in Joda format string until the next closing literal
|
|
|
|
/// sequence single quote. Returns -1 if no literal single quote was found.
|
|
|
|
/// In Joda format string(https://joda-time.sourceforge.net/apidocs/org/joda/time/format/DateTimeFormat.html)
|
|
|
|
/// literal content must be quoted with single quote. and two single quote means literal with one single quote.
|
|
|
|
/// For example:
|
|
|
|
/// Format string: "'aaaa'", unescaped literal: "aaaa";
|
|
|
|
/// Format string: "'aa''aa'", unescaped literal: "aa'aa";
|
|
|
|
/// Format string: "'aaa''aa" is not valid because of missing of end single quote.
|
|
|
|
Int64 numLiteralChars(const char * cur, const char * end)
|
|
|
|
{
|
|
|
|
bool found = false;
|
|
|
|
Int64 count = 0;
|
|
|
|
while (cur < end)
|
|
|
|
{
|
|
|
|
if (*cur == '\'')
|
|
|
|
{
|
|
|
|
if (cur + 1 < end && *(cur + 1) == '\'')
|
|
|
|
{
|
|
|
|
count += 2;
|
|
|
|
cur += 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
found = true;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
++count;
|
|
|
|
++cur;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return found ? count : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Cast value from integer to string, making sure digits number in result string is no less than total_digits by padding leading '0'.
|
|
|
|
String padValue(UInt32 val, size_t min_digits)
|
|
|
|
{
|
|
|
|
String str = std::to_string(val);
|
|
|
|
auto length = str.size();
|
|
|
|
if (length >= min_digits)
|
|
|
|
return str;
|
|
|
|
|
|
|
|
String paddings(min_digits - length, '0');
|
|
|
|
return str.insert(0, paddings);
|
|
|
|
}
|
|
|
|
|
|
|
|
constexpr std::string_view weekdaysFull[] = {"Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"};
|
|
|
|
|
|
|
|
constexpr std::string_view weekdaysShort[] = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
|
|
|
|
|
|
|
|
constexpr std::string_view monthsFull[]
|
|
|
|
= {"January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"};
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
constexpr std::string_view monthsShort[]
|
|
|
|
= {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
|
|
|
|
|
|
|
|
/** formatDateTime(time, 'format')
|
|
|
|
* Performs formatting of time, according to provided format.
|
2018-09-26 00:31:40 +00:00
|
|
|
*
|
|
|
|
* This function is optimized with an assumption, that the resulting strings are fixed width.
|
|
|
|
* (This assumption is fulfilled for currently supported formatting options).
|
|
|
|
*
|
|
|
|
* It is implemented in two steps.
|
2022-12-14 06:28:43 +00:00
|
|
|
* At first step, it creates a template of zeros, literal characters, whitespaces, etc.
|
|
|
|
* and quickly fills resulting character array (string column) with this format.
|
2020-08-08 00:47:03 +00:00
|
|
|
* At second step, it walks across the resulting character array and modifies/replaces specific characters,
|
2018-09-26 00:31:40 +00:00
|
|
|
* by calling some functions by pointers and shifting cursor by specified amount.
|
|
|
|
*
|
|
|
|
* Advantages:
|
|
|
|
* - memcpy is mostly unrolled;
|
2022-12-14 06:28:43 +00:00
|
|
|
* - low number of arithmetic ops due to pre-filled template;
|
2018-09-26 00:31:40 +00:00
|
|
|
* - for somewhat reason, function by pointer call is faster than switch/case.
|
|
|
|
*
|
|
|
|
* Possible further optimization options:
|
|
|
|
* - slightly interleave first and second step for better cache locality
|
|
|
|
* (but it has no sense when character array fits in L1d cache);
|
|
|
|
* - avoid indirect function calls and inline functions with JIT compilation.
|
|
|
|
*
|
|
|
|
* Performance on Intel(R) Core(TM) i7-6700 CPU @ 3.40GHz:
|
|
|
|
*
|
|
|
|
* WITH formatDateTime(now() + number, '%H:%M:%S') AS x SELECT count() FROM system.numbers WHERE NOT ignore(x);
|
|
|
|
* - 97 million rows per second per core;
|
|
|
|
*
|
|
|
|
* WITH formatDateTime(toDateTime('2018-01-01 00:00:00') + number, '%F %T') AS x SELECT count() FROM system.numbers WHERE NOT ignore(x)
|
|
|
|
* - 71 million rows per second per core;
|
|
|
|
*
|
|
|
|
* select count() from (select formatDateTime(t, '%m/%d/%Y %H:%M:%S') from (select toDateTime('2018-01-01 00:00:00')+number as t from numbers(100000000)));
|
|
|
|
* - 53 million rows per second per core;
|
|
|
|
*
|
|
|
|
* select count() from (select formatDateTime(t, 'Hello %Y World') from (select toDateTime('2018-01-01 00:00:00')+number as t from numbers(100000000)));
|
|
|
|
* - 138 million rows per second per core;
|
|
|
|
*
|
|
|
|
* PS. We can make this function to return FixedString. Currently it returns String.
|
|
|
|
*/
|
2022-12-14 06:28:43 +00:00
|
|
|
template <typename Name, FormatDateTimeTraits::SupportInteger support_integer, FormatDateTimeTraits::FormatSyntax format_syntax>
|
2020-07-14 06:04:37 +00:00
|
|
|
class FunctionFormatDateTimeImpl : public IFunction
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
private:
|
|
|
|
/// Time is either UInt32 for DateTime or UInt16 for Date.
|
2020-07-14 06:04:37 +00:00
|
|
|
template <typename F>
|
|
|
|
static bool castType(const IDataType * type, F && f)
|
|
|
|
{
|
|
|
|
return castTypeToEither<
|
|
|
|
DataTypeInt8,
|
|
|
|
DataTypeUInt8,
|
|
|
|
DataTypeInt16,
|
|
|
|
DataTypeUInt16,
|
|
|
|
DataTypeInt32,
|
|
|
|
DataTypeUInt32,
|
|
|
|
DataTypeInt64,
|
|
|
|
DataTypeUInt64>(type, std::forward<F>(f));
|
|
|
|
}
|
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
template <typename Time>
|
|
|
|
class Action
|
|
|
|
{
|
|
|
|
public:
|
2022-12-14 06:28:43 +00:00
|
|
|
/// Using std::function will cause performance degradation in MySQL format by 0.45x.
|
|
|
|
/// But std::function is required for Joda format to capture extra variables.
|
|
|
|
/// This is the reason why we use raw function pointer in MySQL format and std::function
|
|
|
|
/// in Joda format.
|
|
|
|
using Func = std::conditional_t<
|
|
|
|
format_syntax == FormatDateTimeTraits::FormatSyntax::MySQL,
|
2022-12-28 06:44:10 +00:00
|
|
|
size_t (*)(char *, Time, UInt64, UInt32, const DateLUTImpl &),
|
|
|
|
std::function<size_t(char *, Time, UInt64, UInt32, const DateLUTImpl &)>>;
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
Func func;
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
/// extra_shift is only used in MySQL format syntax. It is always 0 in Joda format syntax.
|
|
|
|
size_t extra_shift = 0;
|
|
|
|
|
|
|
|
/// Action for appending date/time related number in specified format.
|
|
|
|
explicit Action(Func && func_) : func(std::move(func_)) {}
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
void perform(char *& dest, Time source, UInt64 fractional_second, UInt32 scale, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-28 06:44:10 +00:00
|
|
|
auto shift = func(dest, source, fractional_second, scale, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
dest += shift + extra_shift;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
template <typename T>
|
2022-12-14 06:28:43 +00:00
|
|
|
static size_t writeNumber2(char * p, T v)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2019-12-03 01:22:25 +00:00
|
|
|
memcpy(p, &digits100[v * 2], 2);
|
2022-12-14 06:28:43 +00:00
|
|
|
return 2;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T>
|
2022-12-14 06:28:43 +00:00
|
|
|
static size_t writeNumber3(char * p, T v)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
writeNumber2(p, v / 10);
|
2022-12-14 06:28:43 +00:00
|
|
|
p[2] = '0' + v % 10;
|
|
|
|
return 3;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T>
|
2022-12-14 06:28:43 +00:00
|
|
|
static size_t writeNumber4(char * p, T v)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
writeNumber2(p, v / 100);
|
|
|
|
writeNumber2(p + 2, v % 100);
|
2022-12-14 06:28:43 +00:00
|
|
|
return 4;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
/// Cast content from integer to string, and append result string to buffer.
|
|
|
|
/// Make sure digits number in result string is no less than total_digits by padding leading '0'
|
|
|
|
/// Notice: '-' is not counted as digit.
|
|
|
|
/// For example:
|
|
|
|
/// val = -123, total_digits = 2 => dest = "-123"
|
|
|
|
/// val = -123, total_digits = 3 => dest = "-123"
|
|
|
|
/// val = -123, total_digits = 4 => dest = "-0123"
|
|
|
|
static size_t writeNumberWithPadding(char * dest, std::integral auto val, size_t min_digits)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
using T = decltype(val);
|
|
|
|
using WeightType = typename NumberTraits::Construct<is_signed_v<T>, /*is_floating*/ false, sizeof(T)>::Type;
|
|
|
|
WeightType w = 1;
|
|
|
|
WeightType n = val;
|
|
|
|
size_t digits = 0;
|
|
|
|
while (n)
|
|
|
|
{
|
|
|
|
w *= 10;
|
|
|
|
n /= 10;
|
|
|
|
++digits;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Possible sign
|
|
|
|
size_t pos = 0;
|
|
|
|
n = val;
|
|
|
|
if constexpr (is_signed_v<T>)
|
|
|
|
if (val < 0)
|
|
|
|
{
|
|
|
|
n = (~n) + 1;
|
|
|
|
dest[pos] = '-';
|
|
|
|
++pos;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Possible leading paddings
|
|
|
|
if (min_digits > digits)
|
|
|
|
{
|
|
|
|
memset(dest, '0', min_digits - digits);
|
|
|
|
pos += min_digits - digits;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Digits
|
2022-12-21 04:20:08 +00:00
|
|
|
while (w >= 100)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
w /= 100;
|
|
|
|
|
|
|
|
writeNumber2(dest + pos, n / w);
|
|
|
|
pos += 2;
|
|
|
|
|
|
|
|
n = n % w;
|
|
|
|
}
|
|
|
|
if (n)
|
|
|
|
{
|
|
|
|
dest[pos] = '0' + n;
|
|
|
|
++pos;
|
|
|
|
}
|
|
|
|
|
|
|
|
return pos;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
2022-12-14 06:28:43 +00:00
|
|
|
public:
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlNoop(char *, Time, UInt64, UInt32, const DateLUTImpl &) { return 0; }
|
2022-12-14 06:28:43 +00:00
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlCentury(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
auto year = ToYearImpl::execute(source, timezone);
|
|
|
|
auto century = year / 100;
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, century);
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlDayOfMonth(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToDayOfMonthImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlAmericanDate(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber2(dest, ToMonthImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 3, ToDayOfMonthImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 6, ToYearImpl::execute(source, timezone) % 100);
|
|
|
|
return 8;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlDayOfMonthSpacePadded(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
auto day = ToDayOfMonthImpl::execute(source, timezone);
|
|
|
|
if (day < 10)
|
2022-12-14 06:28:43 +00:00
|
|
|
dest[1] = '0' + day;
|
2018-09-26 00:31:40 +00:00
|
|
|
else
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber2(dest, day);
|
|
|
|
return 2;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlISO8601Date(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber4(dest, ToYearImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 5, ToMonthImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 8, ToDayOfMonthImpl::execute(source, timezone));
|
|
|
|
return 10;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlDayOfYear(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber3(dest, ToDayOfYearImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlMonth(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToMonthImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlDayOfWeek(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2023-01-13 14:00:16 +00:00
|
|
|
*dest = '0' + ToDayOfWeekImpl::execute(source, 0, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
return 1;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlDayOfWeek0To6(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2023-01-13 14:00:16 +00:00
|
|
|
auto day = ToDayOfWeekImpl::execute(source, 0, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
*dest = '0' + (day == 7 ? 0 : day);
|
|
|
|
return 1;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlISO8601Week(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToISOWeekImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlISO8601Year2(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2020-07-04 18:54:42 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToISOYearImpl::execute(source, timezone) % 100);
|
2020-07-04 18:54:42 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlISO8601Year4(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2020-07-04 18:54:42 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber4(dest, ToISOYearImpl::execute(source, timezone));
|
2020-07-04 18:54:42 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlYear2(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToYearImpl::execute(source, timezone) % 100);
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlYear4(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber4(dest, ToYearImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlHour24(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToHourImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlHour12(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
auto x = ToHourImpl::execute(source, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, x == 0 ? 12 : (x > 12 ? x - 12 : x));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlMinute(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToMinuteImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlAMPM(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
auto hour = ToHourImpl::execute(source, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
dest[0] = hour >= 12 ? 'P' : 'A';
|
|
|
|
dest[1] = 'M';
|
|
|
|
return 2;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlHHMM24(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber2(dest, ToHourImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 3, ToMinuteImpl::execute(source, timezone));
|
|
|
|
return 5;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlSecond(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumber2(dest, ToSecondImpl::execute(source, timezone));
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t
|
|
|
|
mysqlFractionalSecond(char * dest, Time /*source*/, UInt64 fractional_second, UInt32 scale, const DateLUTImpl & /*timezone*/)
|
|
|
|
{
|
|
|
|
if (scale == 0)
|
|
|
|
scale = 1;
|
|
|
|
|
|
|
|
for (Int64 i = scale, value = fractional_second; i > 0; --i)
|
|
|
|
{
|
|
|
|
dest[i - 1] += value % 10;
|
|
|
|
value /= 10;
|
|
|
|
}
|
|
|
|
return scale;
|
|
|
|
}
|
2022-12-21 20:00:39 +00:00
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlISO8601Time(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) // NOLINT
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber2(dest, ToHourImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 3, ToMinuteImpl::execute(source, timezone));
|
|
|
|
writeNumber2(dest + 6, ToSecondImpl::execute(source, timezone));
|
|
|
|
return 8;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
2021-01-18 04:16:32 +00:00
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlTimezoneOffset(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-08-29 02:28:11 +00:00
|
|
|
{
|
|
|
|
auto offset = TimezoneOffsetImpl::execute(source, timezone);
|
|
|
|
if (offset < 0)
|
2022-08-30 00:40:11 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
*dest = '-';
|
2022-08-30 00:40:11 +00:00
|
|
|
offset = -offset;
|
|
|
|
}
|
2022-08-29 02:28:11 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
writeNumber2(dest + 1, offset / 3600);
|
|
|
|
writeNumber2(dest + 3, offset % 3600 / 60);
|
|
|
|
return 5;
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t mysqlQuarter(char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
*dest = '0' + ToQuarterImpl::execute(source, timezone);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Literal>
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaLiteral(const Literal & literal, char * dest, Time, UInt64, UInt32, const DateLUTImpl &)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
memcpy(dest, literal.data(), literal.size());
|
|
|
|
return literal.size();
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaEra(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto year = static_cast<Int32>(ToYearImpl::execute(source, timezone));
|
|
|
|
String res;
|
|
|
|
if (min_represent_digits <= 3)
|
|
|
|
res = static_cast<Int32>(year) > 0 ? "AD" : "BC";
|
|
|
|
else
|
|
|
|
res = static_cast<Int32>(year) > 0 ? "Anno Domini" : "Before Christ";
|
|
|
|
|
|
|
|
memcpy(dest, res.data(), res.size());
|
|
|
|
return res.size();
|
|
|
|
}
|
|
|
|
|
2023-01-29 20:01:30 +00:00
|
|
|
static size_t jodaCenturyOfEra(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto year = static_cast<Int32>(ToYearImpl::execute(source, timezone));
|
|
|
|
year = (year < 0 ? -year : year);
|
|
|
|
return writeNumberWithPadding(dest, year / 100, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaYearOfEra(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto year = static_cast<Int32>(ToYearImpl::execute(source, timezone));
|
|
|
|
if (min_represent_digits == 2)
|
|
|
|
return writeNumberWithPadding(dest, std::abs(year) % 100, 2);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
year = year <= 0 ? std::abs(year - 1) : year;
|
|
|
|
return writeNumberWithPadding(dest, year, min_represent_digits);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaDayOfWeek1Based(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
2023-01-13 14:00:16 +00:00
|
|
|
auto week_day = ToDayOfWeekImpl::execute(source, 0, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
return writeNumberWithPadding(dest, week_day, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaDayOfWeekText(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
2023-01-13 14:00:16 +00:00
|
|
|
auto week_day = ToDayOfWeekImpl::execute(source, 0, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
if (week_day == 7)
|
|
|
|
week_day = 0;
|
|
|
|
|
|
|
|
std::string_view str_view = min_represent_digits <= 3 ? weekdaysShort[week_day] : weekdaysFull[week_day];
|
|
|
|
memcpy(dest, str_view.data(), str_view.size());
|
|
|
|
return str_view.size();
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaYear(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto year = static_cast<Int32>(ToYearImpl::execute(source, timezone));
|
|
|
|
if (min_represent_digits == 2)
|
|
|
|
{
|
|
|
|
year = std::abs(year);
|
|
|
|
auto two_digit_year = year % 100;
|
|
|
|
return writeNumberWithPadding(dest, two_digit_year, 2);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
return writeNumberWithPadding(dest, year, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2023-01-31 02:20:59 +00:00
|
|
|
static size_t jodaWeekYear(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) {
|
|
|
|
auto week_year = ToWeekYearImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, week_year, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
|
|
|
static size_t jodaWeekOfWeekYear(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone) {
|
|
|
|
auto week_of_weekyear = ToWeekOfWeekYearImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, week_of_weekyear, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaDayOfYear(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto day_of_year = ToDayOfYearImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, day_of_year, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaMonthOfYear(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto month_of_year = ToMonthImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, month_of_year, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaMonthOfYearText(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto month = ToMonthImpl::execute(source, timezone);
|
|
|
|
std::string_view str_view = min_represent_digits <= 3 ? monthsShort[month - 1] : monthsFull[month - 1];
|
|
|
|
memcpy(dest, str_view.data(), str_view.size());
|
|
|
|
return str_view.size();
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaDayOfMonth(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto day_of_month = ToDayOfMonthImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, day_of_month, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaHalfDayOfDay(
|
|
|
|
size_t /*min_represent_digits*/, char * dest, Time source, UInt64 fractional_second, UInt32 scale, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
2022-12-28 06:44:10 +00:00
|
|
|
return mysqlAMPM(dest, source, fractional_second, scale, timezone);
|
2022-12-14 06:28:43 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaHourOfHalfDay(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto hour = ToHourImpl::execute(source, timezone) % 12;
|
|
|
|
return writeNumberWithPadding(dest, hour, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaClockHourOfHalfDay(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto hour = ToHourImpl::execute(source, timezone) ;
|
|
|
|
hour = (hour + 11) % 12 + 1;
|
|
|
|
return writeNumberWithPadding(dest, hour, min_represent_digits);
|
2022-08-29 02:28:11 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaHourOfDay(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2021-01-18 04:16:32 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
auto hour = ToHourImpl::execute(source, timezone) ;
|
|
|
|
return writeNumberWithPadding(dest, hour, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaClockHourOfDay(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto hour = ToHourImpl::execute(source, timezone);
|
|
|
|
hour = (hour + 23) % 24 + 1;
|
|
|
|
return writeNumberWithPadding(dest, hour, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaMinuteOfHour(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto minute_of_hour = ToMinuteImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, minute_of_hour, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaSecondOfMinute(size_t min_represent_digits, char * dest, Time source, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto second_of_minute = ToSecondImpl::execute(source, timezone);
|
|
|
|
return writeNumberWithPadding(dest, second_of_minute, min_represent_digits);
|
|
|
|
}
|
|
|
|
|
2023-01-31 02:20:59 +00:00
|
|
|
static size_t jodaFractionOfSecond(size_t min_represent_digits, char * dest, Time /*source*/, UInt64 fractional_second, UInt32 scale, const DateLUTImpl & /*timezone*/)
|
|
|
|
{
|
|
|
|
if (fractional_second == 0)
|
|
|
|
{
|
|
|
|
for (UInt64 i = 0; i < min_represent_digits; ++i)
|
|
|
|
dest[i] = '0';
|
|
|
|
return min_represent_digits;
|
|
|
|
}
|
|
|
|
auto str = toString(fractional_second);
|
|
|
|
if (min_represent_digits > scale)
|
|
|
|
{
|
|
|
|
for (UInt64 i = 0; i < min_represent_digits - scale; ++i)
|
|
|
|
str += '0';
|
|
|
|
}
|
|
|
|
else if (min_represent_digits < scale)
|
|
|
|
{
|
|
|
|
str = str.substr(0, min_represent_digits);
|
|
|
|
}
|
|
|
|
memcpy(dest, str.data(), str.size());
|
|
|
|
return min_represent_digits;
|
|
|
|
}
|
|
|
|
|
2022-12-28 06:44:10 +00:00
|
|
|
static size_t jodaTimezone(size_t min_represent_digits, char * dest, Time /*source*/, UInt64, UInt32, const DateLUTImpl & timezone)
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
if (min_represent_digits <= 3)
|
|
|
|
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "Short name time zone is not yet supported");
|
|
|
|
|
|
|
|
auto str = timezone.getTimeZone();
|
|
|
|
memcpy(dest, str.data(), str.size());
|
|
|
|
return str.size();
|
2021-01-18 04:16:32 +00:00
|
|
|
}
|
2018-09-26 00:31:40 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
public:
|
2020-07-14 06:04:37 +00:00
|
|
|
static constexpr auto name = Name::name;
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2021-06-01 12:20:52 +00:00
|
|
|
static FunctionPtr create(ContextPtr) { return std::make_shared<FunctionFormatDateTimeImpl>(); }
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
String getName() const override
|
|
|
|
{
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool useDefaultImplementationForConstants() const override { return true; }
|
|
|
|
|
2021-06-22 16:21:23 +00:00
|
|
|
bool isSuitableForShortCircuitArgumentsExecution(const DataTypesWithConstInfo & /*arguments*/) const override { return false; }
|
2021-04-29 14:48:26 +00:00
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
ColumnNumbers getArgumentsThatAreAlwaysConstant() const override { return {1, 2}; }
|
|
|
|
|
|
|
|
bool isVariadic() const override { return true; }
|
|
|
|
size_t getNumberOfArguments() const override { return 0; }
|
|
|
|
|
|
|
|
DataTypePtr getReturnTypeImpl(const ColumnsWithTypeAndName & arguments) const override
|
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
if constexpr (support_integer == FormatDateTimeTraits::SupportInteger::Yes)
|
2020-07-23 10:29:59 +00:00
|
|
|
{
|
|
|
|
if (arguments.size() != 1 && arguments.size() != 2 && arguments.size() != 3)
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH,
|
|
|
|
"Number of arguments for function {} doesn't match: passed {}, should be 1, 2 or 3",
|
|
|
|
getName(), arguments.size());
|
2020-07-23 10:29:59 +00:00
|
|
|
if (arguments.size() == 1 && !isInteger(arguments[0].type))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
|
2022-10-28 12:44:47 +00:00
|
|
|
"Illegal type {} of first argument of function {} when arguments size is 1. Should be integer",
|
2022-10-26 16:07:56 +00:00
|
|
|
arguments[0].type->getName(), getName());
|
2022-10-26 08:00:12 +00:00
|
|
|
if (arguments.size() > 1 && !(isInteger(arguments[0].type) || isDate(arguments[0].type) || isDateTime(arguments[0].type) || isDate32(arguments[0].type) || isDateTime64(arguments[0].type)))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
|
2023-01-23 21:13:58 +00:00
|
|
|
"Illegal type {} of first argument of function {} when arguments size is 2 or 3. "
|
|
|
|
"Should be a integer or a date with time",
|
|
|
|
arguments[0].type->getName(), getName());
|
2020-07-23 10:29:59 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (arguments.size() != 2 && arguments.size() != 3)
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH,
|
|
|
|
"Number of arguments for function {} doesn't match: passed {}, should be 2 or 3",
|
|
|
|
getName(), arguments.size());
|
2022-10-26 08:00:12 +00:00
|
|
|
if (!isDate(arguments[0].type) && !isDateTime(arguments[0].type) && !isDate32(arguments[0].type) && !isDateTime64(arguments[0].type))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
|
2022-10-28 12:44:47 +00:00
|
|
|
"Illegal type {} of first argument of function {}. Should be a date or a date with time",
|
2022-10-26 16:07:56 +00:00
|
|
|
arguments[0].type->getName(), getName());
|
2020-07-23 10:29:59 +00:00
|
|
|
}
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2020-07-23 10:29:59 +00:00
|
|
|
if (arguments.size() == 2 && !WhichDataType(arguments[1].type).isString())
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
|
2022-10-28 12:44:47 +00:00
|
|
|
"Illegal type {} of second argument of function {}. Must be String.",
|
2022-10-26 16:07:56 +00:00
|
|
|
arguments[1].type->getName(), getName());
|
2020-07-23 10:29:59 +00:00
|
|
|
|
|
|
|
if (arguments.size() == 3 && !WhichDataType(arguments[2].type).isString())
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
|
2022-10-28 12:44:47 +00:00
|
|
|
"Illegal type {} of third argument of function {}. Must be String.",
|
2022-10-26 16:07:56 +00:00
|
|
|
arguments[2].type->getName(), getName());
|
2020-07-23 10:29:59 +00:00
|
|
|
|
|
|
|
if (arguments.size() == 1)
|
|
|
|
return std::make_shared<DataTypeDateTime>();
|
|
|
|
return std::make_shared<DataTypeString>();
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2020-11-17 13:24:45 +00:00
|
|
|
ColumnPtr executeImpl(const ColumnsWithTypeAndName & arguments, const DataTypePtr & result_type, [[maybe_unused]] size_t input_rows_count) const override
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2020-10-17 21:41:50 +00:00
|
|
|
ColumnPtr res;
|
2022-12-14 06:28:43 +00:00
|
|
|
if constexpr (support_integer == FormatDateTimeTraits::SupportInteger::Yes)
|
2020-07-14 06:04:37 +00:00
|
|
|
{
|
|
|
|
if (arguments.size() == 1)
|
|
|
|
{
|
2020-10-17 21:41:50 +00:00
|
|
|
if (!castType(arguments[0].type.get(), [&](const auto & type)
|
2020-07-14 06:04:55 +00:00
|
|
|
{
|
2020-07-14 06:04:37 +00:00
|
|
|
using FromDataType = std::decay_t<decltype(type)>;
|
2020-10-17 21:41:50 +00:00
|
|
|
res = ConvertImpl<FromDataType, DataTypeDateTime, Name>::execute(arguments, result_type, input_rows_count);
|
2020-07-14 06:04:37 +00:00
|
|
|
return true;
|
|
|
|
}))
|
|
|
|
{
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_COLUMN,
|
2023-01-23 21:13:58 +00:00
|
|
|
"Illegal column {} of function {}, must be Integer, Date, Date32, DateTime "
|
|
|
|
"or DateTime64 when arguments size is 1.",
|
|
|
|
arguments[0].column->getName(), getName());
|
2020-07-14 06:04:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2020-10-17 21:41:50 +00:00
|
|
|
if (!castType(arguments[0].type.get(), [&](const auto & type)
|
2020-07-14 06:04:55 +00:00
|
|
|
{
|
2020-07-14 06:04:37 +00:00
|
|
|
using FromDataType = std::decay_t<decltype(type)>;
|
2020-10-17 21:41:50 +00:00
|
|
|
if (!(res = executeType<FromDataType>(arguments, result_type)))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_COLUMN,
|
2022-10-28 13:11:09 +00:00
|
|
|
"Illegal column {} of function {}, must be Integer, Date, Date32, DateTime or DateTime64.",
|
2022-10-26 16:07:56 +00:00
|
|
|
arguments[0].column->getName(), getName());
|
2020-07-14 06:04:37 +00:00
|
|
|
return true;
|
|
|
|
}))
|
|
|
|
{
|
2020-10-20 13:11:57 +00:00
|
|
|
if (!((res = executeType<DataTypeDate>(arguments, result_type))
|
2022-10-26 08:00:12 +00:00
|
|
|
|| (res = executeType<DataTypeDate32>(arguments, result_type))
|
2020-10-20 13:11:57 +00:00
|
|
|
|| (res = executeType<DataTypeDateTime>(arguments, result_type))
|
|
|
|
|| (res = executeType<DataTypeDateTime64>(arguments, result_type))))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_COLUMN,
|
|
|
|
"Illegal column {} of function {}, must be Integer or DateTime.",
|
|
|
|
arguments[0].column->getName(), getName());
|
2020-07-14 06:04:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2020-10-20 13:11:57 +00:00
|
|
|
if (!((res = executeType<DataTypeDate>(arguments, result_type))
|
2022-10-26 08:00:12 +00:00
|
|
|
|| (res = executeType<DataTypeDate32>(arguments, result_type))
|
2020-10-20 13:11:57 +00:00
|
|
|
|| (res = executeType<DataTypeDateTime>(arguments, result_type))
|
|
|
|
|| (res = executeType<DataTypeDateTime64>(arguments, result_type))))
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_COLUMN,
|
|
|
|
"Illegal column {} of function {}, must be Date or DateTime.",
|
|
|
|
arguments[0].column->getName(), getName());
|
2020-07-14 06:04:37 +00:00
|
|
|
}
|
2020-10-17 21:41:50 +00:00
|
|
|
|
|
|
|
return res;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
template <typename DataType>
|
2020-11-17 13:24:45 +00:00
|
|
|
ColumnPtr executeType(const ColumnsWithTypeAndName & arguments, const DataTypePtr &) const
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2020-10-17 21:41:50 +00:00
|
|
|
auto * times = checkAndGetColumn<typename DataType::ColumnType>(arguments[0].column.get());
|
2019-11-06 12:01:39 +00:00
|
|
|
if (!times)
|
2020-10-17 21:41:50 +00:00
|
|
|
return nullptr;
|
2019-11-06 12:01:39 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
const ColumnConst * format_column = checkAndGetColumnConst<ColumnString>(arguments[1].column.get());
|
|
|
|
if (!format_column)
|
2022-10-26 16:07:56 +00:00
|
|
|
throw Exception(ErrorCodes::ILLEGAL_COLUMN,
|
|
|
|
"Illegal column {} of second ('format') argument of function {}. Must be constant string.",
|
|
|
|
arguments[1].column->getName(), getName());
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
String format = format_column->getValue<String>();
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-21 20:00:39 +00:00
|
|
|
UInt32 scale [[maybe_unused]] = 0;
|
2022-12-08 22:53:16 +00:00
|
|
|
if constexpr (std::is_same_v<DataType, DataTypeDateTime64>)
|
|
|
|
scale = times->getScale();
|
|
|
|
|
2020-07-23 10:23:55 +00:00
|
|
|
using T = typename ActionValueTypeMap<DataType>::ActionValueType;
|
2019-11-06 12:01:39 +00:00
|
|
|
std::vector<Action<T>> instructions;
|
2022-12-14 06:28:43 +00:00
|
|
|
String out_template;
|
2022-12-28 06:44:10 +00:00
|
|
|
auto result_size = parseFormat(format, instructions, scale, out_template);
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
const DateLUTImpl * time_zone_tmp = nullptr;
|
2020-10-17 21:41:50 +00:00
|
|
|
if (castType(arguments[0].type.get(), [&]([[maybe_unused]] const auto & type) { return true; }))
|
|
|
|
time_zone_tmp = &extractTimeZoneFromFunctionArguments(arguments, 2, 0);
|
2020-07-14 06:04:37 +00:00
|
|
|
else if (std::is_same_v<DataType, DataTypeDateTime64> || std::is_same_v<DataType, DataTypeDateTime>)
|
2020-10-20 13:11:57 +00:00
|
|
|
time_zone_tmp = &extractTimeZoneFromFunctionArguments(arguments, 2, 0);
|
2019-11-06 12:01:39 +00:00
|
|
|
else
|
|
|
|
time_zone_tmp = &DateLUT::instance();
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
const DateLUTImpl & time_zone = *time_zone_tmp;
|
|
|
|
const auto & vec = times->getData();
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
auto col_res = ColumnString::create();
|
|
|
|
auto & dst_data = col_res->getChars();
|
|
|
|
auto & dst_offsets = col_res->getOffsets();
|
|
|
|
dst_data.resize(vec.size() * (result_size + 1));
|
|
|
|
dst_offsets.resize(vec.size());
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
if constexpr (format_syntax == FormatDateTimeTraits::FormatSyntax::MySQL)
|
2019-11-06 12:01:39 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
/// Fill result with literals.
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
UInt8 * begin = dst_data.data();
|
|
|
|
UInt8 * end = begin + dst_data.size();
|
|
|
|
UInt8 * pos = begin;
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
if (pos < end)
|
|
|
|
{
|
|
|
|
memcpy(pos, out_template.data(), result_size + 1); /// With zero terminator.
|
|
|
|
pos += result_size + 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Fill by copying exponential growing ranges.
|
|
|
|
while (pos < end)
|
|
|
|
{
|
|
|
|
size_t bytes_to_copy = std::min(pos - begin, end - pos);
|
|
|
|
memcpy(pos, begin, bytes_to_copy);
|
|
|
|
pos += bytes_to_copy;
|
|
|
|
}
|
2019-11-06 12:01:39 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-22 08:31:10 +00:00
|
|
|
auto * begin = reinterpret_cast<char *>(dst_data.data());
|
|
|
|
auto * pos = begin;
|
2019-11-06 12:01:39 +00:00
|
|
|
for (size_t i = 0; i < vec.size(); ++i)
|
|
|
|
{
|
|
|
|
if constexpr (std::is_same_v<DataType, DataTypeDateTime64>)
|
|
|
|
{
|
2022-12-21 20:00:39 +00:00
|
|
|
const auto c = DecimalUtils::split(vec[i], scale);
|
2019-11-06 12:01:39 +00:00
|
|
|
for (auto & instruction : instructions)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-21 20:00:39 +00:00
|
|
|
instruction.perform(pos, static_cast<Int64>(c.whole), c.fractional, scale, time_zone);
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
}
|
2019-11-06 12:01:39 +00:00
|
|
|
else
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2018-11-23 18:52:00 +00:00
|
|
|
for (auto & instruction : instructions)
|
2022-12-21 20:00:39 +00:00
|
|
|
instruction.perform(pos, static_cast<UInt32>(vec[i]), 0, 0, time_zone);
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
2022-12-14 06:28:43 +00:00
|
|
|
*pos++ = '\0';
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
dst_offsets[i] = pos - begin;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
2019-11-06 12:01:39 +00:00
|
|
|
dst_data.resize(pos - begin);
|
2020-10-17 21:41:50 +00:00
|
|
|
return col_res;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T>
|
2022-12-28 06:44:10 +00:00
|
|
|
size_t parseFormat(const String & format, std::vector<Action<T>> & instructions, UInt32 scale, String & out_template) const
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
if constexpr (format_syntax == FormatDateTimeTraits::FormatSyntax::MySQL)
|
2022-12-28 06:44:10 +00:00
|
|
|
return parseMySQLFormat(format, instructions, scale, out_template);
|
2022-12-14 06:28:43 +00:00
|
|
|
else if constexpr (format_syntax == FormatDateTimeTraits::FormatSyntax::Joda)
|
2022-12-28 06:44:10 +00:00
|
|
|
return parseJodaFormat(format, instructions, scale, out_template);
|
2022-12-14 06:28:43 +00:00
|
|
|
else
|
|
|
|
throw Exception(
|
|
|
|
ErrorCodes::NOT_IMPLEMENTED,
|
|
|
|
"Unknown datetime format style {} in function {}",
|
|
|
|
magic_enum::enum_name(format_syntax),
|
|
|
|
getName());
|
|
|
|
}
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
template <typename T>
|
2022-12-28 06:44:10 +00:00
|
|
|
size_t parseMySQLFormat(const String & format, std::vector<Action<T>> & instructions, UInt32 scale, String & out_template) const
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
auto add_extra_shift = [&](size_t amount)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
if (instructions.empty())
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlNoop);
|
|
|
|
instructions.back().extra_shift += amount;
|
2018-09-26 00:31:40 +00:00
|
|
|
};
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
auto add_instruction_or_extra_shift = [&](auto * func [[maybe_unused]], size_t amount [[maybe_unused]])
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
if constexpr (std::is_same_v<T, UInt32> || std::is_same_v<T, Int64>)
|
|
|
|
instructions.emplace_back(std::move(func));
|
2018-09-26 00:31:40 +00:00
|
|
|
else
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(amount);
|
2018-09-26 00:31:40 +00:00
|
|
|
};
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
const char * pos = format.data();
|
|
|
|
const char * const end = pos + format.size();
|
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
const char * percent_pos = find_first_symbols<'%'>(pos, end);
|
|
|
|
|
|
|
|
if (percent_pos < end)
|
|
|
|
{
|
|
|
|
if (pos < percent_pos)
|
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(percent_pos - pos);
|
|
|
|
out_template += String(pos, percent_pos - pos);
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pos = percent_pos + 1;
|
|
|
|
if (pos >= end)
|
2023-01-23 21:13:58 +00:00
|
|
|
throw Exception(ErrorCodes::BAD_ARGUMENTS, "Sign '%' is the last in format, if you need it, use '%%'");
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
switch (*pos)
|
|
|
|
{
|
|
|
|
// Year, divided by 100, zero-padded
|
|
|
|
case 'C':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlCentury);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Day of month, zero-padded (01-31)
|
|
|
|
case 'd':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlDayOfMonth);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Short MM/DD/YY date, equivalent to %m/%d/%y
|
|
|
|
case 'D':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlAmericanDate);
|
|
|
|
out_template += "00/00/00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Day of month, space-padded ( 1-31) 23
|
|
|
|
case 'e':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlDayOfMonthSpacePadded);
|
|
|
|
out_template += " 0";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
2022-12-08 22:53:16 +00:00
|
|
|
// Fractional seconds
|
|
|
|
case 'f':
|
|
|
|
{
|
2022-12-21 20:00:39 +00:00
|
|
|
/// If the time data type has no fractional part, then we print '0' as the fractional part.
|
2022-12-28 06:44:10 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlFractionalSecond);
|
|
|
|
out_template += String(std::max<UInt32>(1, scale), '0');
|
2022-12-08 22:53:16 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
// Short YYYY-MM-DD date, equivalent to %Y-%m-%d 2001-08-23
|
|
|
|
case 'F':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlISO8601Date);
|
|
|
|
out_template += "0000-00-00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
2020-07-04 18:54:42 +00:00
|
|
|
// Last two digits of year of ISO 8601 week number (see %G)
|
|
|
|
case 'g':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlISO8601Year2);
|
|
|
|
out_template += "00";
|
|
|
|
break;
|
2020-07-04 18:54:42 +00:00
|
|
|
|
|
|
|
// Year of ISO 8601 week number (see %V)
|
|
|
|
case 'G':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlISO8601Year4);
|
|
|
|
out_template += "0000";
|
|
|
|
break;
|
2020-07-04 18:54:42 +00:00
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
// Day of the year (001-366) 235
|
|
|
|
case 'j':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlDayOfYear);
|
|
|
|
out_template += "000";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Month as a decimal number (01-12)
|
|
|
|
case 'm':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlMonth);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// ISO 8601 weekday as number with Monday as 1 (1-7)
|
|
|
|
case 'u':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlDayOfWeek);
|
|
|
|
out_template += "0";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// ISO 8601 week number (01-53)
|
|
|
|
case 'V':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlISO8601Week);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Weekday as a decimal number with Sunday as 0 (0-6) 4
|
|
|
|
case 'w':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlDayOfWeek0To6);
|
|
|
|
out_template += "0";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Two digits year
|
|
|
|
case 'y':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlYear2);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Four digits year
|
|
|
|
case 'Y':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlYear4);
|
|
|
|
out_template += "0000";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
2021-01-18 04:16:32 +00:00
|
|
|
// Quarter (1-4)
|
|
|
|
case 'Q':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.template emplace_back(&Action<T>::mysqlQuarter);
|
|
|
|
out_template += "0";
|
2021-01-18 04:16:32 +00:00
|
|
|
break;
|
|
|
|
|
2022-08-29 02:28:11 +00:00
|
|
|
// Offset from UTC timezone as +hhmm or -hhmm
|
|
|
|
case 'z':
|
2022-12-14 06:28:43 +00:00
|
|
|
instructions.emplace_back(&Action<T>::mysqlTimezoneOffset);
|
|
|
|
out_template += "+0000";
|
2022-08-29 02:28:11 +00:00
|
|
|
break;
|
|
|
|
|
2018-09-26 00:31:40 +00:00
|
|
|
/// Time components. If the argument is Date, not a DateTime, then this components will have default value.
|
|
|
|
|
|
|
|
// Minute (00-59)
|
|
|
|
case 'M':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlMinute, 2);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// AM or PM
|
|
|
|
case 'p':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlAMPM, 2);
|
|
|
|
out_template += "AM";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// 24-hour HH:MM time, equivalent to %H:%M 14:55
|
|
|
|
case 'R':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlHHMM24, 5);
|
|
|
|
out_template += "00:00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Seconds
|
|
|
|
case 'S':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlSecond, 2);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// ISO 8601 time format (HH:MM:SS), equivalent to %H:%M:%S 14:55:02
|
|
|
|
case 'T':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlISO8601Time, 8);
|
|
|
|
out_template += "00:00:00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Hour in 24h format (00-23)
|
|
|
|
case 'H':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlHour24, 2);
|
|
|
|
out_template += "00";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Hour in 12h format (01-12)
|
|
|
|
case 'I':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_instruction_or_extra_shift(&Action<T>::mysqlHour12, 2);
|
|
|
|
out_template += "12";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
/// Escaped literal characters.
|
|
|
|
case '%':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(1);
|
|
|
|
out_template += "%";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
case 't':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(1);
|
|
|
|
out_template += "\t";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
case 'n':
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(1);
|
|
|
|
out_template += "\n";
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
// Unimplemented
|
2022-12-14 06:28:43 +00:00
|
|
|
case 'U':
|
|
|
|
[[fallthrough]];
|
2018-09-26 00:31:40 +00:00
|
|
|
case 'W':
|
2022-12-14 06:28:43 +00:00
|
|
|
throw Exception(
|
|
|
|
ErrorCodes::NOT_IMPLEMENTED,
|
|
|
|
"Wrong syntax '{}', symbol '{}' is not implemented for function {}",
|
|
|
|
format,
|
|
|
|
*pos,
|
|
|
|
getName());
|
2018-09-26 00:31:40 +00:00
|
|
|
|
|
|
|
default:
|
2022-12-14 06:28:43 +00:00
|
|
|
throw Exception(
|
|
|
|
ErrorCodes::ILLEGAL_COLUMN,
|
|
|
|
"Wrong syntax '{}', unexpected symbol '{}' for function {}",
|
|
|
|
format,
|
|
|
|
*pos,
|
|
|
|
getName());
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
++pos;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2022-12-14 06:28:43 +00:00
|
|
|
add_extra_shift(end - pos);
|
|
|
|
out_template += String(pos, end - pos);
|
2018-09-26 00:31:40 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
return out_template.size();
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T>
|
2022-12-28 06:44:10 +00:00
|
|
|
size_t parseJodaFormat(const String & format, std::vector<Action<T>> & instructions, UInt32, String &) const
|
2022-12-14 06:28:43 +00:00
|
|
|
{
|
|
|
|
/// If the argument was DateTime, add instruction for printing. If it was date, just append default literal
|
|
|
|
auto add_instruction = [&](auto && func [[maybe_unused]], const String & default_literal [[maybe_unused]])
|
|
|
|
{
|
|
|
|
if constexpr (std::is_same_v<T, UInt32> || std::is_same_v<T, Int64>)
|
|
|
|
instructions.emplace_back(func);
|
|
|
|
else
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::template jodaLiteral<String>, default_literal));
|
|
|
|
};
|
|
|
|
|
|
|
|
size_t reserve_size = 0;
|
|
|
|
const char * pos = format.data();
|
|
|
|
const char * end = pos + format.size();
|
|
|
|
|
|
|
|
while (pos < end)
|
|
|
|
{
|
|
|
|
const char * cur_token = pos;
|
|
|
|
|
|
|
|
// Literal case
|
|
|
|
if (*cur_token == '\'')
|
|
|
|
{
|
|
|
|
// Case 1: 2 consecutive single quote
|
|
|
|
if (pos + 1 < end && *(pos + 1) == '\'')
|
|
|
|
{
|
|
|
|
std::string_view literal(cur_token, 1);
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::template jodaLiteral<decltype(literal)>, literal));
|
|
|
|
++reserve_size;
|
|
|
|
pos += 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// Case 2: find closing single quote
|
|
|
|
Int64 count = numLiteralChars(cur_token + 1, end);
|
|
|
|
if (count == -1)
|
2023-01-23 21:13:58 +00:00
|
|
|
throw Exception(ErrorCodes::BAD_ARGUMENTS, "No closing single quote for literal");
|
2022-12-14 06:28:43 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
for (Int64 i = 1; i <= count; i++)
|
|
|
|
{
|
|
|
|
std::string_view literal(cur_token + i, 1);
|
|
|
|
instructions.emplace_back(
|
|
|
|
std::bind_front(&Action<T>::template jodaLiteral<decltype(literal)>, literal));
|
|
|
|
++reserve_size;
|
|
|
|
if (*(cur_token + i) == '\'')
|
|
|
|
i += 1;
|
|
|
|
}
|
|
|
|
pos += count + 2;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
int repetitions = 1;
|
|
|
|
++pos;
|
|
|
|
while (pos < end && *cur_token == *pos)
|
|
|
|
{
|
|
|
|
++repetitions;
|
|
|
|
++pos;
|
|
|
|
}
|
|
|
|
switch (*cur_token)
|
|
|
|
{
|
|
|
|
case 'G':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaEra, repetitions));
|
|
|
|
reserve_size += repetitions <= 3 ? 2 : 13;
|
|
|
|
break;
|
|
|
|
case 'C':
|
2023-01-29 20:01:30 +00:00
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaCenturyOfEra, repetitions));
|
2022-12-14 06:28:43 +00:00
|
|
|
/// Year range [1900, 2299]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'Y':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaYearOfEra, repetitions));
|
|
|
|
/// Year range [1900, 2299]
|
|
|
|
reserve_size += repetitions == 2 ? 2 : std::max(repetitions, 4);
|
|
|
|
break;
|
|
|
|
case 'x':
|
2023-01-31 02:20:59 +00:00
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaWeekYear, repetitions));
|
|
|
|
/// weekyear range [1900, 2299]
|
|
|
|
reserve_size += std::max(repetitions, 4);
|
|
|
|
break;
|
2022-12-14 06:28:43 +00:00
|
|
|
case 'w':
|
2023-01-31 02:20:59 +00:00
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaWeekOfWeekYear, repetitions));
|
|
|
|
/// Week of weekyear range [1, 52]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
2022-12-14 06:28:43 +00:00
|
|
|
case 'e':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaDayOfWeek1Based, repetitions));
|
|
|
|
/// Day of week range [1, 7]
|
|
|
|
reserve_size += std::max(repetitions, 1);
|
|
|
|
break;
|
|
|
|
case 'E':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaDayOfWeekText, repetitions));
|
|
|
|
/// Maximum length of short name is 3, maximum length of full name is 9.
|
|
|
|
reserve_size += repetitions <= 3 ? 3 : 9;
|
|
|
|
break;
|
|
|
|
case 'y':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaYear, repetitions));
|
|
|
|
/// Year range [1900, 2299]
|
|
|
|
reserve_size += repetitions == 2 ? 2 : std::max(repetitions, 4);
|
|
|
|
break;
|
|
|
|
case 'D':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaDayOfYear, repetitions));
|
|
|
|
/// Day of year range [1, 366]
|
|
|
|
reserve_size += std::max(repetitions, 3);
|
|
|
|
break;
|
|
|
|
case 'M':
|
|
|
|
if (repetitions <= 2)
|
|
|
|
{
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaMonthOfYear, repetitions));
|
|
|
|
/// Month of year range [1, 12]
|
|
|
|
reserve_size += 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaMonthOfYearText, repetitions));
|
|
|
|
/// Maximum length of short name is 3, maximum length of full name is 9.
|
|
|
|
reserve_size += repetitions <= 3 ? 3 : 9;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'd':
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaDayOfMonth, repetitions));
|
|
|
|
/// Day of month range [1, 3]
|
|
|
|
reserve_size += std::max(repetitions, 3);
|
|
|
|
break;
|
|
|
|
case 'a':
|
|
|
|
/// Default half day of day is "AM"
|
|
|
|
add_instruction(std::bind_front(&Action<T>::jodaHalfDayOfDay, repetitions), "AM");
|
|
|
|
reserve_size += 2;
|
|
|
|
break;
|
|
|
|
case 'K':
|
|
|
|
/// Default hour of half day is 0
|
|
|
|
add_instruction(
|
|
|
|
std::bind_front(&Action<T>::jodaHourOfHalfDay, repetitions), padValue(0, repetitions));
|
|
|
|
/// Hour of half day range [0, 11]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'h':
|
|
|
|
/// Default clock hour of half day is 12
|
|
|
|
add_instruction(
|
|
|
|
std::bind_front(&Action<T>::jodaClockHourOfHalfDay, repetitions),
|
|
|
|
padValue(12, repetitions));
|
|
|
|
/// Clock hour of half day range [1, 12]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'H':
|
|
|
|
/// Default hour of day is 0
|
|
|
|
add_instruction(std::bind_front(&Action<T>::jodaHourOfDay, repetitions), padValue(0, repetitions));
|
|
|
|
/// Hour of day range [0, 23]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'k':
|
|
|
|
/// Default clock hour of day is 24
|
|
|
|
add_instruction(std::bind_front(&Action<T>::jodaClockHourOfDay, repetitions), padValue(24, repetitions));
|
|
|
|
/// Clock hour of day range [1, 24]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'm':
|
|
|
|
/// Default minute of hour is 0
|
|
|
|
add_instruction(std::bind_front(&Action<T>::jodaMinuteOfHour, repetitions), padValue(0, repetitions));
|
|
|
|
/// Minute of hour range [0, 59]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 's':
|
|
|
|
/// Default second of minute is 0
|
|
|
|
add_instruction(std::bind_front(&Action<T>::jodaSecondOfMinute, repetitions), padValue(0, repetitions));
|
|
|
|
/// Second of minute range [0, 59]
|
|
|
|
reserve_size += std::max(repetitions, 2);
|
|
|
|
break;
|
|
|
|
case 'S':
|
2023-01-31 02:20:59 +00:00
|
|
|
/// Default fraction of second is 0
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaFractionOfSecond, repetitions));
|
|
|
|
/// 'S' repetitions range [0, 9]
|
|
|
|
reserve_size += repetitions <= 9 ? repetitions : 9;
|
|
|
|
break;
|
2022-12-14 06:28:43 +00:00
|
|
|
case 'z':
|
|
|
|
if (repetitions <= 3)
|
|
|
|
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "Short name time zone is not yet supported");
|
|
|
|
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::jodaTimezone, repetitions));
|
|
|
|
/// Longest length of full name of time zone is 32.
|
|
|
|
reserve_size += 32;
|
|
|
|
break;
|
|
|
|
case 'Z':
|
2023-01-23 21:13:58 +00:00
|
|
|
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for TIMEZONE_OFFSET_ID");
|
2022-12-14 06:28:43 +00:00
|
|
|
default:
|
|
|
|
if (isalpha(*cur_token))
|
|
|
|
throw Exception(ErrorCodes::NOT_IMPLEMENTED, "format is not supported for {}", String(cur_token, repetitions));
|
|
|
|
|
|
|
|
std::string_view literal(cur_token, pos - cur_token);
|
|
|
|
instructions.emplace_back(std::bind_front(&Action<T>::template jodaLiteral<decltype(literal)>, literal));
|
|
|
|
reserve_size += pos - cur_token;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return reserve_size;
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2020-07-14 06:04:37 +00:00
|
|
|
struct NameFormatDateTime
|
|
|
|
{
|
|
|
|
static constexpr auto name = "formatDateTime";
|
|
|
|
};
|
|
|
|
|
2020-07-14 06:29:01 +00:00
|
|
|
struct NameFromUnixTime
|
2020-07-14 06:04:37 +00:00
|
|
|
{
|
2022-08-27 21:58:34 +00:00
|
|
|
static constexpr auto name = "fromUnixTimestamp";
|
2020-07-14 06:04:37 +00:00
|
|
|
};
|
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
struct NameFormatDateTimeInJodaSyntax
|
|
|
|
{
|
|
|
|
static constexpr auto name = "formatDateTimeInJodaSyntax";
|
|
|
|
};
|
|
|
|
|
|
|
|
struct NameFromUnixTimeInJodaSyntax
|
|
|
|
{
|
|
|
|
static constexpr auto name = "fromUnixTimestampInJodaSyntax";
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
using FunctionFormatDateTime = FunctionFormatDateTimeImpl<NameFormatDateTime, FormatDateTimeTraits::SupportInteger::No, FormatDateTimeTraits::FormatSyntax::MySQL>;
|
|
|
|
using FunctionFromUnixTimestamp = FunctionFormatDateTimeImpl<NameFromUnixTime, FormatDateTimeTraits::SupportInteger::Yes, FormatDateTimeTraits::FormatSyntax::MySQL>;
|
|
|
|
using FunctionFormatDateTimeInJodaSyntax = FunctionFormatDateTimeImpl<NameFormatDateTimeInJodaSyntax, FormatDateTimeTraits::SupportInteger::No, FormatDateTimeTraits::FormatSyntax::Joda>;
|
|
|
|
using FunctionFromUnixTimestampInJodaSyntax = FunctionFormatDateTimeImpl<NameFromUnixTimeInJodaSyntax, FormatDateTimeTraits::SupportInteger::Yes, FormatDateTimeTraits::FormatSyntax::Joda>;
|
2020-07-14 06:04:37 +00:00
|
|
|
|
2020-09-07 18:00:37 +00:00
|
|
|
}
|
|
|
|
|
2022-07-04 07:01:39 +00:00
|
|
|
REGISTER_FUNCTION(FormatDateTime)
|
2018-09-26 00:31:40 +00:00
|
|
|
{
|
|
|
|
factory.registerFunction<FunctionFormatDateTime>();
|
2022-08-27 21:58:34 +00:00
|
|
|
factory.registerFunction<FunctionFromUnixTimestamp>();
|
|
|
|
factory.registerAlias("FROM_UNIXTIME", "fromUnixTimestamp");
|
2018-09-26 00:31:40 +00:00
|
|
|
|
2022-12-14 06:28:43 +00:00
|
|
|
factory.registerFunction<FunctionFormatDateTimeInJodaSyntax>();
|
|
|
|
factory.registerFunction<FunctionFromUnixTimestampInJodaSyntax>();
|
|
|
|
}
|
2018-09-26 00:31:40 +00:00
|
|
|
}
|