mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-07 00:50:56 +00:00
101 lines
2.3 KiB
C++
101 lines
2.3 KiB
C++
|
#include <Storages/MergeTree/MergeTreeMutationEntry.h>
|
||
|
|
#include <IO/Operators.h>
|
||
|
|
#include <IO/WriteBufferFromFile.h>
|
||
|
|
#include <IO/ReadBufferFromFile.h>
|
||
|
|
#include <IO/ReadBufferFromString.h>
|
||
|
|
|
||
|
|
#include <Poco/File.h>
|
||
|
|
#include <Poco/Path.h>
|
||
|
|
|
||
|
|
|
||
|
|
namespace DB
|
||
|
|
{
|
||
|
|
|
||
|
|
MergeTreeMutationEntry::MergeTreeMutationEntry(MutationCommands commands_, const String & path_prefix_, Int64 tmp_number)
|
||
|
|
: create_time(time(nullptr))
|
||
|
|
, commands(std::move(commands_))
|
||
|
|
, path_prefix(path_prefix_)
|
||
|
|
, file_name("tmp_mutation_" + toString(tmp_number) + ".txt")
|
||
|
|
, is_temp(true)
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
WriteBufferFromFile out(path_prefix + file_name);
|
||
|
|
out << "format version: 1\n"
|
||
|
|
<< "create time: " << LocalDateTime(create_time) << "\n";
|
||
|
|
out << "commands: ";
|
||
|
|
commands.writeText(out);
|
||
|
|
out << "\n";
|
||
|
|
out.sync();
|
||
|
|
}
|
||
|
|
catch (...)
|
||
|
|
{
|
||
|
|
removeFile();
|
||
|
|
throw;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void MergeTreeMutationEntry::commit(Int64 block_number_)
|
||
|
|
{
|
||
|
|
block_number = block_number_;
|
||
|
|
String new_file_name = "mutation_" + toString(block_number) + ".txt";
|
||
|
|
Poco::File(path_prefix + file_name).renameTo(path_prefix + new_file_name);
|
||
|
|
is_temp = false;
|
||
|
|
file_name = new_file_name;
|
||
|
|
}
|
||
|
|
|
||
|
|
void MergeTreeMutationEntry::removeFile()
|
||
|
|
{
|
||
|
|
if (!file_name.empty())
|
||
|
|
{
|
||
|
|
Poco::File file(path_prefix + file_name);
|
||
|
|
if (!file.exists())
|
||
|
|
return;
|
||
|
|
|
||
|
|
file.remove(false);
|
||
|
|
file_name.clear();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
MergeTreeMutationEntry::MergeTreeMutationEntry(const String & path_prefix_, const String & file_name_)
|
||
|
|
: path_prefix(path_prefix_)
|
||
|
|
, file_name(file_name_)
|
||
|
|
, is_temp(false)
|
||
|
|
{
|
||
|
|
ReadBufferFromString file_name_buf(file_name);
|
||
|
|
file_name_buf >> "mutation_" >> block_number >> ".txt";
|
||
|
|
assertEOF(file_name_buf);
|
||
|
|
|
||
|
|
ReadBufferFromFile buf(path_prefix + file_name);
|
||
|
|
|
||
|
|
buf >> "format version: 1\n";
|
||
|
|
|
||
|
|
LocalDateTime create_time_dt;
|
||
|
|
buf >> "create time: " >> create_time_dt >> "\n";
|
||
|
|
create_time = create_time_dt;
|
||
|
|
|
||
|
|
buf >> "commands: ";
|
||
|
|
commands.readText(buf);
|
||
|
|
buf >> "\n";
|
||
|
|
|
||
|
|
assertEOF(buf);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
MergeTreeMutationEntry::~MergeTreeMutationEntry()
|
||
|
|
{
|
||
|
|
if (is_temp && startsWith(file_name, "tmp_"))
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
removeFile();
|
||
|
|
}
|
||
|
|
catch (...)
|
||
|
|
{
|
||
|
|
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|