#pragma once #include #include #include #include #include #include #include namespace DB { /** Функции работы с датой и временем. * * toYear, toMonth, toDayOfMonth, toDayOfWeek, toHour, toMinute, toSecond, * toMonday, toStartOfMonth, toStartOfYear, toStartOfMinute, toStartOfHour * toTime, * now * TODO: makeDate, makeDateTime * * (toDate - расположена в файле FunctionsConversion.h) * * Возвращаемые типы: * toYear -> UInt16 * toMonth, toDayOfMonth, toDayOfWeek, toHour, toMinute, toSecond -> UInt8 * toMonday, toStartOfMonth, toStartOfYear -> Date * toStartOfMinute, toStartOfHour, toTime, now -> DateTime * * А также: * * timeSlot(EventTime) * - округляет время до получаса. * * timeSlots(StartTime, Duration) * - для интервала времени, начинающегося в StartTime и продолжающегося Duration секунд, * возвращает массив моментов времени, состоящий из округлений вниз до получаса точек из этого интервала. * Например, timeSlots(toDateTime('2012-01-01 12:20:00'), 600) = [toDateTime('2012-01-01 12:00:00'), toDateTime('2012-01-01 12:30:00')]. * Это нужно для поиска хитов, входящих в соответствующий визит. */ #define TIME_SLOT_SIZE 1800 struct ToYearImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toYear(t); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toYear(DayNum_t(d)); } }; struct ToMonthImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toMonth(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toMonth(DayNum_t(d)); } }; struct ToDayOfMonthImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toDayOfMonth(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toDayOfMonth(DayNum_t(d)); } }; struct ToDayOfWeekImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toDayOfWeek(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toDayOfWeek(DayNum_t(d)); } }; struct ToHourImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toHourInaccurate(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toHour", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToMinuteImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toMinuteInaccurate(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toMinute", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToSecondImpl { static inline UInt8 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toSecondInaccurate(t); } static inline UInt8 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toSecond", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToMondayImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfWeek(date_lut.toDayNum(t)); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfWeek(DayNum_t(d)); } }; struct ToStartOfMonthImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfMonth(date_lut.toDayNum(t)); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfMonth(DayNum_t(d)); } }; struct ToStartOfQuarterImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfQuarter(date_lut.toDayNum(t)); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfQuarter(DayNum_t(d)); } }; struct ToStartOfYearImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfYear(date_lut.toDayNum(t)); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toFirstDayNumOfYear(DayNum_t(d)); } }; struct ToTimeImpl { /// При переводе во время, дату будем приравнивать к 1970-01-02. static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toTimeInaccurate(t) + 86400; } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toTime", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToStartOfMinuteImpl { static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toStartOfMinuteInaccurate(t); } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toStartOfMinute", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToStartOfHourImpl { static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toStartOfHourInaccurate(t); } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toStartOfHour", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToRelativeYearNumImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toYear(t); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toYear(DayNum_t(d)); } }; struct ToRelativeMonthNumImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toRelativeMonthNum(t); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toRelativeMonthNum(DayNum_t(d)); } }; struct ToRelativeWeekNumImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toRelativeWeekNum(t); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return date_lut.toRelativeWeekNum(DayNum_t(d)); } }; struct ToRelativeDayNumImpl { static inline UInt16 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toDayNum(t); } static inline UInt16 execute(UInt16 d, DateLUTSingleton & date_lut) { return static_cast(d); } }; struct ToRelativeHourNumImpl { static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toRelativeHourNum(t); } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toRelativeHourNum", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToRelativeMinuteNumImpl { static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return date_lut.toRelativeMinuteNum(t); } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toRelativeMinuteNum", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; struct ToRelativeSecondNumImpl { static inline UInt32 execute(UInt32 t, DateLUTSingleton & date_lut) { return t; } static inline UInt32 execute(UInt16 d, DateLUTSingleton & date_lut) { throw Exception("Illegal type Date of argument for function toRelativeSecondNum", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; template struct DateTimeTransformImpl { static void execute(Block & block, const ColumnNumbers & arguments, size_t result) { DateLUTSingleton & date_lut = DateLUTSingleton::instance(); if (const ColumnVector * col_from = typeid_cast *>(&*block.getByPosition(arguments[0]).column)) { ColumnVector * col_to = new ColumnVector; block.getByPosition(result).column = col_to; const typename ColumnVector::Container_t & vec_from = col_from->getData(); typename ColumnVector::Container_t & vec_to = col_to->getData(); size_t size = vec_from.size(); vec_to.resize(size); for (size_t i = 0; i < size; ++i) vec_to[i] = Transform::execute(vec_from[i], date_lut); } else if (const ColumnConst * col_from = typeid_cast *>(&*block.getByPosition(arguments[0]).column)) { block.getByPosition(result).column = new ColumnConst(col_from->size(), Transform::execute(col_from->getData(), date_lut)); } else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + Name::get(), ErrorCodes::ILLEGAL_COLUMN); } }; template class FunctionDateOrDateTimeToSomething : public IFunction { public: /// Получить имя функции. String getName() const { return Name::get(); } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 1) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); return new ToDataType; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { IDataType * from_type = &*block.getByPosition(arguments[0]).type; if (typeid_cast(from_type)) DateTimeTransformImpl::execute(block, arguments, result); else if (typeid_cast(from_type)) DateTimeTransformImpl::execute(block, arguments, result); else throw Exception("Illegal type " + block.getByPosition(arguments[0]).type->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; /// Получить текущее время. (Оно - константа, вычисляется один раз за весь запрос.) class FunctionNow : public IFunction { public: /// Получить имя функции. String getName() const { return "now"; } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 0) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be 0.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); return new DataTypeDateTime; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { block.getByPosition(result).column = new ColumnConstUInt32( block.rowsInFirstColumn(), time(0)); } }; class FunctionTimeSlot : public IFunction { public: /// Получить имя функции. String getName() const { return "timeSlot"; } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 1) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!typeid_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of first argument of function " + getName() + ". Must be DateTime.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new DataTypeDateTime; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { if (const ColumnUInt32 * times = typeid_cast(&*block.getByPosition(arguments[0]).column)) { ColumnUInt32 * res = new ColumnUInt32; ColumnUInt32::Container_t & res_vec = res->getData(); const ColumnUInt32::Container_t & vec = times->getData(); size_t size = vec.size(); res_vec.resize(size); for (size_t i = 0; i < size; ++i) res_vec[i] = vec[i] / TIME_SLOT_SIZE * TIME_SLOT_SIZE; block.getByPosition(result).column = res; } else if (const ColumnConstUInt32 * const_times = typeid_cast(&*block.getByPosition(arguments[0]).column)) { block.getByPosition(result).column = new ColumnConstUInt32(block.rowsInFirstColumn(), const_times->getData() / TIME_SLOT_SIZE * TIME_SLOT_SIZE); } else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } }; template struct TimeSlotsImpl { static void vector_vector( const PODArray & starts, const PODArray & durations, PODArray & result_values, ColumnArray::Offsets_t & result_offsets) { size_t size = starts.size(); result_offsets.resize(size); result_values.reserve(size); ColumnArray::Offset_t current_offset = 0; for (size_t i = 0; i < size; ++i) { for (UInt32 value = starts[i] / TIME_SLOT_SIZE; value <= (starts[i] + durations[i]) / TIME_SLOT_SIZE; ++value) { result_values.push_back(value * TIME_SLOT_SIZE); ++current_offset; } result_offsets[i] = current_offset; } } static void vector_constant( const PODArray & starts, DurationType duration, PODArray & result_values, ColumnArray::Offsets_t & result_offsets) { size_t size = starts.size(); result_offsets.resize(size); result_values.reserve(size); ColumnArray::Offset_t current_offset = 0; for (size_t i = 0; i < size; ++i) { for (UInt32 value = starts[i] / TIME_SLOT_SIZE; value <= (starts[i] + duration) / TIME_SLOT_SIZE; ++value) { result_values.push_back(value * TIME_SLOT_SIZE); ++current_offset; } result_offsets[i] = current_offset; } } static void constant_vector( UInt32 start, const PODArray & durations, PODArray & result_values, ColumnArray::Offsets_t & result_offsets) { size_t size = durations.size(); result_offsets.resize(size); result_values.reserve(size); ColumnArray::Offset_t current_offset = 0; for (size_t i = 0; i < size; ++i) { for (UInt32 value = start / TIME_SLOT_SIZE; value <= (start + durations[i]) / TIME_SLOT_SIZE; ++value) { result_values.push_back(value * TIME_SLOT_SIZE); ++current_offset; } result_offsets[i] = current_offset; } } static void constant_constant( UInt32 start, DurationType duration, Array & result) { for (UInt32 value = start / TIME_SLOT_SIZE; value <= (start + duration) / TIME_SLOT_SIZE; ++value) result.push_back(static_cast(value * TIME_SLOT_SIZE)); } }; class FunctionTimeSlots : public IFunction { public: /// Получить имя функции. String getName() const { return "timeSlots"; } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 2) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + toString(arguments.size()) + ", should be 2.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!typeid_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of first argument of function " + getName() + ". Must be DateTime.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); if (!typeid_cast(&*arguments[1])) throw Exception("Illegal type " + arguments[1]->getName() + " of second argument of function " + getName() + ". Must be UInt32.", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new DataTypeArray(new DataTypeDateTime); } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { const ColumnUInt32 * starts = typeid_cast(&*block.getByPosition(arguments[0]).column); const ColumnConstUInt32 * const_starts = typeid_cast(&*block.getByPosition(arguments[0]).column); const ColumnUInt32 * durations = typeid_cast(&*block.getByPosition(arguments[1]).column); const ColumnConstUInt32 * const_durations = typeid_cast(&*block.getByPosition(arguments[1]).column); ColumnArray * res = new ColumnArray(new ColumnUInt32); ColumnUInt32::Container_t & res_values = typeid_cast(res->getData()).getData(); if (starts && durations) { TimeSlotsImpl::vector_vector(starts->getData(), durations->getData(), res_values, res->getOffsets()); block.getByPosition(result).column = res; } else if (starts && const_durations) { TimeSlotsImpl::vector_constant(starts->getData(), const_durations->getData(), res_values, res->getOffsets()); block.getByPosition(result).column = res; } else if (const_starts && durations) { TimeSlotsImpl::constant_vector(const_starts->getData(), durations->getData(), res_values, res->getOffsets()); block.getByPosition(result).column = res; } else if (const_starts && const_durations) { Array const_res; TimeSlotsImpl::constant_constant(const_starts->getData(), const_durations->getData(), const_res); block.getByPosition(result).column = new ColumnConstArray(block.rowsInFirstColumn(), const_res, new DataTypeArray(new DataTypeDateTime)); } else throw Exception("Illegal columns " + block.getByPosition(arguments[0]).column->getName() + ", " + block.getByPosition(arguments[1]).column->getName() + " of arguments of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } }; struct NameToYear { static const char * get() { return "toYear"; } }; struct NameToMonth { static const char * get() { return "toMonth"; } }; struct NameToDayOfMonth { static const char * get() { return "toDayOfMonth"; } }; struct NameToDayOfWeek { static const char * get() { return "toDayOfWeek"; } }; struct NameToHour { static const char * get() { return "toHour"; } }; struct NameToMinute { static const char * get() { return "toMinute"; } }; struct NameToSecond { static const char * get() { return "toSecond"; } }; struct NameToMonday { static const char * get() { return "toMonday"; } }; struct NameToStartOfMonth { static const char * get() { return "toStartOfMonth"; } }; struct NameToStartOfQuarter { static const char * get() { return "toStartOfQuarter"; } }; struct NameToStartOfYear { static const char * get() { return "toStartOfYear"; } }; struct NameToStartOfMinute { static const char * get() { return "toStartOfMinute"; } }; struct NameToStartOfHour { static const char * get() { return "toStartOfHour"; } }; struct NameToTime { static const char * get() { return "toTime"; } }; struct NameToRelativeYearNum { static const char * get() { return "toRelativeYearNum"; } }; struct NameToRelativeMonthNum { static const char * get() { return "toRelativeMonthNum"; } }; struct NameToRelativeWeekNum { static const char * get() { return "toRelativeWeekNum"; } }; struct NameToRelativeDayNum { static const char * get() { return "toRelativeDayNum"; } }; struct NameToRelativeHourNum { static const char * get() { return "toRelativeHourNum"; } }; struct NameToRelativeMinuteNum { static const char * get() { return "toRelativeMinuteNum"; } }; struct NameToRelativeSecondNum { static const char * get() { return "toRelativeSecondNum"; } }; typedef FunctionDateOrDateTimeToSomething FunctionToYear; typedef FunctionDateOrDateTimeToSomething FunctionToMonth; typedef FunctionDateOrDateTimeToSomething FunctionToDayOfMonth; typedef FunctionDateOrDateTimeToSomething FunctionToDayOfWeek; typedef FunctionDateOrDateTimeToSomething FunctionToHour; typedef FunctionDateOrDateTimeToSomething FunctionToMinute; typedef FunctionDateOrDateTimeToSomething FunctionToSecond; typedef FunctionDateOrDateTimeToSomething FunctionToMonday; typedef FunctionDateOrDateTimeToSomething FunctionToStartOfMonth; typedef FunctionDateOrDateTimeToSomething FunctionToStartOfQuarter; typedef FunctionDateOrDateTimeToSomething FunctionToStartOfYear; typedef FunctionDateOrDateTimeToSomething FunctionToStartOfMinute; typedef FunctionDateOrDateTimeToSomething FunctionToStartOfHour; typedef FunctionDateOrDateTimeToSomething FunctionToTime; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeYearNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeMonthNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeWeekNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeDayNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeHourNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeMinuteNum; typedef FunctionDateOrDateTimeToSomething FunctionToRelativeSecondNum; }