Skip to content
Snippets Groups Projects
server.hh 2.07 KiB
Newer Older
  • Learn to ignore specific revisions
  • #ifndef __VEREIGN_GRPC_SERVER_API_HH
    #define __VEREIGN_GRPC_SERVER_API_HH
    
    #include <thread>
    #include <grpcpp/server.h>
    #include <boost/asio/io_context.hpp>
    #include <boost/asio/ssl/context.hpp>
    #include <boost/asio/executor_work_guard.hpp>
    
    namespace vereign {
    
    namespace restapi {
    class Client;
    class ClientSession;
    }
    
    namespace grpc {
    
    namespace asio = boost::asio;
    
    /**
     * Server is a grpc server that provides the Vereign services.
     *
     * It bootstraps the grpc server, http client, http client session and
     * Vereign services.
     */
    class Server {
    public:
      /**
       * Constructs and bootstraps the server.
       *
       * @param listenAddress gRPC listen address, for example "localhost:".
       * @param vereignHost Vereign restapi host.
       * @param vereignPort Vereign restapi port - https, 443...
       */
      explicit Server(
        const std::string& listenAddress,
        const std::string& vereignHost,
        const std::string& vereignPort,
        // FIXME: the public key must come from a storage
        const std::string& publicKey
      );
    
      /**
       * Shutdowns the server.
       *
       * @see Server::Shutdown
       */
      ~Server();
    
      // Disable copying
      Server(const Server&) = delete;
      Server& operator=(const Server&) = delete;
    
      /**
       * Shutdown the server.
       *
       * It will cancel all pending http requests to the Vereign restapi.
       * Blocks for all pending gRPC handlers to finish.
       */
      void Shutdown();
    
      /**
       * Returns the port that the gRPC server listens to.
       *
       * This is useful if you construct the server with ephemeral port.
       * Then this method will return the port that the OS assigned to the gRPC
       * socket.
       */
      int SelectedPort() const;
    
    private:
      int selected_port_;
      asio::io_context ioc_;
      asio::executor_work_guard<asio::io_context::executor_type> work_guard_;
      asio::ssl::context ssl_context_;
      std::unique_ptr<vereign::restapi::Client> client_;
      std::unique_ptr<vereign::restapi::ClientSession> client_session_;
      std::vector<std::unique_ptr<::grpc::Service>> services_;
      std::unique_ptr<::grpc::Server> server_;
      std::thread server_thread_;
      std::thread service_thread_;
    };
    
    }
    }
    
    #endif