Mercurial > hg > orthanc
view Core/MultiThreading/BagOfTasksProcessor.cpp @ 2099:bcbc9137a535
refactoring
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 07 Oct 2016 11:13:41 +0200 |
parents | 0ae26237569a |
children | 5b818f677dd6 |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * * This program is free software: you can redistribute it and/or * modify it under the terms of the GNU General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * In addition, as a special exception, the copyright holders of this * program give permission to link the code of its release with the * OpenSSL project's "OpenSSL" library (or with modified versions of it * that use the same license as the "OpenSSL" library), and distribute * the linked executables. You must obey the GNU General Public License * in all respects for all of the code used other than "OpenSSL". If you * modify file(s) with this exception, you may extend this exception to * your version of the file(s), but you are not obligated to do so. If * you do not wish to do so, delete this exception statement from your * version. If you delete this exception statement from all source files * in the program, then also delete it here. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. **/ #include "../PrecompiledHeaders.h" #include "BagOfTasksProcessor.h" #include "../OrthancException.h" namespace Orthanc { class BagOfTasksProcessor::Task : public IDynamicObject { private: uint64_t bag_; std::auto_ptr<ICommand> command_; public: Task(uint64_t bag, ICommand* command) : bag_(bag), command_(command) { } bool Execute() { try { return command_->Execute(); } catch (OrthancException&) { return false; } catch (std::runtime_error&) { return false; } } uint64_t GetBag() { return bag_; } }; void BagOfTasksProcessor::Worker(BagOfTasksProcessor* that) { while (that->continue_) { std::auto_ptr<IDynamicObject> obj(that->queue_.Dequeue(100)); if (obj.get() != NULL) { Task& task = *dynamic_cast<Task*>(obj.get()); { boost::mutex::scoped_lock lock(that->mutex_); Bags::iterator bag = that->bags_.find(task.GetBag()); assert(bag != that->bags_.end()); assert(bag->second.done_ < bag->second.size_); if (bag->second.status_ != BagStatus_Running) { // This bag of task has failed or is tagged as canceled, do nothing bag->second.done_ += 1; continue; } } bool success = task.Execute(); { boost::mutex::scoped_lock lock(that->mutex_); Bags::iterator bag = that->bags_.find(task.GetBag()); assert(bag != that->bags_.end()); if (!success) { bag->second.status_ = BagStatus_Failed; } assert(bag->second.done_ < bag->second.size_); bag->second.done_ += 1; if (bag->second.done_ == bag->second.size_) { that->exitStatus_[task.GetBag()] = (bag->second.status_ == BagStatus_Running); that->bagFinished_.notify_all(); } } } } } void BagOfTasksProcessor::Cancel(int64_t bag) { boost::mutex::scoped_lock lock(mutex_); Bags::iterator it = bags_.find(bag); if (it != bags_.end()) { it->second.status_ = BagStatus_Canceled; } } bool BagOfTasksProcessor::Join(int64_t bag) { boost::mutex::scoped_lock lock(mutex_); while (continue_) { ExitStatus::iterator it = exitStatus_.find(bag); if (it == exitStatus_.end()) // The bag is still running { bagFinished_.wait(lock); } else { bool status = it->second; exitStatus_.erase(it); return status; } } return false; // The processor is stopping } float BagOfTasksProcessor::GetProgress(int64_t bag) { boost::mutex::scoped_lock lock(mutex_); Bags::const_iterator it = bags_.find(bag); if (it == bags_.end()) { // The bag of tasks has finished return 1.0f; } else { return (static_cast<float>(it->second.done_) / static_cast<float>(it->second.size_)); } } bool BagOfTasksProcessor::Handle::Join() { if (hasJoined_) { return status_; } else { status_ = that_.Join(bag_); hasJoined_ = true; return status_; } } BagOfTasksProcessor::BagOfTasksProcessor(size_t countThreads) : countBags_(0), continue_(true) { if (countThreads == 0) { throw OrthancException(ErrorCode_ParameterOutOfRange); } threads_.resize(countThreads); for (size_t i = 0; i < threads_.size(); i++) { threads_[i] = new boost::thread(Worker, this); } } BagOfTasksProcessor::~BagOfTasksProcessor() { continue_ = false; bagFinished_.notify_all(); // Wakes up all the pending "Join()" for (size_t i = 0; i < threads_.size(); i++) { if (threads_[i]) { if (threads_[i]->joinable()) { threads_[i]->join(); } delete threads_[i]; threads_[i] = NULL; } } } BagOfTasksProcessor::Handle* BagOfTasksProcessor::Submit(BagOfTasks& tasks) { if (tasks.GetSize() == 0) { return new Handle(*this, 0, true); } boost::mutex::scoped_lock lock(mutex_); uint64_t id = countBags_; countBags_ += 1; Bag bag(tasks.GetSize()); bags_[id] = bag; while (!tasks.IsEmpty()) { queue_.Enqueue(new Task(id, tasks.Pop())); } return new Handle(*this, id, false); } }