Mercurial > hg > orthanc-client

comparison CppClient/ThreadedCommandProcessor.cpp @ 5:798076adf9e9

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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 }