2020-12-09 08:45:36 +00:00
|
|
|
#pragma once
|
2021-02-01 13:18:17 +00:00
|
|
|
|
2021-03-29 08:24:56 +00:00
|
|
|
#include <Server/KeeperTCPHandler.h>
|
2020-12-09 08:45:36 +00:00
|
|
|
#include <Poco/Net/TCPServerConnectionFactory.h>
|
|
|
|
#include <Poco/Net/NetException.h>
|
2021-10-02 07:13:14 +00:00
|
|
|
#include <base/logger_useful.h>
|
2020-12-09 08:45:36 +00:00
|
|
|
#include <Server/IServer.h>
|
2021-04-12 12:25:52 +00:00
|
|
|
#include <string>
|
2020-12-09 08:45:36 +00:00
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2021-03-29 08:24:56 +00:00
|
|
|
class KeeperTCPHandlerFactory : public Poco::Net::TCPServerConnectionFactory
|
2020-12-09 08:45:36 +00:00
|
|
|
{
|
|
|
|
private:
|
|
|
|
IServer & server;
|
|
|
|
Poco::Logger * log;
|
|
|
|
class DummyTCPHandler : public Poco::Net::TCPServerConnection
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
using Poco::Net::TCPServerConnection::TCPServerConnection;
|
|
|
|
void run() override {}
|
|
|
|
};
|
|
|
|
public:
|
2021-04-12 12:25:52 +00:00
|
|
|
KeeperTCPHandlerFactory(IServer & server_, bool secure)
|
2020-12-09 08:45:36 +00:00
|
|
|
: server(server_)
|
2021-04-12 12:25:52 +00:00
|
|
|
, log(&Poco::Logger::get(std::string{"KeeperTCP"} + (secure ? "S" : "") + "HandlerFactory"))
|
2020-12-09 08:45:36 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
Poco::Net::TCPServerConnection * createConnection(const Poco::Net::StreamSocket & socket) override
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
2021-03-29 08:24:56 +00:00
|
|
|
LOG_TRACE(log, "Keeper request. Address: {}", socket.peerAddress().toString());
|
|
|
|
return new KeeperTCPHandler(server, socket);
|
2020-12-09 08:45:36 +00:00
|
|
|
}
|
|
|
|
catch (const Poco::Net::NetException &)
|
|
|
|
{
|
|
|
|
LOG_TRACE(log, "TCP Request. Client is not connected (most likely RST packet was sent).");
|
|
|
|
return new DummyTCPHandler(socket);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
}
|