mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-09 18:10:48 +00:00
36 lines
920 B
C++
36 lines
920 B
C++
#pragma once
|
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
|
#include <IO/DelimitedReadBuffer.h>
|
|
#include <Interpreters/Context.h>
|
|
|
|
#include <Storages/Kafka/StorageKafka.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class KafkaBlockInputStream : public IBlockInputStream
|
|
{
|
|
public:
|
|
KafkaBlockInputStream(StorageKafka & storage_, const Context & context_, const String & schema, size_t max_block_size_);
|
|
~KafkaBlockInputStream() override;
|
|
|
|
String getName() const override { return storage.getName(); }
|
|
Block readImpl() override { return children.back()->read(); }
|
|
Block getHeader() const override { return storage.getSampleBlock(); }
|
|
|
|
void readPrefixImpl() override;
|
|
void readSuffixImpl() override;
|
|
|
|
private:
|
|
StorageKafka & storage;
|
|
Context context;
|
|
size_t max_block_size;
|
|
|
|
ConsumerPtr consumer;
|
|
std::unique_ptr<DelimitedReadBuffer> buffer;
|
|
bool broken = true, claimed = false;
|
|
};
|
|
|
|
} // namespace DB
|