Mercurial > hg > orthanc-client
comparison CppClient/ThreadedCommandProcessor.cpp @ 5:798076adf9e9
rename
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 02 Jun 2015 10:11:34 +0200 |
parents | OrthancCppClient/ThreadedCommandProcessor.cpp@d5027f9f676a |
children | c584c25a74fd |
comparison
equal
deleted
inserted
replaced
4:2e999f3e84b4 | 5:798076adf9e9 |
---|---|
1 /** | |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
3 * Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics | |
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 | |
35 #include "../Orthanc/Core/OrthancException.h" | |
36 | |
37 namespace Orthanc | |
38 { | |
39 static const int32_t TIMEOUT = 10; | |
40 | |
41 | |
42 void ThreadedCommandProcessor::Processor(ThreadedCommandProcessor* that) | |
43 { | |
44 while (!that->done_) | |
45 { | |
46 std::auto_ptr<IDynamicObject> command(that->queue_.Dequeue(TIMEOUT)); | |
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 { | |
66 success = dynamic_cast<ICommand&>(*command).Execute(); | |
67 } | |
68 } | |
69 else | |
70 { | |
71 // A command has already failed. Skip the execution of this command. | |
72 } | |
73 } | |
74 catch (OrthancException) | |
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 { | |
120 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
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 | |
159 void ThreadedCommandProcessor::Post(ICommand* command) | |
160 { | |
161 if (command == NULL) | |
162 { | |
163 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
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 } |