#pragma once #include #include #include #include #include #include #include #include #include #include #include #include namespace DB { /** Функции работы со строками: * * length, empty, notEmpty, * concat, substring, lower, upper, reverse, // left, right, insert, replace, * // escape, quote, unescape, unquote, trim, trimLeft, trimRight, pad, padLeft * lengthUTF8, substringUTF8, lowerUTF8, upperUTF8, reverseUTF8, // leftUTF8, rightUTF8, insertUTF8, replaceUTF8 * // padUTF8, padLeftUTF8. * * s -> UInt8: empty, notEmpty * s -> UInt64: length, lengthUTF8 * s -> s: lower, upper, lowerUTF8, upperUTF8, reverse, reverseUTF8, escape, quote, unescape, unquote, trim, trimLeft, trimRight * s, s -> s: concat * s, c1, c2 -> s: substring, substringUTF8 * s, c1 -> s: left, right, leftUTF8, rightUTF8 * s, c1, s2 -> s: insert, insertUTF8, pad, padLeft, padUTF8, padLeftUTF8 * s, c1, c2, s2 -> s: replace, replaceUTF8 * * Функции поиска строк и регулярных выражений расположены отдельно. * Функции работы с URL расположены отдельно. * Функции кодирования строк, конвертации в другие типы расположены отдельно. * * Функции length, empty, notEmpty также работают с массивами. */ template struct EmptyImpl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res) { size_t size = offsets.size(); ColumnArray::Offset_t prev_offset = 1; for (size_t i = 0; i < size; ++i) { res[i] = negative ^ (offsets[i] == prev_offset); prev_offset = offsets[i] + 1; } } static void vector_fixed_to_constant(const std::vector & data, size_t n, UInt8 & res) { res = negative ^ (n == 0); } static void vector_fixed_to_vector(const std::vector & data, size_t n, std::vector & res) { } static void constant(const std::string & data, UInt8 & res) { res = negative ^ (data.empty()); } static void array(const ColumnArray::Offsets_t & offsets, std::vector & res) { size_t size = offsets.size(); ColumnArray::Offset_t prev_offset = 0; for (size_t i = 0; i < size; ++i) { res[i] = negative ^ (offsets[i] == prev_offset); prev_offset = offsets[i]; } } static void constant_array(const Array & data, UInt8 & res) { res = negative ^ (data.empty()); } }; /** Вычисляет длину строки в байтах. */ struct LengthImpl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res) { size_t size = offsets.size(); for (size_t i = 0; i < size; ++i) res[i] = i == 0 ? (offsets[i] - 1) : (offsets[i] - 1 - offsets[i - 1]); } static void vector_fixed_to_constant(const std::vector & data, size_t n, UInt64 & res) { res = n; } static void vector_fixed_to_vector(const std::vector & data, size_t n, std::vector & res) { } static void constant(const std::string & data, UInt64 & res) { res = data.size(); } static void array(const ColumnArray::Offsets_t & offsets, std::vector & res) { size_t size = offsets.size(); for (size_t i = 0; i < size; ++i) res[i] = i == 0 ? (offsets[i]) : (offsets[i] - offsets[i - 1]); } static void constant_array(const Array & data, UInt64 & res) { res = data.size(); } }; /** Если строка представляет собой текст в кодировке UTF-8, то возвращает длину текста в кодовых точках. * (не в символах: длина текста "ё" может быть как 1, так и 2, в зависимости от нормализации) * Иначе - поведение не определено. */ struct LengthUTF8Impl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res) { size_t size = offsets.size(); ColumnArray::Offset_t prev_offset = 0; for (size_t i = 0; i < size; ++i) { res[i] = 0; for (const UInt8 * c = &data[prev_offset]; c + 1 < &data[offsets[i]]; ++c) if (*c <= 0x7F || *c >= 0xC0) ++res[i]; prev_offset = offsets[i]; } } static void vector_fixed_to_constant(const std::vector & data, size_t n, UInt64 & res) { } static void vector_fixed_to_vector(const std::vector & data, size_t n, std::vector & res) { size_t size = data.size() / n; for (size_t i = 0; i < size; ++i) { res[i] = 0; for (const UInt8 * c = &data[i * n]; c < &data[(i + 1) * n]; ++c) if (*c <= 0x7F || *c >= 0xC0) ++res[i]; } } static void constant(const std::string & data, UInt64 & res) { res = 0; for (const UInt8 * c = reinterpret_cast(data.data()); c < reinterpret_cast(data.data() + data.size()); ++c) if (*c <= 0x7F || *c >= 0xC0) ++res; } static void array(const ColumnArray::Offsets_t & offsets, std::vector & res) { throw Exception("Cannot apply function lengthUTF8 to Array argument", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } static void constant_array(const Array & data, UInt64 & res) { throw Exception("Cannot apply function lengthUTF8 to Array argument", ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); } }; /** Переводит строку в нижний (верхний) регистр, в текущей локали, в однобайтовой кодировке. */ template struct LowerUpperImpl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.resize(data.size()); res_offsets = offsets; array(&*data.begin(), &*data.end(), &*res_data.begin()); } static void vector_fixed(const std::vector & data, size_t n, std::vector & res_data) { res_data.resize(data.size()); array(&*data.begin(), &*data.end(), &*res_data.begin()); } static void constant(const std::string & data, std::string & res_data) { res_data.resize(data.size()); array(reinterpret_cast(&*data.begin()), reinterpret_cast(&*data.end()), reinterpret_cast(&*res_data.begin())); } private: static void array(const UInt8 * src, const UInt8 * src_end, UInt8 * dst) { for (; src < src_end; ++src, ++dst) *dst = F(*src); } }; /** Если строка содержит текст в кодировке UTF-8 - перевести его в нижний (верхний) регистр. * Замечание: предполагается, что после перевода символа в другой регистр, * длина его мультибайтовой последовательности в UTF-8 не меняется. * Иначе - поведение не определено. */ template struct LowerUpperUTF8Impl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.resize(data.size()); res_offsets = offsets; array(&*data.begin(), &*data.end(), &*res_data.begin()); } static void vector_fixed(const std::vector & data, size_t n, std::vector & res_data) { res_data.resize(data.size()); array(&*data.begin(), &*data.end(), &*res_data.begin()); } static void constant(const std::string & data, std::string & res_data) { res_data.resize(data.size()); array(reinterpret_cast(&*data.begin()), reinterpret_cast(&*data.end()), reinterpret_cast(&*res_data.begin())); } private: static void array(const UInt8 * src, const UInt8 * src_end, UInt8 * dst) { static Poco::UTF8Encoding utf8; while (src < src_end) { int chars = utf8.convert(F(utf8.convert(src)), dst, src_end - src); if (chars) { src += chars; dst += chars; } else { ++src; ++dst; } } } }; /** Разворачивает строку в байтах. */ struct ReverseImpl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.resize(data.size()); res_offsets = offsets; size_t size = offsets.size(); ColumnArray::Offset_t prev_offset = 0; for (size_t i = 0; i < size; ++i) { for (size_t j = prev_offset; j < offsets[i] - 1; ++j) res_data[j] = data[offsets[i] + prev_offset - 2 - j]; res_data[offsets[i] - 1] = 0; prev_offset = offsets[i]; } } static void vector_fixed(const std::vector & data, size_t n, std::vector & res_data) { res_data.resize(data.size()); size_t size = data.size() / n; for (size_t i = 0; i < size; ++i) for (size_t j = i * n; j < (i + 1) * n; ++j) res_data[j] = data[(i * 2 + 1) * n - j - 1]; } static void constant(const std::string & data, std::string & res_data) { res_data.resize(data.size()); for (size_t j = 0; j < data.size(); ++j) res_data[j] = data[data.size() - j - 1]; } }; /** Разворачивает последовательность кодовых точек в строке в кодировке UTF-8. * Результат может не соответствовать ожидаемому, так как модифицирующие кодовые точки (например, диакритика) могут примениться не к тем символам. * Если строка не в кодировке UTF-8, то поведение не определено. */ struct ReverseUTF8Impl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.resize(data.size()); res_offsets = offsets; size_t size = offsets.size(); ColumnArray::Offset_t prev_offset = 0; for (size_t i = 0; i < size; ++i) { ColumnArray::Offset_t j = prev_offset; while (j < offsets[i] - 1) { if (data[j] < 0xBF) { res_data[offsets[i] + prev_offset - 2 - j] = data[j]; j += 1; } else if (data[j] < 0xE0) { memcpy(&res_data[offsets[i] + prev_offset - 2 - j - 1], &data[j], 2); j += 2; } else if (data[j] < 0xF0) { memcpy(&res_data[offsets[i] + prev_offset - 2 - j - 2], &data[j], 3); j += 3; } else { res_data[offsets[i] + prev_offset - 2 - j] = data[j]; j += 1; } } res_data[offsets[i] - 1] = 0; prev_offset = offsets[i]; } } static void vector_fixed(const std::vector & data, size_t n, std::vector & res_data) { throw Exception("Cannot apply function reverseUTF8 to fixed string.", ErrorCodes::ILLEGAL_COLUMN); } static void constant(const std::string & data, std::string & res_data) { res_data.resize(data.size()); size_t j = 0; while (j < data.size()) { if (static_cast(data[j]) < 0xBF) { res_data[data.size() - 1 - j] = data[j]; j += 1; } else if (static_cast(data[j]) < 0xE0) { memcpy(&res_data[data.size() - 1 - j - 1], &data[j], 2); j += 2; } else if (static_cast(data[j]) < 0xF0) { memcpy(&res_data[data.size() - 1 - j - 2], &data[j], 3); j += 3; } else { res_data[data.size() - 1 - j] = data[j]; j += 1; } } } }; /** Склеивает две строки. */ struct ConcatImpl { static void vector_vector( const std::vector & a_data, const ColumnArray::Offsets_t & a_offsets, const std::vector & b_data, const ColumnArray::Offsets_t & b_offsets, std::vector & c_data, ColumnArray::Offsets_t & c_offsets) { size_t size = a_offsets.size(); c_data.resize(a_data.size() + b_data.size() - size); c_offsets.resize(size); ColumnArray::Offset_t offset = 0; ColumnArray::Offset_t a_offset = 0; ColumnArray::Offset_t b_offset = 0; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[offset], &a_data[a_offset], a_offsets[i] - a_offset - 1); offset += a_offsets[i] - a_offset - 1; memcpy(&c_data[offset], &b_data[b_offset], b_offsets[i] - b_offset); offset += b_offsets[i] - b_offset; a_offset = a_offsets[i]; b_offset = b_offsets[i]; c_offsets[i] = offset; } } static void vector_fixed_vector( const std::vector & a_data, const ColumnArray::Offsets_t & a_offsets, const std::vector & b_data, ColumnArray::Offset_t b_n, std::vector & c_data, ColumnArray::Offsets_t & c_offsets) { size_t size = a_offsets.size(); c_data.resize(a_data.size() + b_data.size()); c_offsets.resize(size); ColumnArray::Offset_t offset = 0; ColumnArray::Offset_t a_offset = 0; ColumnArray::Offset_t b_offset = 0; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[offset], &a_data[a_offset], a_offsets[i] - a_offset - 1); offset += a_offsets[i] - a_offset - 1; memcpy(&c_data[offset], &b_data[b_offset], b_n); offset += b_n; c_data[offset] = 0; offset += 1; a_offset = a_offsets[i]; b_offset += b_n; c_offsets[i] = offset; } } static void vector_constant( const std::vector & a_data, const ColumnArray::Offsets_t & a_offsets, const std::string & b, std::vector & c_data, ColumnArray::Offsets_t & c_offsets) { size_t size = a_offsets.size(); c_data.resize(a_data.size() + b.size() * size); c_offsets = a_offsets; for (size_t i = 0; i < size; ++i) c_offsets[i] += b.size() + i; ColumnArray::Offset_t offset = 0; ColumnArray::Offset_t a_offset = 0; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[offset], &a_data[a_offset], a_offsets[i] - a_offset - 1); offset += a_offsets[i] - a_offset - 1; memcpy(&c_data[offset], b.data(), b.size() + 1); offset += b.size() + 1; a_offset = a_offsets[i]; } } static void fixed_vector_vector( const std::vector & a_data, ColumnArray::Offset_t a_n, const std::vector & b_data, const ColumnArray::Offsets_t & b_offsets, std::vector & c_data, ColumnArray::Offsets_t & c_offsets) { size_t size = b_offsets.size(); c_data.resize(a_data.size() + b_data.size()); c_offsets.resize(size); ColumnArray::Offset_t offset = 0; ColumnArray::Offset_t a_offset = 0; ColumnArray::Offset_t b_offset = 0; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[offset], &a_data[a_offset], a_n); offset += a_n; memcpy(&c_data[offset], &b_data[b_offset], b_offsets[i] - b_offset); offset += b_offsets[i] - b_offset; a_offset = a_n; b_offset += b_offsets[i]; c_offsets[i] = offset; } } static void fixed_vector_fixed_vector( const std::vector & a_data, ColumnArray::Offset_t a_n, const std::vector & b_data, ColumnArray::Offset_t b_n, std::vector & c_data) { size_t size = a_data.size() / a_n; c_data.resize(a_data.size() + b_data.size()); ColumnArray::Offset_t c_n = a_n + b_n; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[i * c_n], &a_data[i * a_n], a_n); memcpy(&c_data[i * c_n + a_n], &b_data[i * b_n], b_n); } } static void fixed_vector_constant( const std::vector & a_data, ColumnArray::Offset_t a_n, const std::string & b, std::vector & c_data) { size_t size = a_data.size() / a_n; ColumnArray::Offset_t b_n = b.size(); ColumnArray::Offset_t c_n = a_n + b_n; c_data.resize(a_data.size() + size * b_n); for (size_t i = 0; i < size; ++i) { memcpy(&c_data[i * c_n], &a_data[i * a_n], a_n); memcpy(&c_data[i * c_n + a_n], b.data(), b_n); } } static void constant_vector( const std::string & a, const std::vector & b_data, const ColumnArray::Offsets_t & b_offsets, std::vector & c_data, ColumnArray::Offsets_t & c_offsets) { size_t size = b_offsets.size(); c_data.resize(b_data.size() + a.size() * size); c_offsets = b_offsets; for (size_t i = 0; i < size; ++i) c_offsets[i] += a.size() + i; ColumnArray::Offset_t offset = 0; ColumnArray::Offset_t b_offset = 0; for (size_t i = 0; i < size; ++i) { memcpy(&c_data[offset], a.data(), a.size()); offset += a.size(); memcpy(&c_data[offset], &b_data[b_offset], b_offsets[i] - b_offset); offset += b_offsets[i] - b_offset; b_offset = b_offsets[i]; } } static void constant_fixed_vector( const std::string & a, const std::vector & b_data, ColumnArray::Offset_t b_n, std::vector & c_data) { size_t size = b_data.size() / b_n; ColumnArray::Offset_t a_n = a.size(); ColumnArray::Offset_t c_n = a_n + b_n; c_data.resize(size * a_n + b_data.size()); for (size_t i = 0; i < size; ++i) { memcpy(&c_data[i * c_n], a.data(), a_n); memcpy(&c_data[i * c_n + a_n], &b_data[i * b_n], b_n); } } static void constant_constant( const std::string & a, const std::string & b, std::string & c) { c = a + b; } }; /** Выделяет подстроку в строке, как последовательности байт. */ struct SubstringImpl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, size_t start, size_t length, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.reserve(data.size()); size_t size = offsets.size(); res_offsets.resize(size); ColumnArray::Offset_t prev_offset = 0; ColumnArray::Offset_t res_offset = 0; for (size_t i = 0; i < size; ++i) { if (start + prev_offset >= offsets[i] + 1) { res_data.resize(res_data.size() + 1); res_data[res_offset] = 0; ++res_offset; } else { size_t bytes_to_copy = std::min(offsets[i] - prev_offset - start, length); res_data.resize(res_data.size() + bytes_to_copy + 1); memcpy(&res_data[res_offset], &data[prev_offset + start - 1], bytes_to_copy); res_offset += bytes_to_copy + 1; res_data[res_offset - 1] = 0; } res_offsets[i] = res_offset; prev_offset = offsets[i]; } } static void vector_fixed(const std::vector & data, size_t n, size_t start, size_t length, std::vector & res_data) { if (length == 0 || start + length > n + 1) throw Exception("Index out of bound for function substring of fixed size value", ErrorCodes::ARGUMENT_OUT_OF_BOUND); size_t size = data.size() / n; res_data.resize(length * size); for (size_t i = 0; i < size; ++i) memcpy(&res_data[i * length], &data[i * n + start - 1], length); } static void constant(const std::string & data, size_t start, size_t length, std::string & res_data) { if (start + length > data.size() + 1) throw Exception("Index out of bound for function substring of fixed size value", ErrorCodes::ARGUMENT_OUT_OF_BOUND); res_data = data.substr(start - 1, length); } }; /** Если строка в кодировке UTF-8, то выделяет в ней подстроку кодовых точек. * Иначе - поведение не определено. */ struct SubstringUTF8Impl { static void vector(const std::vector & data, const ColumnArray::Offsets_t & offsets, size_t start, size_t length, std::vector & res_data, ColumnArray::Offsets_t & res_offsets) { res_data.reserve(data.size()); size_t size = offsets.size(); res_offsets.resize(size); ColumnArray::Offset_t prev_offset = 0; ColumnArray::Offset_t res_offset = 0; for (size_t i = 0; i < size; ++i) { ColumnArray::Offset_t j = prev_offset; ColumnArray::Offset_t pos = 1; ColumnArray::Offset_t bytes_start = 0; ColumnArray::Offset_t bytes_length = 0; while (j < offsets[i] - 1) { if (pos == start) bytes_start = j - prev_offset + 1; if (data[j] < 0xBF) j += 1; else if (data[j] < 0xE0) j += 2; else if (data[j] < 0xF0) j += 3; else j += 1; if (pos >= start && pos < start + length) bytes_length = j - prev_offset + 1 - bytes_start; else if (pos >= start + length) break; ++pos; } if (bytes_start == 0) { res_data.resize(res_data.size() + 1); res_data[res_offset] = 0; ++res_offset; } else { size_t bytes_to_copy = std::min(offsets[i] - prev_offset - bytes_start, bytes_length); res_data.resize(res_data.size() + bytes_to_copy + 1); memcpy(&res_data[res_offset], &data[prev_offset + bytes_start - 1], bytes_to_copy); res_offset += bytes_to_copy + 1; res_data[res_offset - 1] = 0; } res_offsets[i] = res_offset; prev_offset = offsets[i]; } } static void vector_fixed(const std::vector & data, ColumnArray::Offset_t n, size_t start, size_t length, std::vector & res_data) { throw Exception("Cannot apply function substringUTF8 to fixed string.", ErrorCodes::ILLEGAL_COLUMN); } static void constant(const std::string & data, size_t start, size_t length, std::string & res_data) { if (start + length > data.size() + 1) throw Exception("Index out of bound for function substring of constant value", ErrorCodes::ARGUMENT_OUT_OF_BOUND); ColumnArray::Offset_t j = 0; ColumnArray::Offset_t pos = 1; ColumnArray::Offset_t bytes_start = 0; ColumnArray::Offset_t bytes_length = 0; while (j < data.size()) { if (pos == start) bytes_start = j + 1; if (static_cast(data[j]) < 0xBF) j += 1; else if (static_cast(data[j]) < 0xE0) j += 2; else if (static_cast(data[j]) < 0xF0) j += 3; else j += 1; if (pos >= start && pos < start + length) bytes_length = j + 1 - bytes_start; else if (pos >= start + length) break; ++pos; } if (bytes_start != 0) res_data = data.substr(bytes_start - 1, bytes_length); } }; template class FunctionStringOrArrayToT : 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 " + Poco::NumberFormatter::format(arguments.size()) + ", should be 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!dynamic_cast(&*arguments[0]) && !dynamic_cast(&*arguments[0]) && !dynamic_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new typename DataTypeFromFieldType::Type; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { const ColumnPtr column = block.getByPosition(arguments[0]).column; if (const ColumnString * col = dynamic_cast(&*column)) { ColumnVector * col_res = new ColumnVector; block.getByPosition(result).column = col_res; typename ColumnVector::Container_t & vec_res = col_res->getData(); vec_res.resize(col->size()); Impl::vector(dynamic_cast(col->getData()).getData(), col->getOffsets(), vec_res); } else if (const ColumnFixedString * col = dynamic_cast(&*column)) { /// Для фиксированной строки, только функция lengthUTF8 возвращает не константу. if ("lengthUTF8" != getName()) { ResultType res = 0; Impl::vector_fixed_to_constant(dynamic_cast(col->getData()).getData(), col->getN(), res); ColumnConst * col_res = new ColumnConst(col->size(), res); block.getByPosition(result).column = col_res; } else { ColumnVector * col_res = new ColumnVector; block.getByPosition(result).column = col_res; typename ColumnVector::Container_t & vec_res = col_res->getData(); vec_res.resize(col->size()); Impl::vector_fixed_to_vector(dynamic_cast(col->getData()).getData(), col->getN(), vec_res); } } else if (const ColumnConstString * col = dynamic_cast(&*column)) { ResultType res = 0; Impl::constant(col->getData(), res); ColumnConst * col_res = new ColumnConst(col->size(), res); block.getByPosition(result).column = col_res; } else if (const ColumnArray * col = dynamic_cast(&*column)) { ColumnVector * col_res = new ColumnVector; block.getByPosition(result).column = col_res; typename ColumnVector::Container_t & vec_res = col_res->getData(); vec_res.resize(col->size()); Impl::array(col->getOffsets(), vec_res); } else if (const ColumnConstArray * col = dynamic_cast(&*column)) { ResultType res = 0; Impl::constant_array(col->getData(), res); ColumnConst * col_res = new ColumnConst(col->size(), res); block.getByPosition(result).column = col_res; } else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } }; template class FunctionStringToString : 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 " + Poco::NumberFormatter::format(arguments.size()) + ", should be 1.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!dynamic_cast(&*arguments[0]) && !dynamic_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new DataTypeString; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { const ColumnPtr column = block.getByPosition(arguments[0]).column; if (const ColumnString * col = dynamic_cast(&*column)) { ColumnString * col_res = new ColumnString; block.getByPosition(result).column = col_res; Impl::vector(dynamic_cast(col->getData()).getData(), col->getOffsets(), dynamic_cast(col_res->getData()).getData(), col_res->getOffsets()); } else if (const ColumnFixedString * col = dynamic_cast(&*column)) { ColumnFixedString * col_res = new ColumnFixedString(col->getN()); block.getByPosition(result).column = col_res; Impl::vector_fixed(dynamic_cast(col->getData()).getData(), col->getN(), dynamic_cast(col_res->getData()).getData()); } else if (const ColumnConstString * col = dynamic_cast(&*column)) { String res; Impl::constant(col->getData(), res); ColumnConstString * col_res = new ColumnConstString(col->size(), res); block.getByPosition(result).column = col_res; } else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } }; template class FunctionStringStringToString : public IFunction { public: /// Получить имя функции. String getName() const { return Name::get(); } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 2) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + Poco::NumberFormatter::format(arguments.size()) + ", should be 2.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!dynamic_cast(&*arguments[0]) && !dynamic_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); if (!dynamic_cast(&*arguments[1]) && !dynamic_cast(&*arguments[1])) throw Exception("Illegal type " + arguments[1]->getName() + " of second argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new DataTypeString; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { IColumn * c0 = &*block.getByPosition(arguments[0]).column; IColumn * c1 = &*block.getByPosition(arguments[1]).column; ColumnString * c0_string = dynamic_cast(c0); ColumnString * c1_string = dynamic_cast(c1); ColumnFixedString * c0_fixed_string = dynamic_cast(c0); ColumnFixedString * c1_fixed_string = dynamic_cast(c1); ColumnConstString * c0_const = dynamic_cast(c0); ColumnConstString * c1_const = dynamic_cast(c1); /// Результат - const string if (c0_const && c1_const) { ColumnConstString * c_res = new ColumnConstString(c0_const->size(), ""); block.getByPosition(result).column = c_res; Impl::constant_constant(c0_const->getData(), c1_const->getData(), c_res->getData()); } else { /// Результат - fixed string if ((c0_fixed_string || c0_const) && (c1_fixed_string || c1_const)) { if (c0_fixed_string && c1_fixed_string) { ColumnFixedString * c_res = new ColumnFixedString(c0_fixed_string->getN() + c1_fixed_string->getN()); block.getByPosition(result).column = c_res; ColumnUInt8::Container_t & vec_res = dynamic_cast(c_res->getData()).getData(); Impl::fixed_vector_fixed_vector( dynamic_cast(c0_fixed_string->getData()).getData(), c0_fixed_string->getN(), dynamic_cast(c1_fixed_string->getData()).getData(), c1_fixed_string->getN(), vec_res); } else if (c0_fixed_string && c1_const) { ColumnFixedString * c_res = new ColumnFixedString(c0_fixed_string->getN() + c1_const->getData().size()); block.getByPosition(result).column = c_res; ColumnUInt8::Container_t & vec_res = dynamic_cast(c_res->getData()).getData(); Impl::fixed_vector_constant( dynamic_cast(c0_fixed_string->getData()).getData(), c0_fixed_string->getN(), c1_const->getData(), vec_res); } else if (c0_const && c1_fixed_string) { ColumnFixedString * c_res = new ColumnFixedString(c1_fixed_string->getN() + c0_const->getData().size()); block.getByPosition(result).column = c_res; ColumnUInt8::Container_t & vec_res = dynamic_cast(c_res->getData()).getData(); Impl::constant_fixed_vector( c0_const->getData(), dynamic_cast(c1_fixed_string->getData()).getData(), c1_fixed_string->getN(), vec_res); } } else /// Результат - string { ColumnString * c_res = new ColumnString; block.getByPosition(result).column = c_res; ColumnUInt8::Container_t & vec_res = dynamic_cast(c_res->getData()).getData(); ColumnString::Offsets_t & offsets_res = c_res->getOffsets(); if (c0_string && c1_string) Impl::vector_vector( dynamic_cast(c0_string->getData()).getData(), c0_string->getOffsets(), dynamic_cast(c1_string->getData()).getData(), c1_string->getOffsets(), vec_res, offsets_res); else if (c0_string && c1_fixed_string) Impl::vector_fixed_vector( dynamic_cast(c0_string->getData()).getData(), c0_string->getOffsets(), dynamic_cast(c1_fixed_string->getData()).getData(), c1_fixed_string->getN(), vec_res, offsets_res); else if (c0_string && c1_const) Impl::vector_constant( dynamic_cast(c0_string->getData()).getData(), c0_string->getOffsets(), c1_const->getData(), vec_res, offsets_res); else if (c0_fixed_string && c1_string) Impl::fixed_vector_vector( dynamic_cast(c0_fixed_string->getData()).getData(), c0_fixed_string->getN(), dynamic_cast(c1_string->getData()).getData(), c1_string->getOffsets(), vec_res, offsets_res); else if (c0_const && c1_string) Impl::constant_vector( c0_const->getData(), dynamic_cast(c1_string->getData()).getData(), c1_string->getOffsets(), vec_res, offsets_res); else throw Exception("Illegal columns " + block.getByPosition(arguments[0]).column->getName() + " and " + block.getByPosition(arguments[1]).column->getName() + " of arguments of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } } } }; template class FunctionStringNumNumToString : public IFunction { public: /// Получить имя функции. String getName() const { return Name::get(); } /// Получить тип результата по типам аргументов. Если функция неприменима для данных аргументов - кинуть исключение. DataTypePtr getReturnType(const DataTypes & arguments) const { if (arguments.size() != 3) throw Exception("Number of arguments for function " + getName() + " doesn't match: passed " + Poco::NumberFormatter::format(arguments.size()) + ", should be 3.", ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH); if (!dynamic_cast(&*arguments[0]) && !dynamic_cast(&*arguments[0])) throw Exception("Illegal type " + arguments[0]->getName() + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); if (!arguments[1]->isNumeric() || !arguments[2]->isNumeric()) throw Exception("Illegal type " + (arguments[1]->isNumeric() ? arguments[2]->getName() : arguments[1]->getName()) + " of argument of function " + getName(), ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT); return new DataTypeString; } /// Выполнить функцию над блоком. void execute(Block & block, const ColumnNumbers & arguments, size_t result) { const ColumnPtr column_string = block.getByPosition(arguments[0]).column; const ColumnPtr column_start = block.getByPosition(arguments[1]).column; const ColumnPtr column_length = block.getByPosition(arguments[2]).column; if (!column_start->isConst() || !column_length->isConst()) throw Exception("2nd and 3rd arguments of function " + getName() + " must be constants."); FieldVisitorConvertToNumber converter; Field start_field = (*block.getByPosition(arguments[1]).column)[0]; Field length_field = (*block.getByPosition(arguments[2]).column)[0]; UInt64 start = boost::apply_visitor(converter, start_field); UInt64 length = boost::apply_visitor(converter, length_field); if (start == 0) throw Exception("Second argument of function substring must be greater than 0.", ErrorCodes::ARGUMENT_OUT_OF_BOUND); if (const ColumnString * col = dynamic_cast(&*column_string)) { ColumnString * col_res = new ColumnString; block.getByPosition(result).column = col_res; Impl::vector(dynamic_cast(col->getData()).getData(), col->getOffsets(), start, length, dynamic_cast(col_res->getData()).getData(), col_res->getOffsets()); } else if (const ColumnFixedString * col = dynamic_cast(&*column_string)) { ColumnFixedString * col_res = new ColumnFixedString(length); block.getByPosition(result).column = col_res; Impl::vector_fixed(dynamic_cast(col->getData()).getData(), col->getN(), start, length, dynamic_cast(col_res->getData()).getData()); } else if (const ColumnConstString * col = dynamic_cast(&*column_string)) { String res; Impl::constant(col->getData(), start, length, res); ColumnConstString * col_res = new ColumnConstString(col->size(), res); block.getByPosition(result).column = col_res; } else throw Exception("Illegal column " + block.getByPosition(arguments[0]).column->getName() + " of first argument of function " + getName(), ErrorCodes::ILLEGAL_COLUMN); } }; struct NameEmpty { static const char * get() { return "empty"; } }; struct NameNotEmpty { static const char * get() { return "notEmpty"; } }; struct NameLength { static const char * get() { return "length"; } }; struct NameLengthUTF8 { static const char * get() { return "lengthUTF8"; } }; struct NameLower { static const char * get() { return "lower"; } }; struct NameUpper { static const char * get() { return "upper"; } }; struct NameLowerUTF8 { static const char * get() { return "lowerUTF8"; } }; struct NameUpperUTF8 { static const char * get() { return "upperUTF8"; } }; struct NameReverse { static const char * get() { return "reverse"; } }; struct NameReverseUTF8 { static const char * get() { return "reverseUTF8"; } }; struct NameConcat { static const char * get() { return "concat"; } }; struct NameSubstring { static const char * get() { return "substring"; } }; struct NameSubstringUTF8 { static const char * get() { return "substringUTF8"; } }; typedef FunctionStringOrArrayToT, NameEmpty, UInt8> FunctionEmpty; typedef FunctionStringOrArrayToT, NameNotEmpty, UInt8> FunctionNotEmpty; typedef FunctionStringOrArrayToT FunctionLength; typedef FunctionStringOrArrayToT FunctionLengthUTF8; typedef FunctionStringToString, NameLower> FunctionLower; typedef FunctionStringToString, NameUpper> FunctionUpper; typedef FunctionStringToString, NameLowerUTF8> FunctionLowerUTF8; typedef FunctionStringToString, NameUpperUTF8> FunctionUpperUTF8; typedef FunctionStringToString FunctionReverse; typedef FunctionStringToString FunctionReverseUTF8; typedef FunctionStringStringToString FunctionConcat; typedef FunctionStringNumNumToString FunctionSubstring; typedef FunctionStringNumNumToString FunctionSubstringUTF8; }