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

65 lines
1.8 KiB
C
Raw Normal View History

2011-09-04 01:42:14 +00:00
#pragma once
#include <Yandex/logger_useful.h>
2011-09-04 01:42:14 +00:00
#include <DB/Core/SortDescription.h>
2011-09-04 21:23:19 +00:00
#include <DB/DataStreams/IProfilingBlockInputStream.h>
2011-09-04 01:42:14 +00:00
namespace DB
{
/** Соединяет поток сортированных по отдельности блоков в сортированный целиком поток.
*/
2011-09-04 21:23:19 +00:00
class MergeSortingBlockInputStream : public IProfilingBlockInputStream
2011-09-04 01:42:14 +00:00
{
public:
/// limit - если не 0, то можно выдать только первые limit строк в сортированном порядке.
MergeSortingBlockInputStream(BlockInputStreamPtr input_, SortDescription & description_, size_t limit_ = 0)
: description(description_), limit(limit_), has_been_read(false), log(&Logger::get("MergeSortingBlockInputStream"))
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-09-04 01:42:14 +00:00
2011-09-04 21:23:19 +00:00
String getName() const { return "MergeSortingBlockInputStream"; }
2011-09-04 01:42:14 +00:00
String getID() const
{
std::stringstream res;
2013-05-04 05:20:07 +00:00
res << "MergeSorting(" << children.back()->getID();
for (size_t i = 0; i < description.size(); ++i)
res << ", " << description[i].getID();
res << ")";
return res.str();
}
2012-10-20 02:10:47 +00:00
protected:
Block readImpl();
2011-09-04 01:42:14 +00:00
private:
SortDescription description;
size_t limit;
2011-09-04 01:42:14 +00:00
2012-07-23 20:01:29 +00:00
/// Всё было прочитано.
2012-03-05 02:34:20 +00:00
bool has_been_read;
2012-07-24 18:21:39 +00:00
Logger * log;
/** Слить сразу много блоков с помощью priority queue.
2012-07-25 20:33:43 +00:00
*/
2012-07-23 06:23:29 +00:00
Block merge(Blocks & blocks);
2013-05-28 16:56:05 +00:00
typedef std::vector<SortCursorImpl> CursorImpls;
/** Делаем поддержку двух разных курсоров - с Collation и без.
* Шаблоны используем вместо полиморфных SortCursor'ов и вызовов виртуальных функций.
*/
template <typename TSortCursor>
Block mergeImpl(Blocks & block, CursorImpls & cursors);
2011-09-04 01:42:14 +00:00
};
}