mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 19:02:04 +00:00
51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Storages/RabbitMQ/StorageRabbitMQ.h>
|
|
#include <Storages/RabbitMQ/ReadBufferFromRabbitMQConsumer.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class RabbitMQBlockInputStream : public IBlockInputStream
|
|
{
|
|
|
|
public:
|
|
RabbitMQBlockInputStream(
|
|
StorageRabbitMQ & storage_,
|
|
const StorageMetadataPtr & metadata_snapshot_,
|
|
const std::shared_ptr<Context> & context_,
|
|
const Names & columns,
|
|
bool ack_in_suffix = true);
|
|
|
|
~RabbitMQBlockInputStream() override;
|
|
|
|
String getName() const override { return storage.getName(); }
|
|
Block getHeader() const override;
|
|
|
|
void readPrefixImpl() override;
|
|
Block readImpl() override;
|
|
void readSuffixImpl() override;
|
|
|
|
void updateChannel();
|
|
bool needManualChannelUpdate();
|
|
bool sendAck();
|
|
|
|
private:
|
|
StorageRabbitMQ & storage;
|
|
StorageMetadataPtr metadata_snapshot;
|
|
const std::shared_ptr<Context> context;
|
|
Names column_names;
|
|
bool ack_in_suffix;
|
|
|
|
bool finished = false;
|
|
const Block non_virtual_header;
|
|
const Block virtual_header;
|
|
|
|
ConsumerBufferPtr buffer;
|
|
};
|
|
|
|
}
|