thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-24 18:50:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update SelectStreamFactory.

Browse Source
This commit is contained in:
Nikolai Kochetov 2020-02-14 12:36:00 +03:00
parent bb6d6a37b3
commit 4cf65451de
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dbms/src/Interpreters/ClusterProxy/SelectStreamFactory.cpp
View File

@ -114,6 +114,7 @@ void SelectStreamFactory::createForShard(
Pipes & res) Pipes & res)
{ {
bool force_add_agg_info = processed_stage == QueryProcessingStage::WithMergeableState; bool force_add_agg_info = processed_stage == QueryProcessingStage::WithMergeableState;
bool add_totals_port = processed_stage == QueryProcessingStage::Complete;
auto modified_query_ast = query_ast->clone(); auto modified_query_ast = query_ast->clone();
if (has_virtual_shard_num_column) if (has_virtual_shard_num_column)
@ -135,6 +136,8 @@ void SelectStreamFactory::createForShard(
stream->setMainTable(main_table); stream->setMainTable(main_table);
auto source = std::make_shared<SourceFromInputStream>(std::move(stream), force_add_agg_info); auto source = std::make_shared<SourceFromInputStream>(std::move(stream), force_add_agg_info);
if (add_totals_port)
source->addTotalsPort(); source->addTotalsPort();
res.emplace_back(std::move(source)); res.emplace_back(std::move(source));
@ -279,6 +282,8 @@ void SelectStreamFactory::createForShard(
}; };
auto source = std::make_shared<SourceFromInputStream>("LazyShardWithLocalReplica", header, lazily_create_stream, force_add_agg_info); auto source = std::make_shared<SourceFromInputStream>("LazyShardWithLocalReplica", header, lazily_create_stream, force_add_agg_info);
if (add_totals_port)
source->addTotalsPort(); source->addTotalsPort();
res.emplace_back(std::move(source)); res.emplace_back(std::move(source));