mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 19:32:07 +00:00
78fc36ca49
This makes the target location consistent with other auto-generated files like config_formats.h, config_core.h, and config_functions.h and simplifies the build of clickhouse_common.
83 lines
2.1 KiB
C++
83 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include <Common/CurrentMetrics.h>
|
|
#include "config.h"
|
|
#include <Core/PostgreSQLProtocol.h>
|
|
#include <Poco/Net/TCPServerConnection.h>
|
|
#include <Common/logger_useful.h>
|
|
#include "IServer.h"
|
|
|
|
#if USE_SSL
|
|
# include <Poco/Net/SecureStreamSocket.h>
|
|
#endif
|
|
|
|
namespace CurrentMetrics
|
|
{
|
|
extern const Metric PostgreSQLConnection;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
class ReadBufferFromPocoSocket;
|
|
class Session;
|
|
class TCPServer;
|
|
|
|
/** PostgreSQL wire protocol implementation.
|
|
* For more info see https://www.postgresql.org/docs/current/protocol.html
|
|
*/
|
|
class PostgreSQLHandler : public Poco::Net::TCPServerConnection
|
|
{
|
|
public:
|
|
PostgreSQLHandler(
|
|
const Poco::Net::StreamSocket & socket_,
|
|
IServer & server_,
|
|
TCPServer & tcp_server_,
|
|
bool ssl_enabled_,
|
|
Int32 connection_id_,
|
|
std::vector<std::shared_ptr<PostgreSQLProtocol::PGAuthentication::AuthenticationMethod>> & auth_methods_);
|
|
|
|
void run() final;
|
|
|
|
private:
|
|
Poco::Logger * log = &Poco::Logger::get("PostgreSQLHandler");
|
|
|
|
IServer & server;
|
|
TCPServer & tcp_server;
|
|
std::unique_ptr<Session> session;
|
|
bool ssl_enabled = false;
|
|
Int32 connection_id = 0;
|
|
Int32 secret_key = 0;
|
|
|
|
std::shared_ptr<ReadBufferFromPocoSocket> in;
|
|
std::shared_ptr<WriteBuffer> out;
|
|
std::shared_ptr<PostgreSQLProtocol::Messaging::MessageTransport> message_transport;
|
|
|
|
#if USE_SSL
|
|
std::shared_ptr<Poco::Net::SecureStreamSocket> ss;
|
|
#endif
|
|
|
|
PostgreSQLProtocol::PGAuthentication::AuthenticationManager authentication_manager;
|
|
|
|
CurrentMetrics::Increment metric_increment{CurrentMetrics::PostgreSQLConnection};
|
|
|
|
void changeIO(Poco::Net::StreamSocket & socket);
|
|
|
|
bool startup();
|
|
|
|
void establishSecureConnection(Int32 & payload_size, Int32 & info);
|
|
|
|
void makeSecureConnectionSSL();
|
|
|
|
void sendParameterStatusData(PostgreSQLProtocol::Messaging::StartupMessage & start_up_message);
|
|
|
|
void cancelRequest();
|
|
|
|
std::unique_ptr<PostgreSQLProtocol::Messaging::StartupMessage> receiveStartupMessage(int payload_size);
|
|
|
|
void processQuery();
|
|
|
|
static bool isEmptyQuery(const String & query);
|
|
};
|
|
|
|
}
|