ClickHouse/dbms/src/DataStreams/RemoteBlockOutputStream.h

#pragma once

#include <Core/Block.h>
#include <DataStreams/IBlockOutputStream.h>
#include <Common/Throttler.h>
#include <IO/ConnectionTimeouts.h>


namespace DB
{

class Connection;
class ReadBuffer;
struct Settings;


/** Allow to execute INSERT query on remote server and send data for it.
  */
class RemoteBlockOutputStream : public IBlockOutputStream
{
public:
    RemoteBlockOutputStream(Connection & connection_,
                            const ConnectionTimeouts & timeouts,
                            const String & query_,
                            const Settings * settings_ = nullptr);

    Block getHeader() const override { return header; }

    void write(const Block & block) override;
    void writeSuffix() override;

    /// Send pre-serialized and possibly pre-compressed block of data, that will be read from 'input'.
    void writePrepared(ReadBuffer & input, size_t size = 0);

    ~RemoteBlockOutputStream() override;

private:
    Connection & connection;
    String query;
    const Settings * settings;
    Block header;
    bool finished = false;
};

}