mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 06:00:49 +00:00
781f1597f2
There are two possible ways to pass the SETTINGS clause: - SELECT 1 FORMAT Null SETTINGS max_block_size = 1 - SELECT 1 SETTINGS max_block_size = 1[ FORMAT Null] And when the SETTINGS goes after FORMAT it is parsed in ParserQueryWithOutput, and these settings are not applied for the query from the ASTSelectQuery::initSettings() So propagate settings from the ParserQueryWithOutput to the ASTSelectQuery settings.
57 lines
1.6 KiB
C++
57 lines
1.6 KiB
C++
#include <Common/SettingsChanges.h>
|
|
#include <Parsers/QueryWithOutputSettingsPushDownVisitor.h>
|
|
#include <Parsers/ASTSelectWithUnionQuery.h>
|
|
#include <Parsers/ASTSelectQuery.h>
|
|
#include <Parsers/ASTSetQuery.h>
|
|
#include <Parsers/ASTSubquery.h>
|
|
|
|
#include <iterator>
|
|
#include <algorithm>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
bool QueryWithOutputSettingsPushDownMatcher::needChildVisit(ASTPtr & node, const ASTPtr & child)
|
|
{
|
|
if (node->as<ASTSelectWithUnionQuery>())
|
|
return true;
|
|
if (node->as<ASTSubquery>())
|
|
return true;
|
|
if (child->as<ASTSelectQuery>())
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
void QueryWithOutputSettingsPushDownMatcher::visit(ASTPtr & ast, Data & data)
|
|
{
|
|
if (auto * select_query = ast->as<ASTSelectQuery>())
|
|
visit(*select_query, ast, data);
|
|
}
|
|
|
|
void QueryWithOutputSettingsPushDownMatcher::visit(ASTSelectQuery & select_query, ASTPtr &, Data & data)
|
|
{
|
|
ASTPtr select_settings_ast = select_query.settings();
|
|
if (!select_settings_ast)
|
|
{
|
|
select_query.setExpression(ASTSelectQuery::Expression::SETTINGS, data.settings_ast->clone());
|
|
return;
|
|
}
|
|
|
|
SettingsChanges & select_settings = select_settings_ast->as<ASTSetQuery &>().changes;
|
|
SettingsChanges & settings = data.settings_ast->as<ASTSetQuery &>().changes;
|
|
|
|
for (auto & setting : settings)
|
|
{
|
|
auto it = std::find_if(select_settings.begin(), select_settings.end(), [&](auto & select_setting)
|
|
{
|
|
return select_setting.name == setting.name;
|
|
});
|
|
if (it == select_settings.end())
|
|
select_settings.push_back(setting);
|
|
else
|
|
it->value = setting.value;
|
|
}
|
|
}
|
|
|
|
}
|