mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 01:12:12 +00:00
97 lines
1.9 KiB
C++
97 lines
1.9 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([this] { thread(); });
|
||
}
|
||
|
||
public:
|
||
AsynchronousWriteBuffer(WriteBuffer & out_) : WriteBuffer(nullptr, 0), out(out_), memory(out.buffer().size()), pool(1), started(false)
|
||
{
|
||
/// Данные пишутся в дублирующий буфер.
|
||
set(&memory[0], memory.size());
|
||
}
|
||
|
||
~AsynchronousWriteBuffer()
|
||
{
|
||
try
|
||
{
|
||
if (started)
|
||
pool.wait();
|
||
|
||
swapBuffers();
|
||
out.next();
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
}
|
||
}
|
||
|
||
ExceptionPtr exception;
|
||
|
||
/// То, что выполняется в отдельном потоке
|
||
void thread()
|
||
{
|
||
try
|
||
{
|
||
out.next();
|
||
}
|
||
catch (...)
|
||
{
|
||
exception = cloneCurrentException();
|
||
}
|
||
}
|
||
};
|
||
|
||
}
|