mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-04 07:30:52 +00:00
47 lines
1.1 KiB
C++
47 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <Core/Block.h>
|
|
#include <DataStreams/IBlockOutputStream.h>
|
|
#include <Common/Throttler.h>
|
|
#include <IO/ConnectionTimeouts.h>
|
|
#include <Interpreters/ClientInfo.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_,
|
|
const ClientInfo & client_info_);
|
|
|
|
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;
|
|
Block header;
|
|
bool finished = false;
|
|
};
|
|
|
|
}
|