mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 18:02:24 +00:00
51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <Processors/IProcessor.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ISource : public IProcessor
|
|
{
|
|
private:
|
|
ReadProgressCounters read_progress;
|
|
bool read_progress_was_set = false;
|
|
bool auto_progress;
|
|
|
|
protected:
|
|
OutputPort & output;
|
|
bool has_input = false;
|
|
bool finished = false;
|
|
bool got_exception = false;
|
|
Port::Data current_chunk;
|
|
|
|
std::shared_ptr<const StorageLimitsList> storage_limits;
|
|
|
|
virtual Chunk generate();
|
|
virtual std::optional<Chunk> tryGenerate();
|
|
|
|
virtual void progress(size_t read_rows, size_t read_bytes);
|
|
|
|
public:
|
|
explicit ISource(Block header, bool enable_auto_progress = true);
|
|
~ISource() override;
|
|
|
|
Status prepare() override;
|
|
void work() override;
|
|
|
|
OutputPort & getPort() { return output; }
|
|
const OutputPort & getPort() const { return output; }
|
|
|
|
void setStorageLimits(const std::shared_ptr<const StorageLimitsList> & storage_limits_) override;
|
|
|
|
/// Default implementation for all the sources.
|
|
std::optional<ReadProgress> getReadProgress() final;
|
|
|
|
void addTotalRowsApprox(size_t value) { read_progress.total_rows_approx += value; }
|
|
};
|
|
|
|
using SourcePtr = std::shared_ptr<ISource>;
|
|
|
|
}
|