agent_control_server.cc

Go to the documentation of this file.

#include "cli/service/agent/agent_control_server.h"
 
#include <grpcpp/server.h>
#include <grpcpp/server_builder.h>
 
#include <iostream>
 
#include "cli/service/agent/emulator_service_impl.h"
 
namespace yaze::agent {
 
AgentControlServer::AgentControlServer(yaze::emu::Emulator* emulator)
    : emulator_(emulator) {}
 
AgentControlServer::~AgentControlServer() {
  Stop();
}
 
void AgentControlServer::Start() {
  server_thread_ = std::thread(&AgentControlServer::Run, this);
}
 
void AgentControlServer::Stop() {
  if (server_) {
    server_->Shutdown();
  }
  if (server_thread_.joinable()) {
    server_thread_.join();
  }
}
 
void AgentControlServer::Run() {
  std::string server_address("0.0.0.0:50051");
  EmulatorServiceImpl service(emulator_);
 
  grpc::ServerBuilder builder;
  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
  builder.RegisterService(&service);
 
  server_ = builder.BuildAndStart();
  if (server_) {
    std::cout << "AgentControlServer listening on " << server_address
              << std::endl;
    server_->Wait();
  } else {
    std::cerr << "Failed to start AgentControlServer on " << server_address
              << std::endl;
  }
}
 
}  // namespace yaze::agent