ClickHouse/dbms/include/DB/Interpreters/InterserverIOHandler.h

#pragma once

#include <DB/IO/ReadBuffer.h>
#include <DB/IO/WriteBuffer.h>
#include <DB/Core/Types.h>
#include <map>
#include <Poco/Net/HTMLForm.h>

namespace DB
{

/** Обработчик запросов от других серверов.
  */
class InterserverIOEndpoint
{
public:
	virtual void processQuery(const Poco::Net::HTMLForm & params, WriteBuffer & out) = 0;

	virtual ~InterserverIOEndpoint() {}
};

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;
};

}