#include <thread>
#include <iostream>
#include "glock.h"
#include "mylogger.h"
extern std::mutex g_mtx;
extern std::mutex g_mtx_main;
extern std::condition_variable cvMain;
extern std::condition_variable cvWorker;
extern int g_numThreads;
int nDevice = 3;
bool allThreadsCompleted = false;
void controlThreadFunction(int id, int casesize, int devicesize, bool bstop = false) {
std::cout << "contorl Thread " << id << " is running." << std::endl;
int index = 0;
int nCount = casesize;
while (!bstop && index++ < casesize) {
{
std::unique_lock<std::mutex> lock(g_mtx);
// 设置工作线程数量
g_numThreads = devicesize;
// 重置标志
allThreadsCompleted = false;
}
// 主线程等待所有工作线程完成
Logger::getInstance().logToConsole("controlthread:main thread wait workthread");
{
std::unique_lock<std::mutex> lock(g_mtx);
cvMain.wait(lock, [] { return allThreadsCompleted; });
}
// 短暂的延时，留出时间确保所有工作线程都进入等待状态
std::this_thread::sleep_for(std::chrono::milliseconds(1000))
// 主线程通知所有工作线程可以继续 Logger::getInstance().logToConsole("controlthread:main thread notify all sub-work thread"); cvWorker.notify_all(); }
Logger::getInstance().logToConsole("controlthread:exit");
}
void workerThreadFunction(int id, int casesize, int devicesize, bool bstop = false) { int nCount = casesize; std::this_thread::sleep_for(std::chrono::seconds(2)); for (int index = 0; index < nCount; ++index) { Logger::getInstance().logToConsole("workerthread:.......do somethings ...." + std::to_string(index) + "_thread:" + std::to_string(id));
{ std::lock_guard<std::mutex> lock(g_mtx); g_numThreads--; // 设置标志表示所有工作线程完成
if (g_numThreads == 0) { allThreadsCompleted = true; cvMain.notify_one(); // 通知主线程 Logger::getInstance().logToConsole("workerthread:.......notify main thread " + std::to_string(index) + "_thread:" + std::to_string(id)); } } Logger::getInstance().logToConsole("workerthread:.......wait main thread notify " + std::to_string(index) + "_thread:" + std::to_string(id));
// 进入等待，等待主线程通知 { std::unique_lock<std::mutex> lock(g_mtx); cvWorker.wait(lock); } }
Logger::getInstance().logToConsole("workthread:exit"); } int main() {
//Logger& ins = Logger::getInstance();
//nDevice //std::thread threads[3];
g_numThreads = 3;
std::thread control_thread = std::thread(controlThreadFunction, 1+ nDevice, 3, nDevice,false); std::vector<std::thread> threads;
for (int i = 0; i < nDevice; ++i) { threads.emplace_back(workerThreadFunction, i + 1, 3, nDevice, false); }
//for (int i = 0; i < nDevice; ++i) { // threads[i] = std::thread(workerThreadFunction, i + 1,2, nDevice,false); //}
control_thread.join();
for (int i = 0; i < nDevice; ++i) { threads[i].join(); }
std::cout << "All threads have finished." << std::endl; } 标签：std,同步,变量,thread,int,lock,nDevice,条件,Logger From： https://www.cnblogs.com/denli/p/17828709.html