mirror of
https://gitee.com/TarsCloud/TarsCpp.git
synced 2025-01-05 17:42:24 +08:00
153 lines
3.3 KiB
C++
153 lines
3.3 KiB
C++
/**
|
|
* Tencent is pleased to support the open source community by making Tars available.
|
|
*
|
|
* Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
|
|
*
|
|
* Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
|
|
* in compliance with the License. You may obtain a copy of the License at
|
|
*
|
|
* https://opensource.org/licenses/BSD-3-Clause
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software distributed
|
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations under the License.
|
|
*/
|
|
|
|
#include "util/tc_thread_pool.h"
|
|
#include "util/tc_common.h"
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
using namespace tars;
|
|
|
|
TC_ThreadPool tpool;
|
|
TC_ThreadLock l;
|
|
|
|
/**
|
|
* 线程私有数据
|
|
*/
|
|
class MyThreadData : public TC_ThreadPool::ThreadData
|
|
{
|
|
public:
|
|
virtual ~MyThreadData()
|
|
{
|
|
cout << std::this_thread::get_id() << endl;
|
|
}
|
|
|
|
public:
|
|
pthread_t _idata;
|
|
};
|
|
|
|
/**
|
|
* 线程初始化
|
|
*/
|
|
void threadInitialize()
|
|
{
|
|
MyThreadData *p = TC_ThreadPool::ThreadData::makeThreadData<MyThreadData>();
|
|
p->_idata = std::this_thread::get_id();
|
|
TC_ThreadPool::setThreadData(p);
|
|
|
|
cout << p->_idata << endl;
|
|
}
|
|
|
|
/**
|
|
* 线程调用过程
|
|
* @param s
|
|
* @param i
|
|
*/
|
|
void TestFunction3(const string &s, int i)
|
|
{
|
|
MyThreadData *p = (MyThreadData*)TC_ThreadPool::getThreadData();
|
|
assert(std::this_thread::get_id() == p->_idata);
|
|
|
|
// cout << std::this_thread::get_id() << " | TestFunction3('" << s << "', " << i << ")" << endl;
|
|
}
|
|
|
|
/**
|
|
* 运行线程池
|
|
*/
|
|
void testThreadPool()
|
|
{
|
|
//4个线程
|
|
tpool.init(4);
|
|
|
|
//启动线程, 指定初始化对象,也可以没有初始化对象:tpool.start();
|
|
tpool.start(threadInitialize);
|
|
|
|
string s("a");
|
|
int i = 1000000;
|
|
|
|
//调用i次
|
|
while(i)
|
|
{
|
|
tpool.exec(std::bind(&TestFunction3, std::cref(s), i));
|
|
--i;
|
|
}
|
|
|
|
//等待线程结束
|
|
cout << "waitForAllDone..." << endl;
|
|
bool b = tpool.waitForAllDone(1000);
|
|
cout << "waitForAllDone..." << b << ":" << tpool.getJobNum() << endl;
|
|
|
|
//停止线程,析够的时候也会自动停止线程
|
|
//线程结束时,会自动释放私有数据
|
|
tpool.stop();
|
|
}
|
|
|
|
void test(int i, string &s)
|
|
{
|
|
cout << i << ":" << s << endl;
|
|
s = TC_Common::tostr(i + 10);
|
|
}
|
|
|
|
/**
|
|
* 运行线程池
|
|
*/
|
|
void testThreadPool1()
|
|
{
|
|
//4个线程
|
|
tpool.init(1);
|
|
|
|
//启动线程, 指定初始化对象,也可以没有初始化对象:tpool.start();
|
|
tpool.start();
|
|
|
|
string bid;
|
|
for(int i=0; i<10; i++)
|
|
{
|
|
bid = TC_Common::tostr(i);
|
|
cout << bid << endl;
|
|
|
|
cout << "index = " << i << ",bid = " << bid << endl;
|
|
|
|
tpool.exec(std::bind(&test, i, std::ref(bid)));
|
|
}
|
|
|
|
//等待线程结束
|
|
cout << "waitForAllDone..." << endl;
|
|
bool b = tpool.waitForAllDone(1000);
|
|
cout << "waitForAllDone..." << b << ":" << tpool.getJobNum() << endl;
|
|
|
|
//停止线程,析够的时候也会自动停止线程
|
|
//线程结束时,会自动释放私有数据
|
|
tpool.stop();
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
try
|
|
{
|
|
testThreadPool1();
|
|
}
|
|
catch(exception &ex)
|
|
{
|
|
cout << ex.what() << endl;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|