2022-07-20 15:25:24 +00:00
|
|
|
#include <Analyzer/TableFunctionNode.h>
|
|
|
|
|
|
|
|
#include <IO/WriteBuffer.h>
|
|
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
#include <IO/Operators.h>
|
|
|
|
|
|
|
|
#include <Storages/IStorage.h>
|
|
|
|
|
|
|
|
#include <Parsers/ASTFunction.h>
|
2023-03-13 11:41:00 +00:00
|
|
|
#include <Parsers/ASTSetQuery.h>
|
2022-07-20 15:25:24 +00:00
|
|
|
|
|
|
|
#include <Interpreters/Context.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
|
|
|
extern const int LOGICAL_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
TableFunctionNode::TableFunctionNode(String table_function_name_)
|
2022-10-07 10:44:28 +00:00
|
|
|
: IQueryTreeNode(children_size)
|
|
|
|
, table_function_name(table_function_name_)
|
2022-07-20 15:25:24 +00:00
|
|
|
, storage_id("system", "one")
|
|
|
|
{
|
|
|
|
children[arguments_child_index] = std::make_shared<ListNode>();
|
|
|
|
}
|
|
|
|
|
2023-05-09 15:14:49 +00:00
|
|
|
void TableFunctionNode::resolve(TableFunctionPtr table_function_value, StoragePtr storage_value, ContextPtr context, std::vector<size_t> unresolved_arguments_indexes_)
|
2022-07-20 15:25:24 +00:00
|
|
|
{
|
|
|
|
table_function = std::move(table_function_value);
|
|
|
|
storage = std::move(storage_value);
|
|
|
|
storage_id = storage->getStorageID();
|
|
|
|
storage_snapshot = storage->getStorageSnapshot(storage->getInMemoryMetadataPtr(), context);
|
2023-05-09 15:14:49 +00:00
|
|
|
unresolved_arguments_indexes = std::move(unresolved_arguments_indexes_);
|
2022-07-20 15:25:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
const StorageID & TableFunctionNode::getStorageID() const
|
|
|
|
{
|
|
|
|
if (!storage)
|
|
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "Table function node {} is not resolved", table_function_name);
|
|
|
|
|
|
|
|
return storage_id;
|
|
|
|
}
|
|
|
|
|
|
|
|
const StorageSnapshotPtr & TableFunctionNode::getStorageSnapshot() const
|
|
|
|
{
|
|
|
|
if (!storage)
|
|
|
|
throw Exception(ErrorCodes::LOGICAL_ERROR, "Table function node {} is not resolved", table_function_name);
|
|
|
|
|
|
|
|
return storage_snapshot;
|
|
|
|
}
|
|
|
|
|
|
|
|
void TableFunctionNode::dumpTreeImpl(WriteBuffer & buffer, FormatState & format_state, size_t indent) const
|
|
|
|
{
|
|
|
|
buffer << std::string(indent, ' ') << "TABLE_FUNCTION id: " << format_state.getNodeId(this);
|
|
|
|
|
|
|
|
if (hasAlias())
|
|
|
|
buffer << ", alias: " << getAlias();
|
|
|
|
|
|
|
|
buffer << ", table_function_name: " << table_function_name;
|
|
|
|
|
2022-09-04 15:20:59 +00:00
|
|
|
if (table_expression_modifiers)
|
|
|
|
{
|
|
|
|
buffer << ", ";
|
|
|
|
table_expression_modifiers->dump(buffer);
|
|
|
|
}
|
|
|
|
|
2022-07-20 15:25:24 +00:00
|
|
|
const auto & arguments = getArguments();
|
|
|
|
if (!arguments.getNodes().empty())
|
|
|
|
{
|
|
|
|
buffer << '\n' << std::string(indent + 2, ' ') << "ARGUMENTS\n";
|
|
|
|
arguments.dumpTreeImpl(buffer, format_state, indent + 4);
|
|
|
|
}
|
2023-03-13 11:41:00 +00:00
|
|
|
|
|
|
|
if (!settings_changes.empty())
|
|
|
|
{
|
2023-03-22 16:58:34 +00:00
|
|
|
buffer << '\n' << std::string(indent + 2, ' ') << "SETTINGS";
|
2023-03-13 11:41:00 +00:00
|
|
|
for (const auto & change : settings_changes)
|
|
|
|
buffer << fmt::format(" {}={}", change.name, toString(change.value));
|
|
|
|
}
|
2022-07-20 15:25:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool TableFunctionNode::isEqualImpl(const IQueryTreeNode & rhs) const
|
|
|
|
{
|
|
|
|
const auto & rhs_typed = assert_cast<const TableFunctionNode &>(rhs);
|
|
|
|
if (table_function_name != rhs_typed.table_function_name)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (storage && rhs_typed.storage)
|
|
|
|
return storage_id == rhs_typed.storage_id;
|
|
|
|
|
2023-03-22 16:58:34 +00:00
|
|
|
if (settings_changes != rhs_typed.settings_changes)
|
|
|
|
return false;
|
|
|
|
|
2023-02-25 19:16:51 +00:00
|
|
|
return table_expression_modifiers == rhs_typed.table_expression_modifiers;
|
2022-07-20 15:25:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void TableFunctionNode::updateTreeHashImpl(HashState & state) const
|
|
|
|
{
|
|
|
|
state.update(table_function_name.size());
|
|
|
|
state.update(table_function_name);
|
|
|
|
|
|
|
|
if (storage)
|
|
|
|
{
|
|
|
|
auto full_name = storage_id.getFullNameNotQuoted();
|
|
|
|
state.update(full_name.size());
|
|
|
|
state.update(full_name);
|
|
|
|
}
|
2022-09-04 15:20:59 +00:00
|
|
|
|
|
|
|
if (table_expression_modifiers)
|
|
|
|
table_expression_modifiers->updateTreeHash(state);
|
2023-03-13 11:41:00 +00:00
|
|
|
|
2023-03-22 16:58:34 +00:00
|
|
|
state.update(settings_changes.size());
|
2023-03-13 11:41:00 +00:00
|
|
|
for (const auto & change : settings_changes)
|
|
|
|
{
|
|
|
|
state.update(change.name.size());
|
|
|
|
state.update(change.name);
|
|
|
|
|
|
|
|
const auto & value_dump = change.value.dump();
|
|
|
|
state.update(value_dump.size());
|
|
|
|
state.update(value_dump);
|
|
|
|
}
|
2022-07-20 15:25:24 +00:00
|
|
|
}
|
|
|
|
|
2022-10-10 10:25:58 +00:00
|
|
|
QueryTreeNodePtr TableFunctionNode::cloneImpl() const
|
|
|
|
{
|
|
|
|
auto result = std::make_shared<TableFunctionNode>(table_function_name);
|
|
|
|
|
|
|
|
result->storage = storage;
|
|
|
|
result->storage_id = storage_id;
|
|
|
|
result->storage_snapshot = storage_snapshot;
|
|
|
|
result->table_expression_modifiers = table_expression_modifiers;
|
2023-03-13 11:41:00 +00:00
|
|
|
result->settings_changes = settings_changes;
|
2022-10-10 10:25:58 +00:00
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2023-03-14 09:14:58 +00:00
|
|
|
ASTPtr TableFunctionNode::toASTImpl(const ConvertToASTOptions & options) const
|
2022-07-20 15:25:24 +00:00
|
|
|
{
|
|
|
|
auto table_function_ast = std::make_shared<ASTFunction>();
|
|
|
|
|
|
|
|
table_function_ast->name = table_function_name;
|
|
|
|
|
|
|
|
const auto & arguments = getArguments();
|
2023-03-07 20:39:26 +00:00
|
|
|
table_function_ast->children.push_back(arguments.toAST(options));
|
2022-09-06 15:25:52 +00:00
|
|
|
table_function_ast->arguments = table_function_ast->children.back();
|
2022-07-20 15:25:24 +00:00
|
|
|
|
2023-03-13 11:41:00 +00:00
|
|
|
if (!settings_changes.empty())
|
|
|
|
{
|
|
|
|
auto settings_ast = std::make_shared<ASTSetQuery>();
|
|
|
|
settings_ast->changes = settings_changes;
|
|
|
|
settings_ast->is_standalone = false;
|
|
|
|
table_function_ast->arguments->children.push_back(std::move(settings_ast));
|
|
|
|
}
|
|
|
|
|
2022-07-20 15:25:24 +00:00
|
|
|
return table_function_ast;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|