signal_waiter.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include <boost/asio/io_context.hpp>
#include <boost/asio/signal_set.hpp>
#include <spdlog/spdlog.h>
 
namespace sol3::net {
 
// Minimal helper to block until SIGINT or SIGTERM is received
class SignalWaiter {
 public:
  SignalWaiter(boost::asio::io_context& io)
      : io_(io), signals_(io_, SIGINT, SIGTERM) {
    signals_.async_wait(
        [this](std::error_code /*ec*/, int /*signal_number*/) { io_.stop(); });
  }
 
  void run(std::function<void()> on_shutdown = [] {}) {
    SPDLOG_INFO("Running... press Ctrl+C to exit");
    io_.run();
    on_shutdown();
  }
 
 private:
  boost::asio::io_context& io_;
  boost::asio::signal_set signals_;
};
 
}  // namespace sol3::net