mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-05 15:21:43 +00:00
69 lines
1.8 KiB
C++
69 lines
1.8 KiB
C++
#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();
|
|
};
|
|
|
|
}
|