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

#pragma once

#include <DB/DataStreams/IBlockInputStream.h>
#include <DB/DataStreams/IBlockOutputStream.h>


namespace DB
{

class ProcessListEntry;

struct BlockIO
{
	/** process_list_entry должен уничтожаться позже, чем in и out,
	  *  так как внутри in и out есть ссылка на объект внутри process_list_entry
	  *  (MemoryTracker * current_memory_tracker),
	  *  которая может использоваться до уничтожения in и out.
	  */
	std::shared_ptr<ProcessListEntry> process_list_entry;

	BlockInputStreamPtr in;
	BlockOutputStreamPtr out;

	Block in_sample;	/// Пример блока, который будет прочитан из in.
	Block out_sample;	/// Пример блока, которого нужно писать в out.

	/// Здесь могут быть установлены колбэки для логгирования запроса.
	std::function<void(IBlockInputStream *)> 	finish_callback;
	std::function<void()> 						exception_callback;

	/// Вызывайте эти функции, если нужно логгировать запрос.
	void onFinish()
	{
		if (finish_callback)
			finish_callback(in.get());
	}

	void onException()
	{
		if (exception_callback)
			exception_callback();
		else
			tryLogCurrentException(__PRETTY_FUNCTION__);
	}

	BlockIO & operator= (const BlockIO & rhs)
	{
		/// Обеспечиваем правильный порядок уничтожения.
		out 				= nullptr;
		in 					= nullptr;
		process_list_entry 	= nullptr;

		process_list_entry 	= rhs.process_list_entry;
		in 					= rhs.in;
		out 				= rhs.out;
		in_sample 			= rhs.in_sample;
		out_sample 			= rhs.out_sample;

		finish_callback		= rhs.finish_callback;
		exception_callback	= rhs.exception_callback;

		return *this;
	}

	~BlockIO();
};

}