mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
51 lines
916 B
C++
51 lines
916 B
C++
#pragma once
|
|
|
|
#include <Poco/Net/TCPServerConnectionFactory.h>
|
|
#include <atomic>
|
|
#include "IServer.h"
|
|
|
|
#if !defined(ARCADIA_BUILD)
|
|
# include <Common/config.h>
|
|
#endif
|
|
|
|
#if USE_SSL
|
|
# include <openssl/rsa.h>
|
|
#endif
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class MySQLHandlerFactory : public Poco::Net::TCPServerConnectionFactory
|
|
{
|
|
private:
|
|
IServer & server;
|
|
Poco::Logger * log;
|
|
|
|
#if USE_SSL
|
|
struct RSADeleter
|
|
{
|
|
void operator()(RSA * ptr) { RSA_free(ptr); }
|
|
};
|
|
using RSAPtr = std::unique_ptr<RSA, RSADeleter>;
|
|
|
|
RSAPtr public_key;
|
|
RSAPtr private_key;
|
|
|
|
bool ssl_enabled = true;
|
|
#else
|
|
bool ssl_enabled = false;
|
|
#endif
|
|
|
|
std::atomic<size_t> last_connection_id = 0;
|
|
public:
|
|
explicit MySQLHandlerFactory(IServer & server_);
|
|
|
|
void readRSAKeys();
|
|
|
|
void generateRSAKeys();
|
|
|
|
Poco::Net::TCPServerConnection * createConnection(const Poco::Net::StreamSocket & socket) override;
|
|
};
|
|
|
|
}
|