mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
114 lines
4.0 KiB
C++
114 lines
4.0 KiB
C++
#include <Processors/Formats/Impl/MySQLOutputFormat.h>
|
|
#include <Core/MySQL/PacketsGeneric.h>
|
|
#include <Core/MySQL/PacketsProtocolText.h>
|
|
#include <Formats/FormatFactory.h>
|
|
#include <Formats/FormatSettings.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Interpreters/ProcessList.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
using namespace MySQLProtocol;
|
|
using namespace MySQLProtocol::Generic;
|
|
using namespace MySQLProtocol::ProtocolText;
|
|
|
|
|
|
MySQLOutputFormat::MySQLOutputFormat(WriteBuffer & out_, const Block & header_, const FormatSettings & settings_)
|
|
: IOutputFormat(header_, out_)
|
|
, client_capabilities(settings_.mysql_wire.client_capabilities)
|
|
{
|
|
/// MySQlWire is a special format that is usually used as output format for MySQL protocol connections.
|
|
/// In this case we have a correct `sequence_id` stored in `settings_.mysql_wire`.
|
|
/// But it's also possible to specify MySQLWire as output format for clickhouse-client or clickhouse-local.
|
|
/// There is no `sequence_id` stored in `settings_.mysql_wire` in this case, so we create a dummy one.
|
|
sequence_id = settings_.mysql_wire.sequence_id ? settings_.mysql_wire.sequence_id : &dummy_sequence_id;
|
|
|
|
const auto & header = getPort(PortKind::Main).getHeader();
|
|
data_types = header.getDataTypes();
|
|
|
|
serializations.reserve(data_types.size());
|
|
for (const auto & type : data_types)
|
|
serializations.emplace_back(type->getDefaultSerialization());
|
|
|
|
packet_endpoint = std::make_shared<MySQLProtocol::PacketEndpoint>(out, *sequence_id);
|
|
}
|
|
|
|
void MySQLOutputFormat::setContext(ContextPtr context_)
|
|
{
|
|
context = context_;
|
|
}
|
|
|
|
void MySQLOutputFormat::writePrefix()
|
|
{
|
|
const auto & header = getPort(PortKind::Main).getHeader();
|
|
|
|
if (header.columns())
|
|
{
|
|
packet_endpoint->sendPacket(LengthEncodedNumber(header.columns()));
|
|
|
|
for (size_t i = 0; i < header.columns(); ++i)
|
|
{
|
|
const auto & column_name = header.getColumnsWithTypeAndName()[i].name;
|
|
packet_endpoint->sendPacket(getColumnDefinition(column_name, data_types[i]->getTypeId()));
|
|
}
|
|
|
|
if (!(client_capabilities & Capability::CLIENT_DEPRECATE_EOF))
|
|
{
|
|
packet_endpoint->sendPacket(EOFPacket(0, 0));
|
|
}
|
|
}
|
|
}
|
|
|
|
void MySQLOutputFormat::consume(Chunk chunk)
|
|
{
|
|
for (size_t i = 0; i < chunk.getNumRows(); ++i)
|
|
{
|
|
ProtocolText::ResultSetRow row_packet(serializations, chunk.getColumns(), static_cast<int>(i));
|
|
packet_endpoint->sendPacket(row_packet);
|
|
}
|
|
}
|
|
|
|
void MySQLOutputFormat::finalizeImpl()
|
|
{
|
|
size_t affected_rows = 0;
|
|
std::string human_readable_info;
|
|
if (QueryStatusPtr process_list_elem = getContext()->getProcessListElement())
|
|
{
|
|
CurrentThread::finalizePerformanceCounters();
|
|
QueryStatusInfo info = process_list_elem->getInfo();
|
|
affected_rows = info.written_rows;
|
|
human_readable_info = fmt::format(
|
|
"Read {} rows, {} in {} sec., {} rows/sec., {}/sec.",
|
|
info.read_rows, ReadableSize(info.read_bytes), info.elapsed_seconds,
|
|
static_cast<size_t>(info.read_rows / info.elapsed_seconds),
|
|
ReadableSize(info.read_bytes / info.elapsed_seconds));
|
|
}
|
|
|
|
const auto & header = getPort(PortKind::Main).getHeader();
|
|
if (header.columns() == 0)
|
|
packet_endpoint->sendPacket(OKPacket(0x0, client_capabilities, affected_rows, 0, 0, "", human_readable_info), true);
|
|
else if (client_capabilities & CLIENT_DEPRECATE_EOF)
|
|
packet_endpoint->sendPacket(OKPacket(0xfe, client_capabilities, affected_rows, 0, 0, "", human_readable_info), true);
|
|
else
|
|
packet_endpoint->sendPacket(EOFPacket(0, 0), true);
|
|
}
|
|
|
|
void MySQLOutputFormat::flush()
|
|
{
|
|
packet_endpoint->out->next();
|
|
}
|
|
|
|
void registerOutputFormatMySQLWire(FormatFactory & factory)
|
|
{
|
|
factory.registerOutputFormat(
|
|
"MySQLWire",
|
|
[](WriteBuffer & buf,
|
|
const Block & sample,
|
|
const RowOutputFormatParams &,
|
|
const FormatSettings & settings) { return std::make_shared<MySQLOutputFormat>(buf, sample, settings); });
|
|
}
|
|
|
|
}
|