ClickHouse/dbms/include/DB/DataStreams/RemoteBlockOutputStream.h

44 lines
976 B
C++
Raw Normal View History

2012-05-21 20:38:34 +00:00
#pragma once
#include <DB/Core/Block.h>
2012-05-21 20:38:34 +00:00
#include <DB/DataStreams/IBlockOutputStream.h>
namespace DB
{
class Connection;
class ReadBuffer;
struct Settings;
/** Allow to execute INSERT query on remote server and send data for it.
2012-05-21 20:38:34 +00:00
*/
class RemoteBlockOutputStream : public IBlockOutputStream
{
public:
RemoteBlockOutputStream(Connection & connection_, const String & query_, const Settings * settings_ = nullptr);
2012-05-21 20:38:34 +00:00
/// You can call this method after 'writePrefix', to get table required structure. (You must send data with that structure).
Block getSampleBlock() const
2012-05-21 20:38:34 +00:00
{
return sample_block;
}
void writePrefix() override;
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);
2012-05-21 20:38:34 +00:00
private:
Connection & connection;
2012-06-07 20:02:41 +00:00
String query;
const Settings * settings;
2014-06-05 12:02:18 +00:00
Block sample_block;
2012-05-21 20:38:34 +00:00
};
}