mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
55 lines
1.1 KiB
C
55 lines
1.1 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <DB/Interpreters/InterserverIOHandler.h>
|
||
|
|
#include <DB/Storages/MergeTree/MergeTreeData.h>
|
||
|
|
#include <DB/IO/WriteBuffer.h>
|
||
|
|
#include <DB/Core/Types.h>
|
||
|
|
#include <common/logger_useful.h>
|
||
|
|
|
||
|
|
namespace DB
|
||
|
|
{
|
||
|
|
|
||
|
|
namespace RemotePartChecker
|
||
|
|
{
|
||
|
|
|
||
|
|
enum class Status : UInt8
|
||
|
|
{
|
||
|
|
OK = 0,
|
||
|
|
NOT_FOUND,
|
||
|
|
INCONSISTENT,
|
||
|
|
ERROR
|
||
|
|
};
|
||
|
|
|
||
|
|
class Service final : public InterserverIOEndpoint
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
Service(StoragePtr & storage_);
|
||
|
|
Service(const Service &) = delete;
|
||
|
|
Service & operator=(const Service &) = delete;
|
||
|
|
std::string getId(const std::string & node_id) const override;
|
||
|
|
void processQuery(const Poco::Net::HTMLForm & params, ReadBuffer & body, WriteBuffer & out) override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
StoragePtr owned_storage;
|
||
|
|
MergeTreeData & data;
|
||
|
|
Logger * log = &Logger::get("RemotePartChecker::Service");
|
||
|
|
};
|
||
|
|
|
||
|
|
class Client final
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
Client() = default;
|
||
|
|
Client(const Client &) = delete;
|
||
|
|
Client & operator=(const Client &) = delete;
|
||
|
|
|
||
|
|
Status check(const std::string & part_name, const std::string & hash,
|
||
|
|
const InterserverIOEndpointLocation & to_location);
|
||
|
|
|
||
|
|
private:
|
||
|
|
Logger * log = &Logger::get("RemotePartChecker::Client");
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|