Mercurial > hg > orthanc
view Core/MultiThreading/BagOfRunnablesBySteps.cpp @ 1678:1a3c20cd1b53 db-changes
renames in BagOfRunnablesBySteps to make things clearer
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 06 Oct 2015 11:12:13 +0200 |
parents | f967bdf8534e |
children |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2015 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 "BagOfRunnablesBySteps.h" #include "../Logging.h" #include <stack> #include <boost/thread.hpp> namespace Orthanc { struct BagOfRunnablesBySteps::PImpl { boost::mutex mutex_; // 1. The set of active runnables, i.e. the runnables that have // not finished their job yet, plus the runnables that have not // been joined yet. typedef std::map<IRunnableBySteps*, boost::thread*> ActiveRunnables; ActiveRunnables activeRunnables_; bool continueActiveRunnables_; // 2. Condition variable that is notified when one active runnable // stops. boost::condition_variable oneRunnableHasStopped_; // 3. The list of runnables that have stopped but are waiting to be // joined by the collector. typedef std::stack<IRunnableBySteps*> StoppedRunnables; StoppedRunnables stoppedRunnables_; // 4. Condition variable that is notified when one stopped // runnable has been joined. boost::condition_variable oneRunnableIsJoined_; // The thread that joins the runnables after they stop bool continueCollector_; std::auto_ptr<boost::thread> collector_; }; void BagOfRunnablesBySteps::RunnableThread(BagOfRunnablesBySteps* bag, IRunnableBySteps* runnable) { while (bag->pimpl_->continueActiveRunnables_) { if (!runnable->Step()) { break; } } { // Register this runnable as having stopped boost::mutex::scoped_lock lock(bag->pimpl_->mutex_); bag->pimpl_->stoppedRunnables_.push(runnable); bag->pimpl_->oneRunnableHasStopped_.notify_one(); } } void BagOfRunnablesBySteps::CollectorThread(BagOfRunnablesBySteps* bag) { boost::mutex::scoped_lock lock(bag->pimpl_->mutex_); while (bag->pimpl_->continueCollector_) { while (!bag->pimpl_->stoppedRunnables_.empty()) { std::auto_ptr<IRunnableBySteps> r(bag->pimpl_->stoppedRunnables_.top()); bag->pimpl_->stoppedRunnables_.pop(); assert(r.get() != NULL); assert(bag->pimpl_->activeRunnables_.find(r.get()) != bag->pimpl_->activeRunnables_.end()); std::auto_ptr<boost::thread> t(bag->pimpl_->activeRunnables_[r.get()]); bag->pimpl_->activeRunnables_.erase(r.get()); assert(t.get() != NULL); assert(bag->pimpl_->activeRunnables_.find(r.get()) == bag->pimpl_->activeRunnables_.end()); if (t->joinable()) { t->join(); } bag->pimpl_->oneRunnableIsJoined_.notify_one(); } bag->pimpl_->oneRunnableHasStopped_.wait(lock); } } BagOfRunnablesBySteps::BagOfRunnablesBySteps() : pimpl_(new PImpl) { pimpl_->continueActiveRunnables_ = true; pimpl_->continueCollector_ = true; // Everyting is set up, the finish listener can be started pimpl_->collector_.reset(new boost::thread(CollectorThread, this)); } BagOfRunnablesBySteps::~BagOfRunnablesBySteps() { if (pimpl_->continueCollector_) { LOG(ERROR) << "INTERNAL ERROR: BagOfRunnablesBySteps::Finalize() should be invoked manually to avoid mess in the destruction order!"; Finalize(); } } void BagOfRunnablesBySteps::Add(IRunnableBySteps* runnable) { // Make sure the runnable is deleted is something goes wrong std::auto_ptr<IRunnableBySteps> runnableRabi(runnable); boost::mutex::scoped_lock lock(pimpl_->mutex_); boost::thread* t(new boost::thread(RunnableThread, this, runnable)); pimpl_->activeRunnables_.insert(std::make_pair(runnableRabi.release(), t)); } void BagOfRunnablesBySteps::StopAll() { boost::mutex::scoped_lock lock(pimpl_->mutex_); pimpl_->continueActiveRunnables_ = false; while (pimpl_->activeRunnables_.size() > 0) { pimpl_->oneRunnableIsJoined_.wait(lock); } pimpl_->continueActiveRunnables_ = true; } void BagOfRunnablesBySteps::Finalize() { if (pimpl_->continueCollector_) { StopAll(); // Stop the finish listener pimpl_->continueCollector_ = false; pimpl_->oneRunnableHasStopped_.notify_one(); // Awakens the listener if (pimpl_->collector_->joinable()) { pimpl_->collector_->join(); } } } }