{

explicit TCPServerConnectionFilter(std::function<bool()> filter_func_) : filter_func(std::move(filter_func_)) {}

bool accept(const Poco::Net::StreamSocket &) override { return filter_func(); }

~TCPServerConnectionFilter() override = default;

std::function<bool()> filter_func;

{

explicit TCPServer(

TCPServerConnectionFactory::Ptr factory,

Poco::ThreadPool & thread_pool,

Poco::Net::ServerSocket & socket,

Poco::Net::TCPServerParams::Ptr params = new Poco::Net::TCPServerParams,

const TCPServerConnectionFilter::Ptr & filter = nullptr);

/// Close the socket and ask existing connections to stop serving queries

void stop()

{

if (!is_open)

return;

// FIXME: On darwin calling shutdown(SHUT_RD) on the socket blocked in accept() leads to ENOTCONN

#ifndef OS_DARWIN

// Shutdown the listen socket before stopping tcp server to avoid 2.5second delay

socket.shutdownReceive();

#endif

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; }

const Poco::Net::ServerSocket& getSocket() { return socket; }

TCPServerConnectionFactory::Ptr factory;

Poco::Net::ServerSocket socket;

std::atomic<bool> is_open;

UInt16 port_number;