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

33 lines
637 B
C++
Raw Normal View History

2016-07-07 01:57:48 +00:00
#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();
};
}