mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
48 lines
1.3 KiB
C++
48 lines
1.3 KiB
C++
|
#pragma once
|
||
|
|
||
|
#include <Poco/Net/TCPServer.h>
|
||
|
|
||
|
#include <base/types.h>
|
||
|
#include <Server/TCPServerConnectionFactory.h>
|
||
|
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
class Context;
|
||
|
|
||
|
class TCPServer : public Poco::Net::TCPServer
|
||
|
{
|
||
|
public:
|
||
|
explicit TCPServer(
|
||
|
TCPServerConnectionFactory::Ptr factory,
|
||
|
Poco::ThreadPool & thread_pool,
|
||
|
Poco::Net::ServerSocket & socket,
|
||
|
Poco::Net::TCPServerParams::Ptr params = new Poco::Net::TCPServerParams);
|
||
|
|
||
|
/// Close the socket and ask existing connections to stop serving queries
|
||
|
void stop()
|
||
|
{
|
||
|
Poco::Net::TCPServer::stop();
|
||
|
// This notifies already established connections that they should stop serving
|
||
|
// queries and close their socket as soon as they can.
|
||
|
is_open = false;
|
||
|
// Poco's stop() stops listening on the socket but leaves it open.
|
||
|
// To be able to hand over control of the listening port to a new server, and
|
||
|
// to get fast connection refusal instead of timeouts, we also need to close
|
||
|
// the listening socket.
|
||
|
socket.close();
|
||
|
}
|
||
|
|
||
|
bool isOpen() const { return is_open; }
|
||
|
|
||
|
UInt16 portNumber() const { return port_number; }
|
||
|
|
||
|
private:
|
||
|
TCPServerConnectionFactory::Ptr factory;
|
||
|
Poco::Net::ServerSocket socket;
|
||
|
std::atomic<bool> is_open;
|
||
|
UInt16 port_number;
|
||
|
};
|
||
|
|
||
|
}
|