mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
43 lines
819 B
C++
43 lines
819 B
C++
#pragma once
|
|
#include <Processors/ISource.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class IMergeTreeSelectAlgorithm;
|
|
using MergeTreeSelectAlgorithmPtr = std::unique_ptr<IMergeTreeSelectAlgorithm>;
|
|
|
|
struct ChunkAndProgress;
|
|
|
|
class MergeTreeSource final : public ISource
|
|
{
|
|
public:
|
|
explicit MergeTreeSource(MergeTreeSelectAlgorithmPtr algorithm_);
|
|
~MergeTreeSource() override;
|
|
|
|
std::string getName() const override;
|
|
|
|
Status prepare() override;
|
|
|
|
#if defined(OS_LINUX)
|
|
int schedule() override;
|
|
#endif
|
|
|
|
protected:
|
|
std::optional<Chunk> tryGenerate() override;
|
|
|
|
void onCancel() override;
|
|
|
|
private:
|
|
MergeTreeSelectAlgorithmPtr algorithm;
|
|
|
|
#if defined(OS_LINUX)
|
|
struct AsyncReadingState;
|
|
std::unique_ptr<AsyncReadingState> async_reading_state;
|
|
#endif
|
|
|
|
Chunk processReadResult(ChunkAndProgress chunk);
|
|
};
|
|
|
|
}
|