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

52 lines
1.0 KiB
C
Raw Normal View History

2012-06-25 02:52:51 +00:00
#pragma once
#include <DB/DataStreams/IProfilingBlockInputStream.h>
namespace DB
{
using Poco::SharedPtr;
/** Объединяет несколько источников в один.
* В отличие от UnionBlockInputStream, делает это последовательно.
* Блоки разных источников не перемежаются друг с другом.
*/
class ConcatBlockInputStream : public IProfilingBlockInputStream
{
public:
ConcatBlockInputStream(BlockInputStreams inputs_)
{
children.insert(children.end(), inputs_.begin(), inputs_.end());
current_stream = children.begin();
}
2012-10-20 02:10:47 +00:00
String getName() const { return "ConcatBlockInputStream"; }
BlockInputStreamPtr clone() { return new ConcatBlockInputStream(children); }
protected:
2012-06-25 02:52:51 +00:00
Block readImpl()
{
Block res;
while (current_stream != children.end())
{
res = (*current_stream)->read();
if (res)
break;
else
++current_stream;
}
return res;
}
private:
BlockInputStreams::iterator current_stream;
};
}