mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 22:22:00 +00:00
2716df8ab2
Added: * IDataTypeDomain interface; * method DataTypeFactory::registerDataTypeDomain for registering domains; * DataTypeDomainWithSimpleSerialization domain base class with simple serialization/deserialization; * Concrete IPv4 and IPv6 domain implementations: DataTypeDomanIPv6 and DataTypeDomanIPv4; Updated: * IDataType text serialization/deserialization methods; * IDataType implementation to use domain for text serialization/deserialization; * Refactored implementation of the IPv4/IPv6 functions to use formatIPv4/v6 and parseIPv4/v6 from Common/formatIPv6.h; Tests: * Added test cases for IPv4 and IPv6 domains. * Updated IPv4/v6 functions tests to validate more cases; * Added performance tests for IPv4 and IPv6 related functions;
121 lines
3.0 KiB
C++
121 lines
3.0 KiB
C++
#include <Formats/JSONCompactRowOutputStream.h>
|
|
#include <Formats/FormatFactory.h>
|
|
#include <Formats/BlockOutputStreamFromRowOutputStream.h>
|
|
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
JSONCompactRowOutputStream::JSONCompactRowOutputStream(WriteBuffer & ostr_, const Block & sample_, const FormatSettings & settings_)
|
|
: JSONRowOutputStream(ostr_, sample_, settings_)
|
|
{
|
|
}
|
|
|
|
|
|
void JSONCompactRowOutputStream::writeField(const IColumn & column, const IDataType & type, size_t row_num)
|
|
{
|
|
type.serializeAsTextJSON(column, row_num, *ostr, settings);
|
|
++field_number;
|
|
}
|
|
|
|
|
|
void JSONCompactRowOutputStream::writeFieldDelimiter()
|
|
{
|
|
writeCString(", ", *ostr);
|
|
}
|
|
|
|
|
|
void JSONCompactRowOutputStream::writeRowStartDelimiter()
|
|
{
|
|
if (row_count > 0)
|
|
writeCString(",\n", *ostr);
|
|
writeCString("\t\t[", *ostr);
|
|
}
|
|
|
|
|
|
void JSONCompactRowOutputStream::writeRowEndDelimiter()
|
|
{
|
|
writeChar(']', *ostr);
|
|
field_number = 0;
|
|
++row_count;
|
|
}
|
|
|
|
|
|
void JSONCompactRowOutputStream::writeTotals()
|
|
{
|
|
if (totals)
|
|
{
|
|
writeCString(",\n", *ostr);
|
|
writeChar('\n', *ostr);
|
|
writeCString("\t\"totals\": [", *ostr);
|
|
|
|
size_t totals_columns = totals.columns();
|
|
for (size_t i = 0; i < totals_columns; ++i)
|
|
{
|
|
if (i != 0)
|
|
writeChar(',', *ostr);
|
|
|
|
const ColumnWithTypeAndName & column = totals.safeGetByPosition(i);
|
|
column.type->serializeAsTextJSON(*column.column.get(), 0, *ostr, settings);
|
|
}
|
|
|
|
writeChar(']', *ostr);
|
|
}
|
|
}
|
|
|
|
|
|
static void writeExtremesElement(const char * title, const Block & extremes, size_t row_num, WriteBuffer & ostr, const FormatSettings & settings)
|
|
{
|
|
writeCString("\t\t\"", ostr);
|
|
writeCString(title, ostr);
|
|
writeCString("\": [", ostr);
|
|
|
|
size_t extremes_columns = extremes.columns();
|
|
for (size_t i = 0; i < extremes_columns; ++i)
|
|
{
|
|
if (i != 0)
|
|
writeChar(',', ostr);
|
|
|
|
const ColumnWithTypeAndName & column = extremes.safeGetByPosition(i);
|
|
column.type->serializeAsTextJSON(*column.column.get(), row_num, ostr, settings);
|
|
}
|
|
|
|
writeChar(']', ostr);
|
|
}
|
|
|
|
void JSONCompactRowOutputStream::writeExtremes()
|
|
{
|
|
if (extremes)
|
|
{
|
|
writeCString(",\n", *ostr);
|
|
writeChar('\n', *ostr);
|
|
writeCString("\t\"extremes\":\n", *ostr);
|
|
writeCString("\t{\n", *ostr);
|
|
|
|
writeExtremesElement("min", extremes, 0, *ostr, settings);
|
|
writeCString(",\n", *ostr);
|
|
writeExtremesElement("max", extremes, 1, *ostr, settings);
|
|
|
|
writeChar('\n', *ostr);
|
|
writeCString("\t}", *ostr);
|
|
}
|
|
}
|
|
|
|
|
|
void registerOutputFormatJSONCompact(FormatFactory & factory)
|
|
{
|
|
factory.registerOutputFormat("JSONCompact", [](
|
|
WriteBuffer & buf,
|
|
const Block & sample,
|
|
const Context &,
|
|
const FormatSettings & format_settings)
|
|
{
|
|
return std::make_shared<BlockOutputStreamFromRowOutputStream>(
|
|
std::make_shared<JSONCompactRowOutputStream>(buf, sample, format_settings), sample);
|
|
});
|
|
}
|
|
|
|
}
|