1
|
1 /**
|
|
2 * Orthanc - A Lightweight, RESTful DICOM Store
|
12
|
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
|
1
|
4 * Department, University Hospital of Liege, Belgium
|
|
5 *
|
|
6 * This program is free software: you can redistribute it and/or
|
|
7 * modify it under the terms of the GNU General Public License as
|
|
8 * published by the Free Software Foundation, either version 3 of the
|
|
9 * License, or (at your option) any later version.
|
|
10 *
|
|
11 * In addition, as a special exception, the copyright holders of this
|
|
12 * program give permission to link the code of its release with the
|
|
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it
|
|
14 * that use the same license as the "OpenSSL" library), and distribute
|
|
15 * the linked executables. You must obey the GNU General Public License
|
|
16 * in all respects for all of the code used other than "OpenSSL". If you
|
|
17 * modify file(s) with this exception, you may extend this exception to
|
|
18 * your version of the file(s), but you are not obligated to do so. If
|
|
19 * you do not wish to do so, delete this exception statement from your
|
|
20 * version. If you delete this exception statement from all source files
|
|
21 * in the program, then also delete it here.
|
|
22 *
|
|
23 * This program is distributed in the hope that it will be useful, but
|
|
24 * WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
26 * General Public License for more details.
|
|
27 *
|
|
28 * You should have received a copy of the GNU General Public License
|
|
29 * along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
30 **/
|
|
31
|
|
32
|
|
33 #include "ThreadedCommandProcessor.h"
|
|
34
|
3
|
35 #include "../Orthanc/Core/OrthancException.h"
|
1
|
36
|
6
|
37 namespace OrthancClient
|
1
|
38 {
|
|
39 static const int32_t TIMEOUT = 10;
|
|
40
|
|
41
|
|
42 void ThreadedCommandProcessor::Processor(ThreadedCommandProcessor* that)
|
|
43 {
|
|
44 while (!that->done_)
|
|
45 {
|
6
|
46 std::auto_ptr<Orthanc::IDynamicObject> command(that->queue_.Dequeue(TIMEOUT));
|
1
|
47
|
|
48 if (command.get() != NULL)
|
|
49 {
|
|
50 bool success = false;
|
|
51
|
|
52 try
|
|
53 {
|
|
54 if (that->success_)
|
|
55 {
|
|
56 // No command has failed so far
|
|
57
|
|
58 if (that->cancel_)
|
|
59 {
|
|
60 // The commands have been canceled. Skip the execution
|
|
61 // of this command, yet mark it as succeeded.
|
|
62 success = true;
|
|
63 }
|
|
64 else
|
|
65 {
|
6
|
66 success = dynamic_cast<Orthanc::ICommand&>(*command).Execute();
|
1
|
67 }
|
|
68 }
|
|
69 else
|
|
70 {
|
|
71 // A command has already failed. Skip the execution of this command.
|
|
72 }
|
|
73 }
|
6
|
74 catch (Orthanc::OrthancException&)
|
1
|
75 {
|
|
76 }
|
|
77
|
|
78 {
|
|
79 boost::mutex::scoped_lock lock(that->mutex_);
|
|
80 assert(that->remainingCommands_ > 0);
|
|
81 that->remainingCommands_--;
|
|
82
|
|
83 if (!success)
|
|
84 {
|
|
85 if (!that->cancel_ && that->listener_ && that->success_)
|
|
86 {
|
|
87 // This is the first command that fails
|
|
88 that->listener_->SignalFailure();
|
|
89 }
|
|
90
|
|
91 that->success_ = false;
|
|
92 }
|
|
93 else
|
|
94 {
|
|
95 if (!that->cancel_ && that->listener_)
|
|
96 {
|
|
97 if (that->remainingCommands_ == 0)
|
|
98 {
|
|
99 that->listener_->SignalSuccess(that->totalCommands_);
|
|
100 }
|
|
101 else
|
|
102 {
|
|
103 that->listener_->SignalProgress(that->totalCommands_ - that->remainingCommands_,
|
|
104 that->totalCommands_);
|
|
105 }
|
|
106 }
|
|
107 }
|
|
108
|
|
109 that->processedCommand_.notify_all();
|
|
110 }
|
|
111 }
|
|
112 }
|
|
113 }
|
|
114
|
|
115
|
|
116 ThreadedCommandProcessor::ThreadedCommandProcessor(unsigned int numThreads)
|
|
117 {
|
|
118 if (numThreads < 1)
|
|
119 {
|
6
|
120 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
|
1
|
121 }
|
|
122
|
|
123 listener_ = NULL;
|
|
124 success_ = true;
|
|
125 done_ = false;
|
|
126 cancel_ = false;
|
|
127 threads_.resize(numThreads);
|
|
128 remainingCommands_ = 0;
|
|
129 totalCommands_ = 0;
|
|
130
|
|
131 for (unsigned int i = 0; i < numThreads; i++)
|
|
132 {
|
|
133 threads_[i] = new boost::thread(Processor, this);
|
|
134 }
|
|
135 }
|
|
136
|
|
137
|
|
138 ThreadedCommandProcessor::~ThreadedCommandProcessor()
|
|
139 {
|
|
140 done_ = true;
|
|
141
|
|
142 for (unsigned int i = 0; i < threads_.size(); i++)
|
|
143 {
|
|
144 boost::thread* t = threads_[i];
|
|
145
|
|
146 if (t != NULL)
|
|
147 {
|
|
148 if (t->joinable())
|
|
149 {
|
|
150 t->join();
|
|
151 }
|
|
152
|
|
153 delete t;
|
|
154 }
|
|
155 }
|
|
156 }
|
|
157
|
|
158
|
6
|
159 void ThreadedCommandProcessor::Post(Orthanc::ICommand* command)
|
1
|
160 {
|
|
161 if (command == NULL)
|
|
162 {
|
6
|
163 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
|
1
|
164 }
|
|
165
|
|
166 boost::mutex::scoped_lock lock(mutex_);
|
|
167 queue_.Enqueue(command);
|
|
168 remainingCommands_++;
|
|
169 totalCommands_++;
|
|
170 }
|
|
171
|
|
172
|
|
173 bool ThreadedCommandProcessor::Join()
|
|
174 {
|
|
175 boost::mutex::scoped_lock lock(mutex_);
|
|
176
|
|
177 while (remainingCommands_ != 0)
|
|
178 {
|
|
179 processedCommand_.wait(lock);
|
|
180 }
|
|
181
|
|
182 if (cancel_ && listener_)
|
|
183 {
|
|
184 listener_->SignalCancel();
|
|
185 }
|
|
186
|
|
187 // Reset the sequence counters for subsequent commands
|
|
188 bool hasSucceeded = success_;
|
|
189 success_ = true;
|
|
190 totalCommands_ = 0;
|
|
191 cancel_ = false;
|
|
192
|
|
193 return hasSucceeded;
|
|
194 }
|
|
195
|
|
196
|
|
197 void ThreadedCommandProcessor::Cancel()
|
|
198 {
|
|
199 boost::mutex::scoped_lock lock(mutex_);
|
|
200
|
|
201 cancel_ = true;
|
|
202 }
|
|
203
|
|
204
|
|
205 void ThreadedCommandProcessor::SetListener(IListener& listener)
|
|
206 {
|
|
207 boost::mutex::scoped_lock lock(mutex_);
|
|
208 listener_ = &listener;
|
|
209 }
|
|
210 }
|