mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-24 10:40:49 +00:00
113 lines
2.8 KiB
C++
113 lines
2.8 KiB
C++
#pragma once
|
||
|
||
#include <DB/IO/ReadBuffer.h>
|
||
#include <DB/IO/WriteBuffer.h>
|
||
#include <DB/Core/Types.h>
|
||
#include <map>
|
||
#include <atomic>
|
||
#include <Poco/Net/HTMLForm.h>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
namespace ErrorCodes
|
||
{
|
||
extern const int DUPLICATE_INTERSERVER_IO_ENDPOINT;
|
||
extern const int NO_SUCH_INTERSERVER_IO_ENDPOINT;
|
||
}
|
||
|
||
|
||
/** Обработчик запросов от других серверов.
|
||
*/
|
||
class InterserverIOEndpoint
|
||
{
|
||
public:
|
||
virtual void processQuery(const Poco::Net::HTMLForm & params, WriteBuffer & out) = 0;
|
||
virtual ~InterserverIOEndpoint() {}
|
||
|
||
void cancel() { is_cancelled = true; }
|
||
|
||
protected:
|
||
/// Нужно остановить передачу данных.
|
||
std::atomic<bool> is_cancelled {false};
|
||
};
|
||
|
||
typedef Poco::SharedPtr<InterserverIOEndpoint> InterserverIOEndpointPtr;
|
||
|
||
|
||
/** Сюда можно зарегистрировать сервис, обрататывающий запросы от других серверов.
|
||
* Используется для передачи кусков в ReplicatedMergeTree.
|
||
*/
|
||
class InterserverIOHandler
|
||
{
|
||
public:
|
||
void addEndpoint(const String & name, InterserverIOEndpointPtr endpoint)
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
if (endpoint_map.count(name))
|
||
throw Exception("Duplicate interserver IO endpoint: " + name, ErrorCodes::DUPLICATE_INTERSERVER_IO_ENDPOINT);
|
||
endpoint_map[name] = endpoint;
|
||
}
|
||
|
||
void removeEndpoint(const String & name)
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
if (!endpoint_map.count(name))
|
||
throw Exception("No interserver IO endpoint named " + name, ErrorCodes::NO_SUCH_INTERSERVER_IO_ENDPOINT);
|
||
endpoint_map.erase(name);
|
||
}
|
||
|
||
InterserverIOEndpointPtr getEndpoint(const String & name)
|
||
{
|
||
Poco::ScopedLock<Poco::FastMutex> lock(mutex);
|
||
if (!endpoint_map.count(name))
|
||
throw Exception("No interserver IO endpoint named " + name, ErrorCodes::NO_SUCH_INTERSERVER_IO_ENDPOINT);
|
||
return endpoint_map[name];
|
||
}
|
||
|
||
private:
|
||
typedef std::map<String, InterserverIOEndpointPtr> EndpointMap;
|
||
|
||
EndpointMap endpoint_map;
|
||
Poco::FastMutex mutex;
|
||
};
|
||
|
||
/// В конструкторе вызывает addEndpoint, в деструкторе - removeEndpoint.
|
||
class InterserverIOEndpointHolder
|
||
{
|
||
public:
|
||
InterserverIOEndpointHolder(const String & name_, InterserverIOEndpointPtr endpoint_, InterserverIOHandler & handler_)
|
||
: name(name_), endpoint(endpoint_), handler(handler_)
|
||
{
|
||
handler.addEndpoint(name, endpoint);
|
||
}
|
||
|
||
InterserverIOEndpointPtr getEndpoint()
|
||
{
|
||
return endpoint;
|
||
}
|
||
|
||
~InterserverIOEndpointHolder()
|
||
{
|
||
try
|
||
{
|
||
handler.removeEndpoint(name);
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException("~InterserverIOEndpointHolder");
|
||
}
|
||
}
|
||
|
||
void cancel() { endpoint->cancel(); }
|
||
|
||
private:
|
||
String name;
|
||
InterserverIOEndpointPtr endpoint;
|
||
InterserverIOHandler & handler;
|
||
};
|
||
|
||
typedef Poco::SharedPtr<InterserverIOEndpointHolder> InterserverIOEndpointHolderPtr;
|
||
|
||
}
|