Boost.Asio - 不执行所有处理程序

Boost.Asio - not executing all handlers

我正在尝试为 boost::io_context 创建一个适配器,它将始终在准备好的处理程序中选择一个具有最高优先级的处理程序来执行。我从 the official example 中获得灵感,但很快 运行 在一个场景中出现意外行为,其中一个处理程序在同一上下文中启动另一个异步操作。

这里是MCVE。我只修改了用户代码(//--- 下方),以调用低优先级处理程序,之后我希望调用高优先级和中优先级处理程序。仅调用低优先级处理程序。

#include <boost/asio.hpp>
#include <boost/function.hpp>
#include <iostream>
#include <queue>

class handler_priority_queue
{
public:
  void add(int priority, boost::function<void()> function)
  {
    handlers_.push(queued_handler(priority, function));
  }

  void execute_all()
  {
    while (!handlers_.empty())
    {
      queued_handler handler = handlers_.top();
      handler.execute();
      handlers_.pop();
    }
  }

  // A generic wrapper class for handlers to allow the invocation to be hooked.
  template <typename Handler>
  class wrapped_handler
  {
  public:
    wrapped_handler(handler_priority_queue& q, int p, Handler h)
      : queue_(q), priority_(p), handler_(h)
    {
    }

    void operator()()
    {
      handler_();
    }

    template <typename Arg1>
    void operator()(Arg1 arg1)
    {
      handler_(arg1);
    }

    template <typename Arg1, typename Arg2>
    void operator()(Arg1 arg1, Arg2 arg2)
    {
      handler_(arg1, arg2);
    }

  //private:
    handler_priority_queue& queue_;
    int priority_;
    Handler handler_;
  };

  template <typename Handler>
  wrapped_handler<Handler> wrap(int priority, Handler handler)
  {
    return wrapped_handler<Handler>(*this, priority, handler);
  }

private:
  class queued_handler
  {
  public:
    queued_handler(int p, boost::function<void()> f)
      : priority_(p), function_(f)
    {
    }

    void execute()
    {
      function_();
    }

    friend bool operator<(const queued_handler& a,
        const queued_handler& b)
    {
      return a.priority_ < b.priority_;
    }

  private:
    int priority_;
    boost::function<void()> function_;
  };

  std::priority_queue<queued_handler> handlers_;
};

// Custom invocation hook for wrapped handlers.
template <typename Function, typename Handler>
void asio_handler_invoke(Function f,
    handler_priority_queue::wrapped_handler<Handler>* h)
{
  h->queue_.add(h->priority_, f);
}

//----------------------------------------------------------------------

void high_priority_handler()
{
  std::cout << "High priority handler\n";
}

void middle_priority_handler()
{
  std::cout << "Middle priority handler\n";
}

void low_priority_handler(
  boost::asio::io_service& io_service,
  handler_priority_queue& pri_queue)
{
  std::cout << "Low priority handler\n";

  io_service.post(pri_queue.wrap(1, middle_priority_handler));
  io_service.post(pri_queue.wrap(2, high_priority_handler));
}

int main()
{
  boost::asio::io_service io_service;
  handler_priority_queue pri_queue;

  // Post a completion handler to be run immediately.
  io_service.post(pri_queue.wrap(
      0, std::bind(low_priority_handler,
                   std::ref(io_service), std::ref(pri_queue))));

  while (io_service.run_one())
  {
    // The custom invocation hook adds the handlers to the priority queue
    // rather than executing them from within the poll_one() call.
    while (io_service.poll_one())
      ;

    pri_queue.execute_all();
  }

  return 0;
}

如果我在 main 中的循环之后调用 io_service.restart() 并在之后复制粘贴同一循环,则其余处理程序将按预期顺序执行。调试时,我可以看到处理程序仅在 asio_handler_invoke 中排队一次。

为什么 boost::io_context 在第一个处理程序之后停止 运行?我要求的可能吗?

io_context 停止,因为在调用 poll_one 时没有任何准备好的 运行 处理程序。

[1] 第一个处理程序已发布:

io_service.post(pri_queue.wrap(0, std::bind(low_priority_handler,
                   std::ref(io_service), std::ref(pri_queue))));

[2] while (io_service.run_one())

等到有一个准备好的处理程序到 运行

[3] in run_one() 处理程序被执行。您已经定义了 asio_handler_invoke(),它提供了一些策略来调用处理程序的函数(主体)。默认策略只调用函数,在您的情况下,函数对象排队进入 handler_priority_queue,但 io_service's 队列不接收要执行的处理程序。那么,当 low_priority_handler 的主体(通过调用 io_service.postio_service 添加新的处理程序)被执行时?这个函数是从pri_queue.execute_all()开始执行的(是在poll_one之后调用的),而不是在poll_one()的调用期间。当 运行 没有任何准备好的处理程序时,io_service.poll_one()io_service 标记为已停止。这是你的情况。您可以在 pri_queue.execute_all() 之后重置 io_service,然后将调用所有处理程序。