mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-02 06:30:55 +00:00
63 lines
2.0 KiB
C++
63 lines
2.0 KiB
C++
#include <Storages/StorageJoin.h>
|
|
#include <Interpreters/Join.h>
|
|
#include <Interpreters/Limits.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NO_SUCH_COLUMN_IN_TABLE;
|
|
extern const int INCOMPATIBLE_TYPE_OF_JOIN;
|
|
}
|
|
|
|
|
|
StorageJoin::StorageJoin(
|
|
const String & path_,
|
|
const String & name_,
|
|
const Names & key_names_,
|
|
ASTTableJoin::Kind kind_, ASTTableJoin::Strictness strictness_,
|
|
const NamesAndTypes & columns_,
|
|
const NamesAndTypes & materialized_columns_,
|
|
const NamesAndTypes & alias_columns_,
|
|
const ColumnDefaults & column_defaults_)
|
|
: StorageSetOrJoinBase{path_, name_, columns_, materialized_columns_, alias_columns_, column_defaults_},
|
|
key_names(key_names_), kind(kind_), strictness(strictness_)
|
|
{
|
|
/// Check that key exists in table definition.
|
|
const auto check_key_exists = [] (const NamesAndTypes & columns, const String & key)
|
|
{
|
|
for (const auto & column : columns)
|
|
if (column.name == key)
|
|
return true;
|
|
return false;
|
|
};
|
|
|
|
for (const auto & key : key_names)
|
|
if (!check_key_exists(columns, key) && !check_key_exists(materialized_columns, key))
|
|
throw Exception{
|
|
"Key column (" + key + ") does not exist in table declaration.",
|
|
ErrorCodes::NO_SUCH_COLUMN_IN_TABLE};
|
|
|
|
/// NOTE StorageJoin doesn't use join_use_nulls setting.
|
|
|
|
join = std::make_shared<Join>(key_names, key_names, false /* use_nulls */, Limits(), kind, strictness);
|
|
join->setSampleBlock(getSampleBlock().sortColumns());
|
|
restore();
|
|
}
|
|
|
|
|
|
void StorageJoin::assertCompatible(ASTTableJoin::Kind kind_, ASTTableJoin::Strictness strictness_) const
|
|
{
|
|
/// NOTE Could be more loose.
|
|
if (!(kind == kind_ && strictness == strictness_))
|
|
throw Exception("Table " + name + " has incompatible type of JOIN.", ErrorCodes::INCOMPATIBLE_TYPE_OF_JOIN);
|
|
}
|
|
|
|
|
|
void StorageJoin::insertBlock(const Block & block) { join->insertFromBlock(block); }
|
|
size_t StorageJoin::getSize() const { return join->getTotalRowCount(); };
|
|
|
|
}
|