ClickHouse/dbms/include/DB/DataStreams/ExpressionBlockInputStream.h

68 lines
1.6 KiB
C
Raw Normal View History

2011-08-14 00:49:30 +00:00
#pragma once
#include <Poco/SharedPtr.h>
#include <DB/Interpreters/ExpressionActions.h>
2011-09-04 21:23:19 +00:00
#include <DB/DataStreams/IProfilingBlockInputStream.h>
2011-08-14 00:49:30 +00:00
namespace DB
{
using Poco::SharedPtr;
/** Выполняет над блоком вычисление некоторого выражения.
* Выражение состоит из идентификаторов столбцов из блока, констант, обычных функций.
* Например: hits * 2 + 3, instr("yandex", url)
* Выражение не меняет количество строк в потоке, и обрабатывает каждую строку независимо от других.
*/
2011-09-04 21:23:19 +00:00
class ExpressionBlockInputStream : public IProfilingBlockInputStream
2011-08-14 00:49:30 +00:00
{
public:
ExpressionBlockInputStream(BlockInputStreamPtr input_, ExpressionActionsPtr expression_)
: expression(expression_)
2011-09-04 21:23:19 +00:00
{
2013-05-04 04:05:15 +00:00
children.push_back(input_);
2011-09-04 21:23:19 +00:00
}
2011-08-14 00:49:30 +00:00
2012-10-20 02:10:47 +00:00
String getName() const { return "ExpressionBlockInputStream"; }
String getID() const
{
std::stringstream res;
res << "Expression(" << children.back()->getID() << ", " << expression->getID() << ")";
return res.str();
}
const Block & getTotals()
{
if (IProfilingBlockInputStream * child = dynamic_cast<IProfilingBlockInputStream *>(&*children.back()))
{
totals = child->getTotals();
if (totals)
expression->execute(totals);
}
return totals;
}
2012-10-20 02:10:47 +00:00
protected:
2011-09-04 21:23:19 +00:00
Block readImpl()
2011-08-14 00:49:30 +00:00
{
2013-05-04 05:20:07 +00:00
Block res = children.back()->read();
2011-08-14 00:49:30 +00:00
if (!res)
return res;
expression->execute(res);
2011-08-28 05:13:24 +00:00
return res;
2011-08-14 00:49:30 +00:00
}
private:
ExpressionActionsPtr expression;
2011-08-14 00:49:30 +00:00
};
}