2019-01-21 14:02:03 +00:00
|
|
|
#pragma once
|
|
|
|
|
2019-01-24 12:44:58 +00:00
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
2019-01-21 14:02:03 +00:00
|
|
|
#include <Interpreters/Context.h>
|
2019-01-30 17:41:06 +00:00
|
|
|
|
|
|
|
#include <Storages/Kafka/StorageKafka.h>
|
2019-01-21 14:02:03 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2019-01-24 12:44:58 +00:00
|
|
|
class KafkaBlockInputStream : public IBlockInputStream
|
2019-01-21 14:02:03 +00:00
|
|
|
{
|
|
|
|
public:
|
|
|
|
KafkaBlockInputStream(StorageKafka & storage_, const Context & context_, const String & schema, size_t max_block_size_);
|
|
|
|
~KafkaBlockInputStream() override;
|
|
|
|
|
2019-01-30 17:41:06 +00:00
|
|
|
String getName() const override { return storage.getName(); }
|
|
|
|
Block readImpl() override { return children.back()->read(); }
|
|
|
|
Block getHeader() const override { return storage.getSampleBlock(); }
|
|
|
|
|
2019-01-21 14:02:03 +00:00
|
|
|
void readPrefixImpl() override;
|
|
|
|
void readSuffixImpl() override;
|
|
|
|
|
|
|
|
private:
|
|
|
|
StorageKafka & storage;
|
|
|
|
Context context;
|
2019-02-11 11:54:30 +00:00
|
|
|
UInt64 max_block_size;
|
2019-01-25 12:48:59 +00:00
|
|
|
|
2019-04-22 13:23:05 +00:00
|
|
|
BufferPtr buffer;
|
2019-01-30 17:41:06 +00:00
|
|
|
bool broken = true, claimed = false;
|
2019-01-21 14:02:03 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace DB
|