mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
83 lines
2.6 KiB
C++
83 lines
2.6 KiB
C++
#include <DataTypes/DataTypeObject.h>
|
|
#include <DataTypes/DataTypeFactory.h>
|
|
#include <DataTypes/Serializations/SerializationObject.h>
|
|
|
|
#include <Parsers/IAST.h>
|
|
#include <Parsers/ASTLiteral.h>
|
|
#include <Parsers/ASTFunction.h>
|
|
#include <IO/Operators.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
|
|
extern const int UNEXPECTED_AST_STRUCTURE;
|
|
}
|
|
|
|
DataTypeObject::DataTypeObject(const String & schema_format_, bool is_nullable_)
|
|
: schema_format(Poco::toLower(schema_format_))
|
|
, is_nullable(is_nullable_)
|
|
{
|
|
}
|
|
|
|
bool DataTypeObject::equals(const IDataType & rhs) const
|
|
{
|
|
if (const auto * object = typeid_cast<const DataTypeObject *>(&rhs))
|
|
return schema_format == object->schema_format && is_nullable == object->is_nullable;
|
|
return false;
|
|
}
|
|
|
|
SerializationPtr DataTypeObject::doGetDefaultSerialization() const
|
|
{
|
|
return getObjectSerialization(schema_format);
|
|
}
|
|
|
|
String DataTypeObject::doGetName() const
|
|
{
|
|
WriteBufferFromOwnString out;
|
|
if (is_nullable)
|
|
out << "Object(Nullable(" << quote << schema_format << "))";
|
|
else
|
|
out << "Object(" << quote << schema_format << ")";
|
|
return out.str();
|
|
}
|
|
|
|
static DataTypePtr create(const ASTPtr & arguments)
|
|
{
|
|
if (!arguments || arguments->children.size() != 1)
|
|
throw Exception(ErrorCodes::NUMBER_OF_ARGUMENTS_DOESNT_MATCH,
|
|
"Object data type family must have one argument - name of schema format");
|
|
|
|
ASTPtr schema_argument = arguments->children[0];
|
|
bool is_nullable = false;
|
|
|
|
if (const auto * func = schema_argument->as<ASTFunction>())
|
|
{
|
|
if (func->name != "Nullable" || func->arguments->children.size() != 1)
|
|
throw Exception(ErrorCodes::UNEXPECTED_AST_STRUCTURE,
|
|
"Expected 'Nullable(<schema_name>)' as parameter for type Object (function: {})", func->name);
|
|
|
|
schema_argument = func->arguments->children[0];
|
|
is_nullable = true;
|
|
}
|
|
|
|
const auto * literal = schema_argument->as<ASTLiteral>();
|
|
if (!literal || literal->value.getType() != Field::Types::String)
|
|
throw Exception(ErrorCodes::UNEXPECTED_AST_STRUCTURE,
|
|
"Object data type family must have a const string as its schema name parameter");
|
|
|
|
return std::make_shared<DataTypeObject>(literal->value.get<const String &>(), is_nullable);
|
|
}
|
|
|
|
void registerDataTypeObject(DataTypeFactory & factory)
|
|
{
|
|
factory.registerDataType("Object", create);
|
|
factory.registerSimpleDataType("JSON",
|
|
[] { return std::make_shared<DataTypeObject>("JSON", false); },
|
|
DataTypeFactory::CaseInsensitive);
|
|
}
|
|
|
|
}
|