c++ - 在每个线程完成后停止 io_service

Question

我想让程序等到它完成所有正在运行的线程，这与 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;
}

编译:-lboost_serialization -lboost_thread -lboost_system

Answer 1

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

Destructor notifies the io_service that the work is complete.

由于工作是在堆栈上的 main 中创建的，因此它的生命周期比您希望的要长。在 main 退出之前，它不会被销毁。相反，在堆上创建它，因此您可以显式销毁它。将其更改为:

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

然后，稍后，当您想要告诉 io_service 在完成所有未完成的工作后停止时，不要停止 io_service 而是销毁“工作”，然后等待线程完成。

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 post 操作的引用。变量 str 必须存在足够长的时间才能完成任务。我确信这只是为了举例。在现实世界的应用程序中，确保传递给工作对象的参数不会过早销毁可能出奇地困难。我用 shared_ptr<>很多，或者在不可能的情况下，我有时会用 boost::atomic
计算未完成的 io_service 操作的数量