ClickHouse/dbms/include/DB/IO/AsynchronousWriteBuffer.h

97 lines
1.9 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#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();
}
}
};
}