mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 19:32:07 +00:00
52 lines
1.5 KiB
C++
52 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <ostream>
|
|
#include <optional>
|
|
|
|
#include <Core/Names.h>
|
|
#include <Core/Types.h>
|
|
#include <Parsers/IAST_fwd.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ASTIdentifier;
|
|
|
|
/// Information about table and column names extracted from ASTSelectQuery block. Do not include info from subselects.
|
|
struct RequiredSourceColumnsData
|
|
{
|
|
struct NameInfo
|
|
{
|
|
std::set<String> aliases;
|
|
size_t appears = 0;
|
|
|
|
void addInclusion(const String & alias)
|
|
{
|
|
if (!alias.empty())
|
|
aliases.insert(alias);
|
|
++appears;
|
|
}
|
|
};
|
|
|
|
std::unordered_map<String, NameInfo> required_names;
|
|
NameSet private_aliases; /// lambda aliases that should not be interpreted as required columns
|
|
NameSet complex_aliases; /// aliases to functions results: they are not required cause calculated by query itself
|
|
NameSet masked_columns; /// columns names masked by function aliases: we still need them in required columns
|
|
NameSet array_join_columns; /// Tech debt: we exclude ArrayJoin columns from general logic cause they have own logic outside
|
|
|
|
bool has_table_join = false;
|
|
bool has_array_join = false;
|
|
|
|
bool addColumnAliasIfAny(const IAST & ast);
|
|
void addColumnIdentifier(const ASTIdentifier & node);
|
|
bool addArrayJoinAliasIfAny(const IAST & ast);
|
|
void addArrayJoinIdentifier(const ASTIdentifier & node);
|
|
|
|
NameSet requiredColumns() const;
|
|
size_t nameInclusion(const String & name) const;
|
|
};
|
|
|
|
std::ostream & operator << (std::ostream & os, const RequiredSourceColumnsData & cols);
|
|
|
|
}
|