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
|
|
37 namespace Orthanc
|
|
38 {
|
|
39 class ServerScheduler : public ServerFilterInstance::IListener
|
|
40 {
|
|
41 private:
|
|
42 struct JobInfo
|
|
43 {
|
|
44 bool watched_;
|
|
45 bool cancel_;
|
|
46 size_t size_;
|
|
47 size_t success_;
|
|
48 size_t failures_;
|
|
49 std::string description_;
|
|
50 };
|
|
51
|
|
52 enum JobStatus
|
|
53 {
|
|
54 JobStatus_Running = 1,
|
|
55 JobStatus_Success = 2,
|
|
56 JobStatus_Failure = 3
|
|
57 };
|
|
58
|
|
59 typedef IServerFilter::ListOfStrings ListOfStrings;
|
|
60 typedef std::map<std::string, JobInfo> Jobs;
|
|
61
|
|
62 boost::mutex mutex_;
|
|
63 boost::condition_variable jobFinished_;
|
|
64 Jobs jobs_;
|
|
65 SharedMessageQueue queue_;
|
|
66 bool finish_;
|
|
67 boost::thread worker_;
|
|
68 std::map<std::string, JobStatus> watchedJobStatus_;
|
|
69
|
|
70 JobInfo& GetJobInfo(const std::string& jobId);
|
|
71
|
|
72 virtual void SignalSuccess(const std::string& jobId);
|
|
73
|
|
74 virtual void SignalFailure(const std::string& jobId);
|
|
75
|
|
76 static void Worker(ServerScheduler* that);
|
|
77
|
|
78 void SubmitInternal(ServerJob& job,
|
|
79 bool watched);
|
|
80
|
|
81 public:
|
|
82 ServerScheduler();
|
|
83
|
|
84 ~ServerScheduler();
|
|
85
|
|
86 void Submit(ServerJob& job);
|
|
87
|
|
88 bool SubmitAndWait(ListOfStrings& outputs,
|
|
89 ServerJob& job);
|
|
90
|
|
91 bool IsRunning(const std::string& jobId);
|
|
92
|
|
93 void Cancel(const std::string& jobId);
|
|
94
|
|
95 // Returns a number between 0 and 1
|
|
96 float GetProgress(const std::string& jobId);
|
|
97
|
|
98 bool IsRunning(const ServerJob& job)
|
|
99 {
|
|
100 return IsRunning(job.GetId());
|
|
101 }
|
|
102
|
|
103 void Cancel(const ServerJob& job)
|
|
104 {
|
|
105 Cancel(job.GetId());
|
|
106 }
|
|
107
|
|
108 float GetProgress(const ServerJob& job)
|
|
109 {
|
|
110 return GetProgress(job.GetId());
|
|
111 }
|
|
112
|
|
113 void GetListOfJobs(ListOfStrings& jobs);
|
|
114 };
|
|
115 }
|