mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 11:32:03 +00:00
55 lines
1.3 KiB
C++
55 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <Poco/SharedPtr.h>
|
|
|
|
#include <DB/Interpreters/ExpressionActions.h>
|
|
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
using Poco::SharedPtr;
|
|
|
|
|
|
/** Выполняет над блоком вычисление некоторого выражения.
|
|
* Выражение состоит из идентификаторов столбцов из блока, констант, обычных функций.
|
|
* Например: hits * 2 + 3, instr("yandex", url)
|
|
* Выражение не меняет количество строк в потоке, и обрабатывает каждую строку независимо от других.
|
|
*/
|
|
class ExpressionBlockInputStream : public IProfilingBlockInputStream
|
|
{
|
|
public:
|
|
ExpressionBlockInputStream(BlockInputStreamPtr input_, ExpressionActionsPtr expression_)
|
|
: expression(expression_)
|
|
{
|
|
children.push_back(input_);
|
|
}
|
|
|
|
String getName() const { return "ExpressionBlockInputStream"; }
|
|
|
|
String getID() const
|
|
{
|
|
std::stringstream res;
|
|
res << "Expression(" << children.back()->getID() << ", " << expression->getID() << ")";
|
|
return res.str();
|
|
}
|
|
|
|
protected:
|
|
Block readImpl()
|
|
{
|
|
Block res = children.back()->read();
|
|
if (!res)
|
|
return res;
|
|
|
|
expression->execute(res);
|
|
|
|
return res;
|
|
}
|
|
|
|
private:
|
|
ExpressionActionsPtr expression;
|
|
};
|
|
|
|
}
|