mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 09:22:05 +00:00
107 lines
2.2 KiB
C++
107 lines
2.2 KiB
C++
#pragma once
|
||
|
||
#include <math.h>
|
||
|
||
#include <vector>
|
||
|
||
#include <Poco/SharedPtr.h>
|
||
#include <Poco/Net/NetException.h>
|
||
|
||
#include <statdaemons/threadpool.hpp>
|
||
|
||
#include <DB/IO/WriteBuffer.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
using Poco::SharedPtr;
|
||
|
||
|
||
/** Записывает данные асинхронно с помощью двойной буферизации.
|
||
*/
|
||
class AsynchronousWriteBuffer : public WriteBuffer
|
||
{
|
||
private:
|
||
WriteBuffer & out; /// Основной буфер, отвечает за запись данных.
|
||
std::vector<char> memory; /// Кусок памяти для дублирования буфера.
|
||
boost::threadpool::pool pool; /// Для асинхронной записи данных.
|
||
bool started; /// Была ли запущена асинхронная запись данных.
|
||
|
||
/// Менять местами основной и дублирующий буферы.
|
||
void swapBuffers()
|
||
{
|
||
buffer().swap(out.buffer());
|
||
std::swap(position(), out.position());
|
||
}
|
||
|
||
void nextImpl()
|
||
{
|
||
if (!offset())
|
||
return;
|
||
|
||
if (started)
|
||
pool.wait();
|
||
else
|
||
started = true;
|
||
|
||
if (exception)
|
||
exception->rethrow();
|
||
|
||
swapBuffers();
|
||
|
||
/// Данные будут записываться в отельном потоке.
|
||
pool.schedule(boost::bind(&AsynchronousWriteBuffer::thread, this));
|
||
}
|
||
|
||
public:
|
||
AsynchronousWriteBuffer(WriteBuffer & out_) : WriteBuffer(NULL, 0), out(out_), memory(out.buffer().size()), pool(1), started(false)
|
||
{
|
||
/// Данные пишутся в дублирующий буфер.
|
||
set(&memory[0], memory.size());
|
||
}
|
||
|
||
~AsynchronousWriteBuffer()
|
||
{
|
||
if (std::uncaught_exception())
|
||
return;
|
||
|
||
if (started)
|
||
pool.wait();
|
||
if (exception)
|
||
exception->rethrow();
|
||
|
||
swapBuffers();
|
||
out.next();
|
||
}
|
||
|
||
ExceptionPtr exception;
|
||
|
||
/// То, что выполняется в отдельном потоке
|
||
void thread()
|
||
{
|
||
try
|
||
{
|
||
out.next();
|
||
}
|
||
catch (const Exception & e)
|
||
{
|
||
exception = e.clone();
|
||
}
|
||
catch (const Poco::Exception & e)
|
||
{
|
||
exception = e.clone();
|
||
}
|
||
catch (const std::exception & e)
|
||
{
|
||
exception = new Exception(e.what(), ErrorCodes::STD_EXCEPTION);
|
||
}
|
||
catch (...)
|
||
{
|
||
exception = new Exception("Unknown exception", ErrorCodes::UNKNOWN_EXCEPTION);
|
||
}
|
||
}
|
||
};
|
||
|
||
}
|