thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-01 03:52:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
avogar 2023-08-01 10:14:09 +00:00
parent a71cd56a90
commit fa905ebd27
2 changed files with 0 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/Formats/FormatFactory.cpp
View File

@ -1,7 +1,6 @@
#include <Formats/FormatFactory.h>
#include <algorithm>
//#include <Core/Settings.h>
#include <Formats/FormatSettings.h>
#include <Interpreters/Context.h>
#include <Interpreters/ProcessList.h>

75
src/Processors/Formats/OutputFormatWithExceptionHandlerAdaptor.h
View File

@ -1,75 +0,0 @@
#pragma once
#include <Processors/Formats/IOutputFormat.h>
#include <Processors/Formats/IRowOutputFormat.h>
#include <IO/WriteBuffer.h>
#include <IO/PeekableWriteBuffer.h>
namespace DB
{
template <typename Base, typename... Args>
class RowOutputFormatWithExceptionHandlerAdaptorBase : public Base
{
public:
RowOutputFormatWithExceptionHandlerAdaptorBase(bool handle_exceptions, const Block & header, WriteBuffer & out_, Args... args)
: Base(header, out_, std::forward<Args>(args)...)
{
if (handle_exceptions)
peekable_out = std::make_unique<PeekableWriteBuffer>(*Base::getWriteBufferPtr());
}
void write(const Columns & columns, size_t row_num)
{
if (!peekable_out)
Base::write(columns, row_num);
PeekableWriteBufferCheckpoint checkpoint(*peekable_out);
try
{
Base::write(columns, row_num);
}
catch (...)
{
peekable_out->rollbackToCheckpoint();
throw;
}
}
void flush() override
{
getWriteBufferPtr()->next();
if (peekable_out)
Base::getWriteBufferPtr()->next();
}
void finalizeBuffers() override
{
if (peekable_out)
peekable_out->finalize();
}
void resetFormatterImpl() override
{
peekable_out = std::make_unique<PeekableWriteBuffer>(*Base::getWriteBufferPtr());
}
protected:
/// Returns buffer that should be used in derived classes instead of out.
WriteBuffer * getWriteBufferPtr() override
{
if (peekable_out)
peekable_out.get();
return Base::getWriteBufferPtr();
}
private:
std::unique_ptr<PeekableWriteBuffer> peekable_out;
};
}