Mercurial > hg > orthanc
annotate OrthancServer/Scheduler/ServerScheduler.h @ 1294:910478b2d4e4
refactoring
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Fri, 06 Feb 2015 13:48:05 +0100 |
| parents | 6e7e5ed91c2d |
| children | c0bdc47165ef |
| rev | line source |
|---|---|
| 781 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
|
1288
6e7e5ed91c2d
upgrade to year 2015
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1000
diff
changeset
|
3 * Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics |
|
6e7e5ed91c2d
upgrade to year 2015
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1000
diff
changeset
|
4 * Department, University Hospital of Liege, Belgium |
| 781 | 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 } |