server.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <chrono>
 
#include <viam/sdk/common/grpc_fwd.hpp>
#include <viam/sdk/resource/resource.hpp>
#include <viam/sdk/resource/resource_api.hpp>
#include <viam/sdk/resource/resource_server_base.hpp>
 
namespace grpc {
 
class Service;
 
}  // namespace grpc
 
namespace viam {
 
// needed for later `friend` declaration.
namespace sdktests {
class TestServer;
}
 
namespace sdk {
 
class Server {
   public:
    Server();
    ~Server();
 
    void start();
 
    void register_service(::grpc::Service* service);
 
    std::shared_ptr<ResourceServer> lookup_resource_server(const API& api);
 
    void add_resource(std::shared_ptr<Resource> resource);
 
    void add_resource(std::shared_ptr<Resource> resource,
                      boost::optional<std::chrono::system_clock::time_point> deadline);
 
    void add_listening_port(const std::string& address,
                            std::shared_ptr<GrpcServerCredentials> creds = nullptr);
 
    void wait();
 
    void shutdown();
 
    // friend declaration to allow TestServer to access the private server_ field.
    friend class ::viam::sdktests::TestServer;
 
   private:
    std::unordered_map<API, std::shared_ptr<ResourceServer>> managed_servers_;
    std::unique_ptr<GrpcServerBuilder> builder_;
    std::unique_ptr<GrpcServer> server_;
};
 
}  // namespace sdk
}  // namespace viam