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

#pragma once

#include <DB/DataStreams/IBlockOutputStream.h>
#include <DB/Core/NamesAndTypes.h>


namespace DB
{

namespace ErrorCodes
{
	extern const int ILLEGAL_COLUMN;
}


/// Throws exception on encountering prohibited column in block
class ProhibitColumnsBlockOutputStream : public IBlockOutputStream
{
public:
	ProhibitColumnsBlockOutputStream(const BlockOutputStreamPtr & output, const NamesAndTypesList & columns)
		: output{output}, columns{columns}
	{
	}

private:
	void write(const Block & block) override;

	void flush() override { output->flush(); }

	void writePrefix() override { output->writePrefix(); }
	void writeSuffix() override { output->writeSuffix(); }

	BlockOutputStreamPtr output;
	NamesAndTypesList columns;
};


}