Tars框架线程池TC_ThreadPool的使用--执行lambda形式任务task

比眉伴天荒 2022-03-17 10:36 241阅读 0赞

线程池是我们在开发多线程程序中几乎是必不可少的一部分。  
今天实践一下Tars框架的线程池TC\_ThreadPool，本例采用一个lambda函数作为线程执行的任务，源码如下：

#pragma once
    
    #include "../../tars/util/include/util/tc_thread_pool.h"
    #include <iostream>
    #include <set>
    
    using namespace std;
    using namespace tars;
    
    namespace lesliefishtest
    {
        class TestThreadPool
        {
        public:
            static void testRunLambdaTask()
            {
                TC_ThreadPool tpool;
                
                // 4个线程
                tpool.init(4);
    
                // 启动线程池,tpool.start();
                tpool.start();
    
                // 执行任务的线程id集合
                std::set<pthread_t> threadIds{};
    
                for (int i = 0; i < 100000; i++)
                {
                    tpool.exec([i, &threadIds]
                    {
                        cout << "index = " << i << ", current thread id : " << pthread_self() << endl;
                        threadIds.insert(pthread_self());
                    });
                }
    
                // 等待线程结束
                cout << "waitForAllDone..." << endl;
                bool b = tpool.waitForAllDone(-1); // -1 表示等待执行完
                cout << "waitForAllDone..." << b << ":" << tpool.getJobNum() << endl;
    
                // 停止线程,析够的时候也会自动停止线程
                // 线程结束时,会自动释放私有数据
                tpool.stop();
    
                // 打印出执行任务的线程id
                cout << "执行任务的线程id : " << endl;
                for(const auto& iter: threadIds)
                {
                    cout << iter << endl;
                }
            }
    
        };
    }

--------------------

程序执行结果如下：

...
    index = 99987, current thread id : 140079190062848
    index = 99988, current thread id : 140079190062848
    index = 99989, current thread id : 140079190062848
    index = 99990, current thread id : 140079190062848
    index = 99991, current thread id : 140079190062848
    index = 99992, current thread id : 140079190062848
    index = 99993, current thread id : 140079190062848
    index = 99994, current thread id : 140079190062848
    index = 99995, current thread id : 140079190062848
    index = 99996, current thread id : 140079190062848
    index = 99997, current thread id : 140079190062848
    index = 99998, current thread id : 140079190062848
    index = 99999, current thread id : 140079190062848
    waitForAllDone...1:0
    执行任务的线程id :
    140079190062848
    140079198455552
    140079215240960
    140079223633664

--------------------

说明：  
1、本例开4个线程，用一个std::set<pthread\_t>类型的变量threadIds存储执行的线程的id，用来验证我们的确是4个线程的线程池。  
2、tpool.waitForAllDone()这个函数参数传入-1表示执行完任务停止，如果传入大于0的数值，如1000，线程池会在1000毫秒后终止执行任务。  
3、pthread\_self()函数是linux线程库中返回线程id的接口，C++11中对应的写法为：

std::this_thread::get_id();

和pthread\_self()获取的结果是一样的。

--------------------

实践环境：vs2017+ubuntu14.04  
代码路径：  
[https://github.com/lesliefish/FishTars/blob/master/code/FishTars/test/utiltest/test\_thread\_pool.h][https_github.com_lesliefish_FishTars_blob_master_code_FishTars_test_utiltest_test_thread_pool.h]

[https_github.com_lesliefish_FishTars_blob_master_code_FishTars_test_utiltest_test_thread_pool.h]: https://github.com/lesliefish/FishTars/blob/master/code/FishTars/test/utiltest/test_thread_pool.h