mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-23 10:10:50 +00:00
77 lines
3.0 KiB
C++
77 lines
3.0 KiB
C++
#pragma once
|
||
|
||
#include <DB/Interpreters/Context.h>
|
||
#include <DB/DataStreams/AddingConstColumnBlockInputStream.h>
|
||
#include <DB/DataStreams/OneBlockInputStream.h>
|
||
#include <DB/DataTypes/DataTypeString.h>
|
||
#include <DB/DataTypes/DataTypesNumberFixed.h>
|
||
#include <DB/Parsers/ASTIdentifier.h>
|
||
#include <DB/Parsers/ASTExpressionList.h>
|
||
#include <DB/Parsers/ASTLiteral.h>
|
||
#include <DB/Parsers/ASTSelectQuery.h>
|
||
#include <DB/Columns/ColumnString.h>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
namespace VirtualColumnUtils
|
||
{
|
||
|
||
/// Вычислить минимальный числовый суффикс, который надо добавить к строке, чтобы она не присутствовала в множестве
|
||
String chooseSuffix(const NamesAndTypesList & columns, const String & name);
|
||
|
||
/// Вычислить минимальный общий числовый суффикс, который надо добавить к каждой строке,
|
||
/// чтобы ни одна не присутствовала в множестве.
|
||
String chooseSuffixForSet(const NamesAndTypesList & columns, const std::vector<String> & names);
|
||
|
||
/// Добавляет в селект запрос секцию select column_name as value
|
||
/// Например select _port as 9000.
|
||
void rewriteEntityInAst(ASTPtr ast, const String & column_name, const Field & value);
|
||
|
||
/// Получить поток блоков содержащий интересующие нас значения виртуальных столбцов
|
||
/// На вход подается исходный запрос, блок с значениями виртуальных столбцов и контекст
|
||
BlockInputStreamPtr getVirtualColumnsBlocks(ASTPtr query, const Block & input, const Context & context);
|
||
|
||
/// Извлечь из входного потока множество значений столбца name
|
||
template<typename T1>
|
||
std::set<T1> extractSingleValueFromBlocks(BlockInputStreamPtr input, const String & name)
|
||
{
|
||
std::set<T1> res;
|
||
input->readPrefix();
|
||
while(1)
|
||
{
|
||
Block block = input->read();
|
||
if (!block) break;
|
||
const ColumnWithNameAndType & data = block.getByName(name);
|
||
for (size_t i = 0; i < block.rows(); ++i)
|
||
res.insert((*data.column)[i].get<T1>());
|
||
}
|
||
return res;
|
||
}
|
||
|
||
/// Извлечь из входного потока множество пар значений в столбцах first_name и second_name
|
||
template<typename T1, typename T2>
|
||
std::set< std::pair<T1, T2> > extractTwoValuesFromBlocks(BlockInputStreamPtr input,
|
||
const String & first_name, const String & second_name)
|
||
{
|
||
std::set< std::pair<T1, T2> > res;
|
||
input->readPrefix();
|
||
while(1)
|
||
{
|
||
Block block = input->read();
|
||
if (!block) break;
|
||
const ColumnWithNameAndType & first = block.getByName(first_name);
|
||
const ColumnWithNameAndType & second = block.getByName(second_name);
|
||
for (size_t i = 0; i < block.rows(); ++i)
|
||
{
|
||
T1 val1 = (*first.column)[i].get<T1>();
|
||
T2 val2 = (*second.column)[i].get<T2>();
|
||
res.insert(std::make_pair(val1, val2));
|
||
}
|
||
}
|
||
return res;
|
||
}
|
||
|
||
}
|
||
}
|