mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-12 02:23:14 +00:00
33 lines
637 B
C++
33 lines
637 B
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include <DB/DataStreams/IBlockOutputStream.h>
|
||
|
|
#include <DB/DataStreams/SquashingTransform.h>
|
||
|
|
|
||
|
|
|
||
|
|
namespace DB
|
||
|
|
{
|
||
|
|
|
||
|
|
/** Merging consecutive blocks of stream to specified minimum size.
|
||
|
|
*/
|
||
|
|
class SquashingBlockOutputStream : public IBlockOutputStream
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
SquashingBlockOutputStream(BlockOutputStreamPtr & dst, size_t min_block_size_rows, size_t min_block_size_bytes);
|
||
|
|
|
||
|
|
void write(const Block & block) override;
|
||
|
|
|
||
|
|
void flush() override;
|
||
|
|
void writePrefix() override;
|
||
|
|
void writeSuffix() override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
BlockOutputStreamPtr output;
|
||
|
|
|
||
|
|
SquashingTransform transform;
|
||
|
|
bool all_written = false;
|
||
|
|
|
||
|
|
void finalize();
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|