ZBLOG

bool OBSThreadPool::init(size_t num) { std::unique_lock lock(_mutex); if (!_threads.empty()) { return false; } _threadNum = num; return true; } void OBSThreadPo...

为了实现修改线程池中线程的个数和使每个线程都能独立控制是否退出,您需要对代码进行一些修改。下面是一个可能的解决方案:

void OBSThreadPool::modify(size_t newNum)
{
    std::unique_lock<std::mutex> lock(_mutex);

    if (newNum == _threadNum) {
        return;  // 如果新旧线程数量相同,则直接返回
    }

    if (newNum < _threadNum) {
        size_t diff = _threadNum - newNum;
        for (size_t i = 0; i < diff; ++i) {
            _tasks.push_back(nullptr); // 添加空任务以退出多余线程
            _condition.notify_one();   // 通知等待的线程处理退出信号
        }
    } else {
        size_t diff = newNum - _threadNum;
        for (size_t i = 0; i < diff; ++i) {
            _threads.push_back(new thread(&OBSThreadPool::run, this));
        }
    }

    _threadNum = newNum;
}

void OBSThreadPool::stop()
{
    std::unique_lock<std::mutex> lock(_mutex);

    for(auto& thread : _threads)
    {
        if(thread->joinable())
        {
            thread->join();
        }
        delete thread;
    }
    
	_threads.clear();

	_bTerminate = true;
}

bool OBSThreadPool::isTerminate(size_t index)
{
    std::unique_lock<std::mutex> lock(_terminateMutex);
    
	if(index >= 0 && index <_threads.size()){
		return terminateFlags[index];
	}
	
	return false;
}

void OBSThreadPool::setTerminate(size_t index, bool value)
{
	std::unique_lock<std::mutex> lock(_terminateMutex);
	
	if(index >= 0 && index <_threads.size()){
		terminateFlags[index] = value;
	}
}

在上述代码中,我们增加了 modify 方法来动态修改线程池中的线程数量,并且添加了 _terminateFlags 数组来记录每个线程的退出状态。同时,修改了 stop 方法以确保所有子线程安全退出,并设置 _bTerminate 标志为 true

请根据具体需求和代码架构进行适当的调整和完善。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=18581

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?