2019-02-19 18:41:18 +00:00
|
|
|
#include <cmath>
|
|
|
|
|
|
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
#include <IO/WriteBufferFromString.h>
|
|
|
|
#include <Processors/Formats/Impl/VerticalRowOutputFormat.h>
|
|
|
|
#include <Formats/FormatFactory.h>
|
|
|
|
#include <Common/UTF8Helpers.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
VerticalRowOutputFormat::VerticalRowOutputFormat(
|
2019-04-09 09:25:57 +00:00
|
|
|
WriteBuffer & out_, const Block & header, const FormatSettings & format_settings)
|
|
|
|
: IRowOutputFormat(header, out_), format_settings(format_settings)
|
2019-02-19 18:41:18 +00:00
|
|
|
{
|
|
|
|
auto & sample = getPort(PortKind::Main).getHeader();
|
|
|
|
size_t columns = sample.columns();
|
|
|
|
|
|
|
|
using Widths = std::vector<size_t>;
|
|
|
|
Widths name_widths(columns);
|
|
|
|
size_t max_name_width = 0;
|
|
|
|
|
|
|
|
String serialized_value;
|
|
|
|
|
|
|
|
for (size_t i = 0; i < columns; ++i)
|
|
|
|
{
|
|
|
|
/// Note that number of code points is just a rough approximation of visible string width.
|
|
|
|
const String & name = sample.getByPosition(i).name;
|
|
|
|
|
|
|
|
name_widths[i] = UTF8::computeWidth(reinterpret_cast<const UInt8 *>(name.data()), name.size());
|
|
|
|
|
|
|
|
if (name_widths[i] > max_name_width)
|
|
|
|
max_name_width = name_widths[i];
|
|
|
|
}
|
|
|
|
|
|
|
|
names_and_paddings.resize(columns);
|
|
|
|
for (size_t i = 0; i < columns; ++i)
|
|
|
|
{
|
2019-04-05 12:32:47 +00:00
|
|
|
WriteBufferFromString buf(names_and_paddings[i]);
|
|
|
|
writeString(sample.getByPosition(i).name, buf);
|
|
|
|
writeCString(": ", buf);
|
2019-02-19 18:41:18 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < columns; ++i)
|
|
|
|
{
|
|
|
|
size_t new_size = max_name_width - name_widths[i] + names_and_paddings[i].size();
|
|
|
|
names_and_paddings[i].resize(new_size, ' ');
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeField(const IColumn & column, const IDataType & type, size_t row_num)
|
|
|
|
{
|
|
|
|
if (row_number > format_settings.pretty.max_rows)
|
|
|
|
return;
|
|
|
|
|
|
|
|
writeString(names_and_paddings[field_number], out);
|
|
|
|
writeValue(column, type, row_num);
|
|
|
|
writeChar('\n', out);
|
|
|
|
|
|
|
|
++field_number;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeValue(const IColumn & column, const IDataType & type, size_t row_num) const
|
|
|
|
{
|
|
|
|
type.serializeAsText(column, row_num, out, format_settings);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeRowStartDelimiter()
|
|
|
|
{
|
|
|
|
++row_number;
|
|
|
|
|
|
|
|
if (row_number > format_settings.pretty.max_rows)
|
|
|
|
return;
|
|
|
|
|
|
|
|
writeCString("Row ", out);
|
|
|
|
writeIntText(row_number, out);
|
|
|
|
writeCString(":\n", out);
|
|
|
|
|
|
|
|
size_t width = log10(row_number + 1) + 1 + strlen("Row :");
|
|
|
|
for (size_t i = 0; i < width; ++i)
|
|
|
|
writeCString("─", out);
|
|
|
|
writeChar('\n', out);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeRowBetweenDelimiter()
|
|
|
|
{
|
|
|
|
if (row_number > format_settings.pretty.max_rows)
|
|
|
|
return;
|
|
|
|
|
|
|
|
writeCString("\n", out);
|
|
|
|
field_number = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeSuffix()
|
|
|
|
{
|
|
|
|
if (row_number > format_settings.pretty.max_rows)
|
|
|
|
{
|
|
|
|
writeCString("Showed first ", out);
|
|
|
|
writeIntText(format_settings.pretty.max_rows, out);
|
|
|
|
writeCString(".\n", out);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeBeforeTotals()
|
|
|
|
{
|
|
|
|
writeCString("\n", out);
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeBeforeExtremes()
|
|
|
|
{
|
|
|
|
if (!was_totals_written)
|
|
|
|
writeCString("\n", out);
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeMinExtreme(const Columns & columns, size_t row_num)
|
|
|
|
{
|
|
|
|
writeSpecialRow(columns, row_num, PortKind::Totals, "Min");
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeMaxExtreme(const Columns & columns, size_t row_num)
|
|
|
|
{
|
|
|
|
writeSpecialRow(columns, row_num, PortKind::Totals, "Max");
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeTotals(const Columns & columns, size_t row_num)
|
|
|
|
{
|
|
|
|
writeSpecialRow(columns, row_num, PortKind::Totals, "Totals");
|
|
|
|
was_totals_written = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void VerticalRowOutputFormat::writeSpecialRow(const Columns & columns, size_t row_num, PortKind port_kind, const char * title)
|
|
|
|
{
|
|
|
|
writeCString("\n", out);
|
|
|
|
|
|
|
|
row_number = 0;
|
|
|
|
field_number = 0;
|
|
|
|
|
|
|
|
auto & header = getPort(port_kind).getHeader();
|
|
|
|
size_t num_columns = columns.size();
|
|
|
|
|
|
|
|
writeCString(title, out);
|
|
|
|
writeCString(":\n", out);
|
|
|
|
|
|
|
|
size_t width = strlen(title) + 1;
|
|
|
|
for (size_t i = 0; i < width; ++i)
|
|
|
|
writeCString("─", out);
|
|
|
|
writeChar('\n', out);
|
|
|
|
|
|
|
|
for (size_t i = 0; i < num_columns; ++i)
|
|
|
|
{
|
|
|
|
if (i != 0)
|
|
|
|
writeFieldDelimiter();
|
|
|
|
|
|
|
|
auto & col = header.getByPosition(i);
|
|
|
|
writeField(*columns[i], *col.type, row_num);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void registerOutputFormatProcessorVertical(FormatFactory & factory)
|
|
|
|
{
|
|
|
|
factory.registerOutputFormatProcessor("Vertical", [](
|
|
|
|
WriteBuffer & buf,
|
|
|
|
const Block & sample,
|
|
|
|
const Context &,
|
|
|
|
const FormatSettings & settings)
|
|
|
|
{
|
|
|
|
return std::make_shared<VerticalRowOutputFormat>(buf, sample, settings);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|