每个线程完成后停止io_service

时间:2018-04-28 12:10:41

标签: c++ boost boost-asio boost-thread

我想让程序等到它完成所有正在运行的线程,而不是ioService.stop();,这会在不等待的情况下停止ioService。我尝试了下面的代码,它运行正常,但在不等待线程完成的情况下停止ioService

#include <iostream>
#include <boost/asio/io_service.hpp>
#include <boost/bind.hpp>
#include <boost/thread/thread.hpp>


void myTask (std::string &str);

int main(int argc, char **argv){

    uint16_t total_threads = 4;

    /*
     * Create an asio::io_service and a thread_group
     */
    boost::asio::io_service ioService;
    boost::thread_group threadpool;    

    /*
     * This will start the ioService processing loop. 
     */     
    boost::asio::io_service::work work(ioService);

    /*
     * This will add threads to the thread pool.
     */
    for (std::size_t i = 0; i < total_threads; ++i)
        threadpool.create_thread(
                boost::bind(&boost::asio::io_service::run, &ioService));    

    /*
     * This will assign tasks to the thread pool.
     */
    std::string str = "Hello world";
    ioService.post(boost::bind(myTask, std::ref(str) ));



    ioService.stop();

    /*
     * thread pool are finished with
     * their assigned tasks and 'join' them.
     */
    threadpool.join_all();

    return 0;

}


void myTask (std::string &str){
    std::cout << str << std::endl;
}

编译:{{1​​}}

1 个答案:

答案 0 :(得分:3)

您的问题是您正在创建work作为堆栈上的变量。 work告诉io_service仍有工作要做。从手册:

  

析构函数通知io_service工作已完成。

由于工作是在堆栈的main中创建的,因此它的生命周期比你想要的要长。直到主要出口才会被破坏。而是在堆上创建它,因此您可以显式地销毁它。将其更改为:

using namespace boost::asio;
boost::scoped_ptr<io_service::work> work(new io_service::work(ioService));

然后,稍后,当你想告诉io_service在完成所有未完成的工作后停止时,不要停止io_service但是要破坏工作&#39;相反,然后等待线程完成。

work.reset();
threadpool.join_all();

这将调用~work(),它将从io_service中删除工作对象。这反过来会导致io_service::run在最后一个挂起操作完成时退出。

还有一些说明:

  • 我会避免给出与其类同名的变量。我不会写io_service::work work(io_service);这太令人困惑了。我会写一些像io_service::work some_work(io_service);
  • 这样的东西
  • 小心io_service.post(... std::ref(str));您正在传递对io_service帖子操作的引用。变量str必须足够长时间才能完成任务。我相信这只是为了这个例子。在现实世界的应用程序中,令人惊讶地难以确保传递给工作对象的参数不会过早地被破坏。我经常使用shared_ptr<>,或者在不可能的情况下,我有时会使用boost :: atomic
  • 计算未完成的io_service操作的数量