Mercurial > hg > orthanc

comparison OrthancFramework/Sources/MultiThreading/RunnableWorkersPool.cpp @ 4044:d25f4c0fa160 framework

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
splitting code into OrthancFramework and OrthancServer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Jun 2020 20:30:34 +0200
parents Core/MultiThreading/RunnableWorkersPool.cpp@2a170a8f1faf
children bf7b9edf6b81
comparison
equal deleted inserted replaced
4043:6c6239aec462 4044:d25f4c0fa160
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2020 Osimis S.A., Belgium
6 *
7 * This program is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License as
9 * published by the Free Software Foundation, either version 3 of the
10 * License, or (at your option) any later version.
11 *
12 * In addition, as a special exception, the copyright holders of this
13 * program give permission to link the code of its release with the
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
15 * that use the same license as the "OpenSSL" library), and distribute
16 * the linked executables. You must obey the GNU General Public License
17 * in all respects for all of the code used other than "OpenSSL". If you
18 * modify file(s) with this exception, you may extend this exception to
19 * your version of the file(s), but you are not obligated to do so. If
20 * you do not wish to do so, delete this exception statement from your
21 * version. If you delete this exception statement from all source files
22 * in the program, then also delete it here.
23 *
24 * This program is distributed in the hope that it will be useful, but
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
27 * General Public License for more details.
28 *
29 * You should have received a copy of the GNU General Public License
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
31 **/
32
33
34 #include "../PrecompiledHeaders.h"
35 #include "RunnableWorkersPool.h"
36
37 #include "SharedMessageQueue.h"
38 #include "../Compatibility.h"
39 #include "../OrthancException.h"
40 #include "../Logging.h"
41
42 namespace Orthanc
43 {
44 struct RunnableWorkersPool::PImpl
45 {
46 class Worker
47 {
48 private:
49 const bool& continue_;
50 SharedMessageQueue& queue_;
51 boost::thread thread_;
52
53 static void WorkerThread(Worker* that)
54 {
55 while (that->continue_)
56 {
57 try
58 {
59 std::unique_ptr<IDynamicObject> obj(that->queue_.Dequeue(100));
60 if (obj.get() != NULL)
61 {
62 IRunnableBySteps& runnable = *dynamic_cast<IRunnableBySteps*>(obj.get());
63
64 bool wishToContinue = runnable.Step();
65
66 if (wishToContinue)
67 {
68 // The runnable wishes to continue, reinsert it at the beginning of the queue
69 that->queue_.Enqueue(obj.release());
70 }
71 }
72 }
73 catch (OrthancException& e)
74 {
75 LOG(ERROR) << "Exception while handling some runnable object: " << e.What();
76 }
77 catch (std::bad_alloc&)
78 {
79 LOG(ERROR) << "Not enough memory to handle some runnable object";
80 }
81 catch (std::exception& e)
82 {
83 LOG(ERROR) << "std::exception while handling some runnable object: " << e.what();
84 }
85 catch (...)
86 {
87 LOG(ERROR) << "Native exception while handling some runnable object";
88 }
89 }
90 }
91
92 public:
93 Worker(const bool& globalContinue,
94 SharedMessageQueue& queue) :
95 continue_(globalContinue),
96 queue_(queue)
97 {
98 thread_ = boost::thread(WorkerThread, this);
99 }
100
101 void Join()
102 {
103 if (thread_.joinable())
104 {
105 thread_.join();
106 }
107 }
108 };
109
110
111 bool continue_;
112 std::vector<Worker*> workers_;
113 SharedMessageQueue queue_;
114 };
115
116
117
118 RunnableWorkersPool::RunnableWorkersPool(size_t countWorkers) : pimpl_(new PImpl)
119 {
120 pimpl_->continue_ = true;
121
122 if (countWorkers == 0)
123 {
124 throw OrthancException(ErrorCode_ParameterOutOfRange);
125 }
126
127 pimpl_->workers_.resize(countWorkers);
128
129 for (size_t i = 0; i < countWorkers; i++)
130 {
131 pimpl_->workers_[i] = new PImpl::Worker(pimpl_->continue_, pimpl_->queue_);
132 }
133 }
134
135
136 void RunnableWorkersPool::Stop()
137 {
138 if (pimpl_->continue_)
139 {
140 pimpl_->continue_ = false;
141
142 for (size_t i = 0; i < pimpl_->workers_.size(); i++)
143 {
144 PImpl::Worker* worker = pimpl_->workers_[i];
145
146 if (worker != NULL)
147 {
148 worker->Join();
149 delete worker;
150 }
151 }
152 }
153 }
154
155
156 RunnableWorkersPool::~RunnableWorkersPool()
157 {
158 Stop();
159 }
160
161
162 void RunnableWorkersPool::Add(IRunnableBySteps* runnable)
163 {
164 if (!pimpl_->continue_)
165 {
166 throw OrthancException(ErrorCode_BadSequenceOfCalls);
167 }
168
169 pimpl_->queue_.Enqueue(runnable);
170 }
171 }