Mercurial > hg > orthanc
annotate OrthancServer/Scheduler/ServerScheduler.h @ 1143:8bff277e6886
note
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 11 Sep 2014 10:26:10 +0200 |
parents | 13e230bbd882 |
children | 6e7e5ed91c2d |
rev | line source |
---|---|
781 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
3 * Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, | |
4 * 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 #pragma once | |
34 | |
35 #include "ServerJob.h" | |
36 | |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
781
diff
changeset
|
37 #include "../../Core/MultiThreading/Semaphore.h" |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
781
diff
changeset
|
38 |
781 | 39 namespace Orthanc |
40 { | |
1000
13e230bbd882
rename filter to command
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
999
diff
changeset
|
41 class ServerScheduler : public ServerCommandInstance::IListener |
781 | 42 { |
43 private: | |
44 struct JobInfo | |
45 { | |
46 bool watched_; | |
47 bool cancel_; | |
48 size_t size_; | |
49 size_t success_; | |
50 size_t failures_; | |
51 std::string description_; | |
52 }; | |
53 | |
54 enum JobStatus | |
55 { | |
56 JobStatus_Running = 1, | |
57 JobStatus_Success = 2, | |
58 JobStatus_Failure = 3 | |
59 }; | |
60 | |
1000
13e230bbd882
rename filter to command
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
999
diff
changeset
|
61 typedef IServerCommand::ListOfStrings ListOfStrings; |
781 | 62 typedef std::map<std::string, JobInfo> Jobs; |
63 | |
64 boost::mutex mutex_; | |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
781
diff
changeset
|
65 boost::condition_variable watchedJobFinished_; |
781 | 66 Jobs jobs_; |
67 SharedMessageQueue queue_; | |
68 bool finish_; | |
69 boost::thread worker_; | |
70 std::map<std::string, JobStatus> watchedJobStatus_; | |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
781
diff
changeset
|
71 Semaphore availableJob_; |
781 | 72 |
73 JobInfo& GetJobInfo(const std::string& jobId); | |
74 | |
75 virtual void SignalSuccess(const std::string& jobId); | |
76 | |
77 virtual void SignalFailure(const std::string& jobId); | |
78 | |
79 static void Worker(ServerScheduler* that); | |
80 | |
81 void SubmitInternal(ServerJob& job, | |
82 bool watched); | |
83 | |
84 public: | |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
781
diff
changeset
|
85 ServerScheduler(unsigned int maxjobs); |
781 | 86 |
87 ~ServerScheduler(); | |
88 | |
89 void Submit(ServerJob& job); | |
90 | |
91 bool SubmitAndWait(ListOfStrings& outputs, | |
92 ServerJob& job); | |
93 | |
999 | 94 bool SubmitAndWait(ServerJob& job); |
95 | |
781 | 96 bool IsRunning(const std::string& jobId); |
97 | |
98 void Cancel(const std::string& jobId); | |
99 | |
100 // Returns a number between 0 and 1 | |
101 float GetProgress(const std::string& jobId); | |
102 | |
103 bool IsRunning(const ServerJob& job) | |
104 { | |
105 return IsRunning(job.GetId()); | |
106 } | |
107 | |
108 void Cancel(const ServerJob& job) | |
109 { | |
110 Cancel(job.GetId()); | |
111 } | |
112 | |
113 float GetProgress(const ServerJob& job) | |
114 { | |
115 return GetProgress(job.GetId()); | |
116 } | |
117 | |
118 void GetListOfJobs(ListOfStrings& jobs); | |
119 }; | |
120 } |