server.cc

浏览该文件的文档.

 
#include "modules/contrib/cyber_bridge/server.h"
 
#include <signal.h>
 
#include <cstdint>
#include <functional>
#include <memory>
#include <utility>
 
#include "boost/bind.hpp"
#include "cyber/common/log.h"
#include "gflags/gflags.h"
#include "modules/contrib/cyber_bridge/client.h"
 
DEFINE_int32(port, 9090, "tcp listen port");
 
Server::Server(Node* node)
    : node(*node),
      signals(io),
      endpoint(boost::asio::ip::tcp::v4(), (uint16_t)FLAGS_port),
      acceptor(io, endpoint),
      socket(io) {
  signals.add(SIGTERM);
  signals.add(SIGINT);
}
 
Server::~Server() {}
 
void Server::run() {
  signals.async_wait(boost::bind(&Server::stop, shared_from_this(),
                                 boost::asio::placeholders::error,
                                 boost::asio::placeholders::signal_number));
 
  begin_accept();
 
  io.run();
}
 
void Server::stop(const boost::system::error_code& ec, int signal_number) {
  if (ec) {
    AERROR << "Error waiting on signals: " << ec.message();
  } else {
    AINFO << "Signal " << signal_number << " received, stopping server";
  }
  acceptor.close();
  clients.stop_all();
}
 
void Server::begin_accept() {
  acceptor.async_accept(socket,
                        boost::bind(&Server::end_accept, shared_from_this(),
                                    boost::asio::placeholders::error));
}
 
void Server::end_accept(const boost::system::error_code& ec) {
  if (!acceptor.is_open()) {
    return;
  }
 
  if (ec) {
    AERROR << "Error accepting connection: " << ec.message();
  } else {
    auto client = std::make_shared<Client>(&node, &clients, std::move(socket));
    clients.start(client);
  }
 
  begin_accept();
}