mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-29 05:00:47 +00:00
143 lines
4.1 KiB
C++
143 lines
4.1 KiB
C++
#pragma once
|
||
|
||
#include <DB/Interpreters/InterserverIOHandler.h>
|
||
#include <DB/Storages/MergeTree/MergeTreeData.h>
|
||
#include <DB/IO/ReadBufferFromHTTP.h>
|
||
#include <DB/IO/HashingWriteBuffer.h>
|
||
#include <DB/IO/copyData.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
class ReplicatedMergeTreePartsServer : public InterserverIOEndpoint
|
||
{
|
||
public:
|
||
ReplicatedMergeTreePartsServer(MergeTreeData & data_, StoragePtr owned_storage_) : data(data_),
|
||
owned_storage(owned_storage_), log(&Logger::get(data.getLogName() + " (Replicated PartsServer)")) {}
|
||
|
||
void processQuery(const Poco::Net::HTMLForm & params, WriteBuffer & out) override
|
||
{
|
||
String part_name = params.get("part");
|
||
LOG_TRACE(log, "Sending part " << part_name);
|
||
|
||
auto storage_lock = owned_storage->lockStructure(false);
|
||
|
||
MergeTreeData::DataPartPtr part = findPart(part_name);
|
||
|
||
Poco::ScopedReadRWLock part_lock(part->columns_lock);
|
||
|
||
/// Список файлов возьмем из списка контрольных сумм.
|
||
MergeTreeData::DataPart::Checksums checksums = part->checksums;
|
||
/// Добавим файлы, которых нет в списке контрольных сумм.
|
||
checksums.files["checksums.txt"];
|
||
checksums.files["columns.txt"];
|
||
|
||
writeBinary(checksums.files.size(), out);
|
||
for (const auto & it : checksums.files)
|
||
{
|
||
String path = data.getFullPath() + part_name + "/" + it.first;
|
||
UInt64 size = Poco::File(path).getSize();
|
||
|
||
writeStringBinary(it.first, out);
|
||
writeBinary(size, out);
|
||
|
||
ReadBufferFromFile file_in(path);
|
||
HashingWriteBuffer hashing_out(out);
|
||
copyData(file_in, hashing_out);
|
||
|
||
if (hashing_out.count() != size)
|
||
throw Exception("Unexpected size of file " + path, ErrorCodes::BAD_SIZE_OF_FILE_IN_DATA_PART);
|
||
|
||
writeBinary(hashing_out.getHash(), out);
|
||
}
|
||
}
|
||
|
||
private:
|
||
MergeTreeData & data;
|
||
StoragePtr owned_storage;
|
||
|
||
Logger * log;
|
||
|
||
MergeTreeData::DataPartPtr findPart(const String & name)
|
||
{
|
||
MergeTreeData::DataPartPtr part = data.getContainingPart(name, true);
|
||
if (part && part->name == name)
|
||
return part;
|
||
throw Exception("No part " + name + " in table");
|
||
}
|
||
};
|
||
|
||
class ReplicatedMergeTreePartsFetcher
|
||
{
|
||
public:
|
||
ReplicatedMergeTreePartsFetcher(MergeTreeData & data_) : data(data_), log(&Logger::get("ReplicatedMergeTreePartsFetcher")) {}
|
||
|
||
/// Скачивает кусок в tmp_директорию.
|
||
MergeTreeData::MutableDataPartPtr fetchPart(
|
||
const String & part_name,
|
||
const String & replica_path,
|
||
const String & host,
|
||
int port)
|
||
{
|
||
ReadBufferFromHTTP::Params params = {
|
||
std::make_pair("endpoint", "ReplicatedMergeTree:" + replica_path),
|
||
std::make_pair("part", part_name),
|
||
std::make_pair("compress", "false")};
|
||
ReadBufferFromHTTP in(host, port, params);
|
||
|
||
String part_path = data.getFullPath() + "tmp_" + part_name + "/";
|
||
if (!Poco::File(part_path).createDirectory())
|
||
throw Exception("Directory " + part_path + " already exists");
|
||
|
||
MergeTreeData::MutableDataPartPtr new_data_part = std::make_shared<MergeTreeData::DataPart>(data);
|
||
new_data_part->name = "tmp_" + part_name;
|
||
new_data_part->is_temp = true;
|
||
|
||
size_t files;
|
||
readBinary(files, in);
|
||
MergeTreeData::DataPart::Checksums checksums;
|
||
for (size_t i = 0; i < files; ++i)
|
||
{
|
||
String file_name;
|
||
UInt64 file_size;
|
||
|
||
readStringBinary(file_name, in);
|
||
readBinary(file_size, in);
|
||
|
||
WriteBufferFromFile file_out(part_path + file_name);
|
||
HashingWriteBuffer hashing_out(file_out);
|
||
copyData(in, hashing_out, file_size);
|
||
|
||
uint128 expected_hash;
|
||
readBinary(expected_hash, in);
|
||
|
||
if (expected_hash != hashing_out.getHash())
|
||
throw Exception("Checksum mismatch for file " + part_path + file_name + " transferred from " + replica_path);
|
||
|
||
if (file_name != "checksums.txt" &&
|
||
file_name != "columns.txt")
|
||
checksums.addFile(file_name, file_size, expected_hash);
|
||
}
|
||
|
||
assertEOF(in);
|
||
|
||
ActiveDataPartSet::parsePartName(part_name, *new_data_part);
|
||
new_data_part->modification_time = time(0);
|
||
new_data_part->loadColumns();
|
||
new_data_part->loadChecksums();
|
||
new_data_part->loadIndex();
|
||
|
||
new_data_part->checksums.checkEqual(checksums, false);
|
||
|
||
return new_data_part;
|
||
}
|
||
|
||
private:
|
||
MergeTreeData & data;
|
||
|
||
Logger * log;
|
||
};
|
||
|
||
}
|