2020-05-30 17:04:02 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <Poco/Net/TCPServerConnectionFactory.h>
|
2020-05-30 20:02:11 +00:00
|
|
|
#include <atomic>
|
|
|
|
#include <memory>
|
|
|
|
#include <Server/IServer.h>
|
2020-05-30 17:04:02 +00:00
|
|
|
#include <Core/PostgreSQLProtocol.h>
|
|
|
|
|
2021-11-01 14:03:20 +00:00
|
|
|
#if !defined(ARCADIA_BUILD)
|
|
|
|
# include <Common/config.h>
|
|
|
|
#endif
|
|
|
|
|
2020-05-30 17:04:02 +00:00
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
class PostgreSQLHandlerFactory : public Poco::Net::TCPServerConnectionFactory
|
|
|
|
{
|
|
|
|
private:
|
|
|
|
IServer & server;
|
|
|
|
Poco::Logger * log;
|
2020-06-08 12:05:49 +00:00
|
|
|
|
|
|
|
#if USE_SSL
|
2020-05-30 17:04:02 +00:00
|
|
|
bool ssl_enabled = true;
|
|
|
|
#else
|
|
|
|
bool ssl_enabled = false;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
std::atomic<Int32> last_connection_id = 0;
|
|
|
|
std::vector<std::shared_ptr<PostgreSQLProtocol::PGAuthentication::AuthenticationMethod>> auth_methods;
|
|
|
|
|
|
|
|
public:
|
|
|
|
explicit PostgreSQLHandlerFactory(IServer & server_);
|
|
|
|
|
|
|
|
Poco::Net::TCPServerConnection * createConnection(const Poco::Net::StreamSocket & socket) override;
|
|
|
|
};
|
|
|
|
}
|