#pragma once #include #include #include #include #include #include #include #include #include #include #include namespace DB { using Poco::SharedPtr; class IAggregateFunction; /** Discriminated union из нескольких типов. * Сделан для замены boost::variant: * является не обобщённым, * зато несколько более эффективным, и более простым. * * Используется для представления единичного значения одного из нескольких типов в оперативке. * Внимание! Предпочтительно вместо единичных значений хранить кусочки столбцов. См. Column.h */ class Field { public: struct Types { /// Идентификатор типа. enum Which { Null = 0, UInt64 = 1, Int64 = 2, Float64 = 3, /// не POD типы String = 16, AggregateFunction = 17, /// Состояние агрегатной функции Array = 18, }; static const int MIN_NON_POD = 16; static const char * toString(Which which) { switch (which) { case Null: return "Null"; case UInt64: return "UInt64"; case Int64: return "Int64"; case Float64: return "Float64"; case String: return "String"; case AggregateFunction: return "AggregateFunctionPtr"; case Array: return "Array"; default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } }; /// Позволяет получить идентификатор для типа или наоборот. template struct TypeToEnum; template struct EnumToType; Field() : which(Types::Null) { // std::cerr << "Field()" << std::endl; } /** Не смотря на наличие шаблонного конструктора, этот конструктор всё-равно нужен, * так как при его отсутствии, не смотря на наличие шаблонного конструктора, * компилятор всё-равно сгенерирует конструктор по-умолчанию. */ Field(const Field & rhs) { // std::cerr << this << " Field::Field(const Field &)" << std::endl; create(rhs); } Field & operator= (const Field & rhs) { // std::cerr << this << " Field::operator=(const Field &)" << std::endl; destroy(); create(rhs); return *this; } template Field(const T & rhs) : which(Types::Null) { // std::cerr << this << " Field::Field(" << Types::toString(TypeToEnum::value) << ")" << std::endl; create(rhs); } /// Создать строку inplace. Field(const char * data, size_t size) { create(data, size); } Field(const unsigned char * data, size_t size) { create(data, size); } void assignString(const char * data, size_t size) { destroy(); create(data, size); } void assignString(const unsigned char * data, size_t size) { destroy(); create(data, size); } template Field & operator= (const T & rhs) { // std::cerr << this << " Field::operator=(" << Types::toString(TypeToEnum::value) << ")" << std::endl; destroy(); create(rhs); return *this; } ~Field() { // std::cerr << this << " Field::~Field()" << std::endl; destroy(); } Types::Which getType() const { return which; } const char * getTypeName() const { return Types::toString(which); } template T & get() { typedef typename std::tr1::remove_reference::type TWithoutRef; TWithoutRef * __attribute__((__may_alias__)) ptr = reinterpret_cast(storage); return *ptr; }; template const T & get() const { typedef typename std::tr1::remove_reference::type TWithoutRef; const TWithoutRef * __attribute__((__may_alias__)) ptr = reinterpret_cast(storage); return *ptr; }; template T & safeGet() { const Types::Which requested = TypeToEnum::type>::type>::value; if (which != requested) throw Exception("Bad get: has " + std::string(getTypeName()) + ", requested " + std::string(Types::toString(requested)), ErrorCodes::BAD_GET); return get(); } template const T & safeGet() const { const Types::Which requested = TypeToEnum::type>::type>::value; if (which != requested) throw Exception("Bad get: has " + std::string(getTypeName()) + ", requested " + std::string(Types::toString(requested)), ErrorCodes::BAD_GET); return get(); } bool operator< (const Field & rhs) const { if (which < rhs.which) return true; if (which > rhs.which) return false; switch (which) { case Types::Null: return get() < rhs.get(); case Types::UInt64: return get() < rhs.get(); case Types::Int64: return get() < rhs.get(); case Types::Float64: return get() < rhs.get(); case Types::String: return get() < rhs.get(); case Types::AggregateFunction: return get() < rhs.get(); case Types::Array: return get() < rhs.get(); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } bool operator> (const Field & rhs) const { return rhs < *this; } bool operator<= (const Field & rhs) const { if (which < rhs.which) return true; if (which > rhs.which) return false; switch (which) { case Types::Null: return get() <= rhs.get(); case Types::UInt64: return get() <= rhs.get(); case Types::Int64: return get() <= rhs.get(); case Types::Float64: return get() <= rhs.get(); case Types::String: return get() <= rhs.get(); case Types::AggregateFunction: return get() <= rhs.get(); case Types::Array: return get() <= rhs.get(); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } bool operator>= (const Field & rhs) const { return rhs <= *this; } bool operator== (const Field & rhs) const { if (which != rhs.which) return false; switch (which) { case Types::Null: return get() == rhs.get(); case Types::UInt64: return get() == rhs.get(); case Types::Int64: return get() == rhs.get(); case Types::Float64: return get() == rhs.get(); case Types::String: return get() == rhs.get(); case Types::AggregateFunction: return get() == rhs.get(); case Types::Array: return get() == rhs.get(); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } bool operator!= (const Field & rhs) const { return !(*this == rhs); } typedef SharedPtr AggregateFunctionPtr; typedef std::vector Array; private: static const size_t storage_size = 24; BOOST_STATIC_ASSERT(storage_size >= sizeof(Null)); BOOST_STATIC_ASSERT(storage_size >= sizeof(UInt64)); BOOST_STATIC_ASSERT(storage_size >= sizeof(Int64)); BOOST_STATIC_ASSERT(storage_size >= sizeof(Float64)); BOOST_STATIC_ASSERT(storage_size >= sizeof(String)); BOOST_STATIC_ASSERT(storage_size >= sizeof(AggregateFunctionPtr)); BOOST_STATIC_ASSERT(storage_size >= sizeof(Array)); char storage[storage_size]; Types::Which which; template void create(const T & x) { which = TypeToEnum::value; // std::cerr << this << " Creating " << getTypeName() << std::endl; T * __attribute__((__may_alias__)) ptr = reinterpret_cast(storage); new (ptr) T(x); } void create(const Null & x) { which = Types::Null; // std::cerr << this << " Creating " << getTypeName() << std::endl; } void create(const Field & x) { // std::cerr << this << " Creating Field" << std::endl; switch (x.which) { case Types::Null: create(Null()); break; case Types::UInt64: create(x.get()); break; case Types::Int64: create(x.get()); break; case Types::Float64: create(x.get()); break; case Types::String: create(x.get()); break; case Types::AggregateFunction: create(x.get()); break; case Types::Array: create(x.get()); break; } } void create(const char * data, size_t size) { which = Types::String; String * __attribute__((__may_alias__)) ptr = reinterpret_cast(storage); new (ptr) String(data, size); } void create(const unsigned char * data, size_t size) { create(reinterpret_cast(data), size); } __attribute__((__always_inline__)) void destroy() { // std::cerr << this << " Destroying " << getTypeName() << std::endl; if (which < Types::MIN_NON_POD) return; switch (which) { case Types::String: destroy(); break; case Types::AggregateFunction: destroy(); break; case Types::Array: destroy(); break; default: break; } } template void destroy() { T * __attribute__((__may_alias__)) ptr = reinterpret_cast(storage); ptr->~T(); } }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::Null; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::UInt64; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::Int64; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::Float64; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::String; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::AggregateFunction; }; template <> struct Field::TypeToEnum { static const Types::Which value = Types::Array; }; template <> struct Field::EnumToType { typedef Null Type; }; template <> struct Field::EnumToType { typedef UInt64 Type; }; template <> struct Field::EnumToType { typedef Int64 Type; }; template <> struct Field::EnumToType { typedef Float64 Type; }; template <> struct Field::EnumToType { typedef String Type; }; template <> struct Field::EnumToType { typedef AggregateFunctionPtr Type; }; template <> struct Field::EnumToType { typedef Array Type; }; template T get(const Field & field) { return field.template get(); } template T get(Field & field) { return field.template get(); } template T safeGet(const Field & field) { return field.template safeGet(); } template T safeGet(Field & field) { return field.template safeGet(); } /** StaticVisitor (его наследники) - класс с перегруженными для разных типов операторами (). * Вызвать visitor для field можно с помощью функции apply_visitor. * Также поддерживается visitor, в котором оператор () принимает два аргумента. */ template struct StaticVisitor { typedef R ResultType; }; template typename Visitor::ResultType apply_visitor_impl(Visitor & visitor, F & field) { switch (field.getType()) { case Field::Types::Null: return visitor(field.template get()); case Field::Types::UInt64: return visitor(field.template get()); case Field::Types::Int64: return visitor(field.template get()); case Field::Types::Float64: return visitor(field.template get()); case Field::Types::String: return visitor(field.template get()); case Field::Types::AggregateFunction: return visitor(field.template get()); case Field::Types::Array: return visitor(field.template get()); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } /** Эти штуки нужны, чтобы принимать временный объект по константной ссылке. * В шаблон выше, типы форвардятся уже с const-ом. */ template typename Visitor::ResultType apply_visitor(const Visitor & visitor, Field & field) { return apply_visitor_impl(visitor, field); } template typename Visitor::ResultType apply_visitor(const Visitor & visitor, const Field & field) { return apply_visitor_impl(visitor, field); } template typename Visitor::ResultType apply_visitor(Visitor & visitor, Field & field) { return apply_visitor_impl(visitor, field); } template typename Visitor::ResultType apply_visitor(Visitor & visitor, const Field & field) { return apply_visitor_impl(visitor, field); } template typename Visitor::ResultType apply_binary_visitor_impl2(Visitor & visitor, F1 & field1, F2 & field2) { switch (field2.getType()) { case Field::Types::Null: return visitor(field1, field2.template get()); case Field::Types::UInt64: return visitor(field1, field2.template get()); case Field::Types::Int64: return visitor(field1, field2.template get()); case Field::Types::Float64: return visitor(field1, field2.template get()); case Field::Types::String: return visitor(field1, field2.template get()); case Field::Types::AggregateFunction: return visitor(field1, field2.template get()); case Field::Types::Array: return visitor(field1, field2.template get()); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } template typename Visitor::ResultType apply_binary_visitor_impl1(Visitor & visitor, F1 & field1, F2 & field2) { switch (field1.getType()) { case Field::Types::Null: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::UInt64: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::Int64: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::Float64: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::String: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::AggregateFunction: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); case Field::Types::Array: return apply_binary_visitor_impl2(visitor, field1.template get(), field2); default: throw Exception("Bad type of Field", ErrorCodes::BAD_TYPE_OF_FIELD); } } template typename Visitor::ResultType apply_visitor(Visitor & visitor, Field & field1, Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(Visitor & visitor, Field & field1, const Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(Visitor & visitor, const Field & field1, Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(Visitor & visitor, const Field & field1, const Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(const Visitor & visitor, Field & field1, Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(const Visitor & visitor, Field & field1, const Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(const Visitor & visitor, const Field & field1, Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } template typename Visitor::ResultType apply_visitor(const Visitor & visitor, const Field & field1, const Field & field2) { return apply_binary_visitor_impl1(visitor, field1, field2); } typedef std::vector Array; /// Значение типа "массив" template <> struct TypeName { static std::string get() { return "Array"; } }; /** Возвращает строковый дамп типа */ class FieldVisitorDump : public StaticVisitor { public: String operator() (const Null & x) const { return "NULL"; } String operator() (const UInt64 & x) const { return "UInt64_" + Poco::NumberFormatter::format(x); } String operator() (const Int64 & x) const { return "Int64_" + Poco::NumberFormatter::format(x); } String operator() (const Float64 & x) const { return "Float64_" + Poco::NumberFormatter::format(x); } String operator() (const SharedPtr & x) const { return "AggregateFunction"; } String operator() (const String & x) const { std::stringstream s; s << mysqlxx::quote << x; return s.str(); } String operator() (const Array & x) const { std::stringstream s; s << "Array_["; for (Array::const_iterator it = x.begin(); it != x.end(); ++it) { if (it != x.begin()) s << ", "; s << apply_visitor(FieldVisitorDump(), *it); } s << "]"; return s.str(); } }; /** Выводит текстовое представление типа, как литерала в SQL запросе */ class FieldVisitorToString : public StaticVisitor { public: String operator() (const Null & x) const { return "NULL"; } String operator() (const UInt64 & x) const { return Poco::NumberFormatter::format(x); } String operator() (const Int64 & x) const { return Poco::NumberFormatter::format(x); } String operator() (const Float64 & x) const { return Poco::NumberFormatter::format(x); } String operator() (const SharedPtr & x) const { return "AggregateFunction"; } String operator() (const String & x) const { std::stringstream s; s << mysqlxx::quote << x; return s.str(); } String operator() (const Array & x) const { std::stringstream s; FieldVisitorToString visitor; s << "["; for (Array::const_iterator it = x.begin(); it != x.end(); ++it) { if (it != x.begin()) s << ", "; s << apply_visitor(visitor, *it); } s << "]"; return s.str(); } }; /** Числовой тип преобразует в указанный. */ template class FieldVisitorConvertToNumber : public StaticVisitor { public: T operator() (const Null & x) const { throw Exception("Cannot convert NULL to " + TypeName::get(), ErrorCodes::CANNOT_CONVERT_TYPE); } T operator() (const String & x) const { throw Exception("Cannot convert String to " + TypeName::get(), ErrorCodes::CANNOT_CONVERT_TYPE); } T operator() (const Array & x) const { throw Exception("Cannot convert Array to " + TypeName::get(), ErrorCodes::CANNOT_CONVERT_TYPE); } T operator() (const SharedPtr & x) const { throw Exception("Cannot convert AggregateFunctionPtr to " + TypeName::get(), ErrorCodes::CANNOT_CONVERT_TYPE); } T operator() (const UInt64 & x) const { return x; } T operator() (const Int64 & x) const { return x; } T operator() (const Float64 & x) const { return x; } }; class FieldVisitorLess : public StaticVisitor { public: template bool operator() (const T &, const U &) const { return false; } template bool operator() (const T & lhs, const T & rhs) const { return lhs < rhs; } }; class FieldVisitorGreater : public StaticVisitor { public: template bool operator() (const T &, const U &) const { return false; } template bool operator() (const T & lhs, const T & rhs) const { return lhs > rhs; } }; template struct NearestFieldType; template <> struct NearestFieldType { typedef UInt64 Type; }; template <> struct NearestFieldType { typedef UInt64 Type; }; template <> struct NearestFieldType { typedef UInt64 Type; }; template <> struct NearestFieldType { typedef UInt64 Type; }; template <> struct NearestFieldType { typedef Int64 Type; }; template <> struct NearestFieldType { typedef Int64 Type; }; template <> struct NearestFieldType { typedef Int64 Type; }; template <> struct NearestFieldType { typedef Int64 Type; }; template <> struct NearestFieldType { typedef Float64 Type; }; template <> struct NearestFieldType { typedef Float64 Type; }; template <> struct NearestFieldType { typedef String Type; }; template <> struct NearestFieldType { typedef Array Type; }; template <> struct NearestFieldType { typedef UInt64 Type; }; } /// Заглушки, чтобы DBObject-ы с полем типа Array компилировались. namespace mysqlxx { inline std::ostream & operator<< (mysqlxx::EscapeManipResult res, const DB::Array & value) { throw Poco::Exception("Cannot escape Array with mysqlxx::escape."); } inline std::ostream & operator<< (mysqlxx::QuoteManipResult res, const DB::Array & value) { throw Poco::Exception("Cannot quote Array with mysqlxx::quote."); } inline std::istream & operator>> (mysqlxx::UnEscapeManipResult res, DB::Array & value) { throw Poco::Exception("Cannot unescape Array with mysqlxx::unescape."); } inline std::istream & operator>> (mysqlxx::UnQuoteManipResult res, DB::Array & value) { throw Poco::Exception("Cannot unquote Array with mysqlxx::unquote."); } } namespace DB { class ReadBuffer; class WriteBuffer; inline void readBinary(Array & x, ReadBuffer & buf) { throw Exception("Cannot read Array.", ErrorCodes::NOT_IMPLEMENTED); } inline void readText(Array & x, ReadBuffer & buf) { throw Exception("Cannot read Array.", ErrorCodes::NOT_IMPLEMENTED); } inline void readQuoted(Array & x, ReadBuffer & buf) { throw Exception("Cannot read Array.", ErrorCodes::NOT_IMPLEMENTED); } inline void writeBinary(const Array & x, WriteBuffer & buf) { throw Exception("Cannot write Array.", ErrorCodes::NOT_IMPLEMENTED); } inline void writeText(const Array & x, WriteBuffer & buf) { throw Exception("Cannot write Array.", ErrorCodes::NOT_IMPLEMENTED); } inline void writeQuoted(const Array & x, WriteBuffer & buf) { throw Exception("Cannot write Array.", ErrorCodes::NOT_IMPLEMENTED); } }