thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-22 23:52:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make sure that remote query has non-zero columns

Browse Source
This commit is contained in:
Alexander Gololobov 2023-07-04 23:45:49 +02:00
parent 627e924a1b
commit 6d712f595a
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/QueryPipeline/RemoteQueryExecutor.cpp
View File

@ -48,7 +48,9 @@ RemoteQueryExecutor::RemoteQueryExecutor(
: header(header_), query(query_), context(context_), scalars(scalars_)
, external_tables(external_tables_), stage(stage_)
, extension(extension_)
{}
{
assertHeaderIsNotEmpty(header);
}
RemoteQueryExecutor::RemoteQueryExecutor(
Connection & connection,
@ -91,6 +93,8 @@ RemoteQueryExecutor::RemoteQueryExecutor(
, scalars(scalars_), external_tables(external_tables_), stage(stage_)
, extension(extension_)
{
assertHeaderIsNotEmpty(header);
create_connections = [this, connections_, throttler, extension_](AsyncCallback) mutable {
auto res = std::make_unique<MultiplexedConnections>(std::move(connections_), context->getSettingsRef(), throttler);
if (extension_ && extension_->replica_info)
@ -108,6 +112,8 @@ RemoteQueryExecutor::RemoteQueryExecutor(
, scalars(scalars_), external_tables(external_tables_), stage(stage_)
, extension(extension_)
{
assertHeaderIsNotEmpty(header);
create_connections = [this, pool, throttler, extension_](AsyncCallback async_callback)->std::unique_ptr<IConnections>
{
const Settings & current_settings = context->getSettingsRef();
@ -754,4 +760,10 @@ bool RemoteQueryExecutor::hasThrownException() const
return got_exception_from_replica || got_unknown_packet_from_replica;
}
void RemoteQueryExecutor::assertHeaderIsNotEmpty(const Block & header)
{
if (header.columns() == 0)
throw Exception(ErrorCodes::LOGICAL_ERROR, "Remote query executor must not have header without columns");
}
}

4
src/QueryPipeline/RemoteQueryExecutor.h
View File

@ -304,6 +304,10 @@ private:
/// Reads packet by packet
Block readPackets();
/// Block does not support passing 0 columns but non-zero rows.
/// We need to ensure that this is not happening.
static void assertHeaderIsNotEmpty(const Block & header);
};
}