blob: b77f22f046800c8ffb6f818ccd0ffbc865c366e4 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#pragma once
#include "clickhouse_config.h"
#if USE_GRPC
#include <Poco/Net/SocketAddress.h>
#include <base/types.h>
#error #include "clickhouse_grpc.grpc.pb.h"
namespace Poco { class Logger; }
namespace grpc
{
class Server;
class ServerCompletionQueue;
}
namespace DB
{
class IServer;
class GRPCServer
{
public:
GRPCServer(IServer & iserver_, const Poco::Net::SocketAddress & address_to_listen_);
~GRPCServer();
/// Starts the server. A new thread will be created that waits for and accepts incoming connections.
void start();
/// Stops the server. No new connections will be accepted.
void stop();
/// Returns the port this server is listening to.
UInt16 portNumber() const { return address_to_listen.port(); }
/// Returns the number of currently handled connections.
size_t currentConnections() const;
/// Returns the number of current threads.
size_t currentThreads() const { return currentConnections(); }
private:
using GRPCService = clickhouse::grpc::ClickHouse::AsyncService;
class Runner;
IServer & iserver;
const Poco::Net::SocketAddress address_to_listen;
Poco::Logger * log;
GRPCService grpc_service;
std::unique_ptr<grpc::Server> grpc_server;
std::unique_ptr<grpc::ServerCompletionQueue> queue;
std::unique_ptr<Runner> runner;
};
}
#endif
|
