6 Star 3 Fork 32

OpenHarmony/developtools_global_resource_tool

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
thread_pool.cpp 2.80 KB
一键复制 编辑 原始数据 按行查看 历史
/*
* Copyright (c) 2021-2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* 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 "thread_pool.h"
#include "restool_errors.h"
#include <iostream>
#include <string>
namespace OHOS {
namespace Global {
namespace Restool {
using namespace std;
ThreadPool::ThreadPool()
{}
ThreadPool &ThreadPool::GetInstance()
{
static ThreadPool pool;
return pool;
}
uint32_t ThreadPool::Start(const size_t &threadCount)
{
if (!workerThreads_.empty()) {
cout << "Warning: ThreadPool is already started." << endl;
return RESTOOL_SUCCESS;
}
size_t hardwareCount = std::thread::hardware_concurrency();
cout << "Info: hardware concurrency count is : " << hardwareCount << endl;
size_t count = threadCount <= 0 ? (hardwareCount <= 0 ? DEFAULT_POOL_SIZE : hardwareCount) : threadCount;
if (count == 1) {
count++;
}
cout << "Info: thread count is : " << count << endl;
running_ = true;
workerThreads_.reserve(count);
for (size_t i = 0; i < count; ++i) {
workerThreads_.emplace_back([this] { this->WorkInThread(); });
}
cout << "Info: thread pool is started" << endl;
return RESTOOL_SUCCESS;
}
void ThreadPool::Stop()
{
{
std::unique_lock<std::mutex> lock(queueMutex_);
running_ = false;
}
condition_.notify_all();
for (std::thread &worker : workerThreads_) {
if (worker.joinable()) {
worker.join();
}
}
cout << "Info: thread pool is stopped" << endl;
}
ThreadPool::~ThreadPool()
{
if (running_) {
Stop();
}
}
void ThreadPool::WorkInThread()
{
while (this->running_) {
std::function<void()> task;
{
std::unique_lock<std::mutex> lock(this->queueMutex_);
// wake up when there's a task or when the pool is stopped
this->condition_.wait(lock, [this] { return !this->running_ || !this->tasks_.empty(); });
if (!this->running_) {
// exit thread when the pool is stopped
return;
}
if (!this->tasks_.empty()) {
task = std::move(this->tasks_.front());
this->tasks_.pop();
}
}
if (task) {
task();
}
}
}
} // namespace Restool
} // namespace Global
} // namespace OHOS
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C/C++
1
https://gitee.com/openharmony/developtools_global_resource_tool.git
git@gitee.com:openharmony/developtools_global_resource_tool.git
openharmony
developtools_global_resource_tool
developtools_global_resource_tool
master

搜索帮助