thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-21 15:12:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

dictGet* with table name

Browse Source
This commit is contained in:
Amos Bird 2019-12-22 01:48:33 +08:00 committed by Vitaly Baranov
parent bc5f68f1ee
commit ce3c53a00a
5 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Interpreters/ActionsVisitor.cpp
View File

@ -471,7 +471,7 @@ void ActionsMatcher::visit(const ASTFunction & node, const ASTPtr & ast, Data &
argument_types.push_back(column.type);
argument_names.push_back(column.name);
}
else if (identifier && node.name == "joinGet" && arg == 0)
else if (identifier && functionIsJoinGetOrDictGet(node.name) && arg == 0)
{
auto table_id = IdentifierSemantic::extractDatabaseAndTable(*identifier);
table_id = data.context.resolveStorageID(table_id, Context::ResolveOrdinary);
@ -480,7 +480,7 @@ void ActionsMatcher::visit(const ASTFunction & node, const ASTPtr & ast, Data &
ColumnWithTypeAndName column(
ColumnConst::create(std::move(column_string), 1),
std::make_shared<DataTypeString>(),
data.getUniqueName("__joinGet"));
data.getUniqueName("__joinGetOrDictGet"));
data.addAction(ExpressionAction::addColumn(column));
argument_types.push_back(column.type);
argument_names.push_back(column.name);

2
src/Interpreters/MarkTableIdentifiersVisitor.cpp
View File

@ -44,7 +44,7 @@ void MarkTableIdentifiersMatcher::visit(const ASTFunction & func, ASTPtr &, Data
}
// first argument of joinGet can be a table identifier
if (func.name == "joinGet")
if (functionIsJoinGetOrDictGet(func.name))
{
auto & ast = func.arguments->children.at(0);
if (auto opt_name = tryGetIdentifierName(ast))

7
src/Interpreters/misc.h
View File

@ -1,5 +1,7 @@
#pragma once
#include <Common/StringUtils/StringUtils.h>
namespace DB
{
@ -18,4 +20,9 @@ inline bool functionIsLikeOperator(const std::string & name)
return name == "like" || name == "notLike";
}
inline bool functionIsJoinGetOrDictGet(const std::string & name)
{
return name == "joinGet" || startsWith(name, "dictGet");
}
}

4
tests/queries/0_stateless/01018_ddl_dictionaries_select.reference
View File

@ -17,3 +17,7 @@ database_for_dict dict1 ComplexKeyCache
database_for_dict dict2 Hashed
6
6
6
6
6
6

10
tests/queries/0_stateless/01018_ddl_dictionaries_select.sql
View File

@ -105,6 +105,16 @@ LAYOUT(HASHED());
SELECT dictGetString('database_for_dict.dict3', 'some_column', toUInt64(12));
-- dictGet with table name
USE database_for_dict;
SELECT dictGetString(dict3, 'some_column', toUInt64(12));
SELECT dictGetString(database_for_dict.dict3, 'some_column', toUInt64(12));
SELECT dictGetString(default.dict3, 'some_column', toUInt64(12)); -- {serverError 36}
SELECT dictGet(dict3, 'some_column', toUInt64(12));
SELECT dictGet(database_for_dict.dict3, 'some_column', toUInt64(12));
SELECT dictGet(default.dict3, 'some_column', toUInt64(12)); -- {serverError 36}
USE default;
DROP TABLE database_for_dict.table_for_dict;
SYSTEM RELOAD DICTIONARIES; -- {serverError 60}