mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-30 05:30:51 +00:00
42 lines
1.0 KiB
C++
42 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <DataStreams/IBlockOutputStream.h>
|
|
#include <Storages/ConstraintsDescription.h>
|
|
#include <Interpreters/StorageID.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Check for constraints violation. If anything is found - throw an exception with detailed error message.
|
|
* Otherwise just pass block to output unchanged.
|
|
*/
|
|
|
|
class CheckConstraintsBlockOutputStream : public IBlockOutputStream
|
|
{
|
|
public:
|
|
CheckConstraintsBlockOutputStream(
|
|
const StorageID & table_,
|
|
const BlockOutputStreamPtr & output_,
|
|
const Block & header_,
|
|
const ConstraintsDescription & constraints_,
|
|
const Context & context_);
|
|
|
|
Block getHeader() const override { return header; }
|
|
void write(const Block & block) override;
|
|
|
|
void flush() override;
|
|
|
|
void writePrefix() override;
|
|
void writeSuffix() override;
|
|
|
|
private:
|
|
StorageID table_id;
|
|
BlockOutputStreamPtr output;
|
|
Block header;
|
|
const ConstraintsDescription constraints;
|
|
const ConstraintsExpressions expressions;
|
|
size_t rows_written = 0;
|
|
};
|
|
}
|