server.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <grpcpp/impl/service_type.h>
#include <grpcpp/security/server_credentials.h>
#include <grpcpp/server_builder.h>
 
#include <viam/sdk/resource/resource.hpp>
#include <viam/sdk/resource/resource_api.hpp>
#include <viam/sdk/resource/resource_server_base.hpp>
 
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_listening_port(const std::string& address,
                            std::shared_ptr<grpc::ServerCredentials> 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<grpc::ServerBuilder> builder_;
    std::unique_ptr<grpc::Server> server_;
};
 
}  // namespace sdk
}  // namespace viam