orthanc-stone: Framework/Loaders/OracleScheduler.cpp annotate

annotate Framework/Loaders/OracleScheduler.cpp @ 1271:0ca50d275b9a broker

upgrade to year 2020

author	Sebastien Jodogne <s.jodogne@gmail.com>
date	Fri, 31 Jan 2020 17:34:57 +0100
parents	c471a0aa137b
children	c38c89684d83

rev	line source
1228 c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	1 /**
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	2 * Stone of Orthanc
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	4 * Department, University Hospital of Liege, Belgium
1271 0ca50d275b9a upgrade to year 2020 Sebastien Jodogne <s.jodogne@gmail.com> parents: 1228 diff changeset	5 * Copyright (C) 2017-2020 Osimis S.A., Belgium
1228 c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	6 *
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	7 * This program is free software: you can redistribute it and/or
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	8 * modify it under the terms of the GNU Affero General Public License
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	9 * as published by the Free Software Foundation, either version 3 of
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	10 * the License, or (at your option) any later version.
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	11 *
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	12 * This program is distributed in the hope that it will be useful, but
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	13 * WITHOUT ANY WARRANTY; without even the implied warranty of
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	15 * Affero General Public License for more details.
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	16 *
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	17 * You should have received a copy of the GNU Affero General Public License
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	18 * along with this program. If not, see <http://www.gnu.org/licenses/>.
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	19 **/
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	20
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	21
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	22 #include "OracleScheduler.h"
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	23
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	24 #include "../Oracle/ParseDicomFromFileCommand.h"
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	25
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	26 namespace OrthancStone
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	27 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	28 class OracleScheduler::ReceiverPayload : public Orthanc::IDynamicObject
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	29 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	30 private:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	31 Priority priority_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	32 boost::weak_ptr<IObserver> receiver_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	33 std::auto_ptr<IOracleCommand> command_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	34
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	35 public:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	36 ReceiverPayload(Priority priority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	37 boost::weak_ptr<IObserver> receiver,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	38 IOracleCommand* command) :
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	39 priority_(priority),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	40 receiver_(receiver),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	41 command_(command)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	42 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	43 if (command == NULL)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	44 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	45 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	46 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	47 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	48
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	49 Priority GetActivePriority() const
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	50 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	51 return priority_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	52 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	53
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	54 boost::weak_ptr<IObserver> GetOriginalReceiver() const
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	55 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	56 return receiver_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	57 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	58
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	59 const IOracleCommand& GetOriginalCommand() const
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	60 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	61 assert(command_.get() != NULL);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	62 return *command_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	63 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	64 };
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	65
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	66
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	67 class OracleScheduler::ScheduledCommand : public boost::noncopyable
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	68 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	69 private:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	70 boost::weak_ptr<IObserver> receiver_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	71 std::auto_ptr<IOracleCommand> command_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	72
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	73 public:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	74 ScheduledCommand(boost::shared_ptr<IObserver> receiver,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	75 IOracleCommand* command) :
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	76 receiver_(receiver),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	77 command_(command)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	78 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	79 if (command == NULL)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	80 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	81 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	82 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	83 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	84
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	85 boost::weak_ptr<IObserver> GetReceiver()
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	86 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	87 return receiver_;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	88 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	89
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	90 bool IsSameReceiver(boost::shared_ptr<OrthancStone::IObserver> receiver) const
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	91 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	92 boost::shared_ptr<IObserver> lock(receiver_.lock());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	93
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	94 return (lock &&
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	95 lock.get() == receiver.get());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	96 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	97
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	98 IOracleCommand* WrapCommand(Priority priority)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	99 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	100 if (command_.get() == NULL)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	101 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	102 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	103 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	104 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	105 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	106 std::auto_ptr<IOracleCommand> wrapped(command_->Clone());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	107 dynamic_cast<OracleCommandBase&>(*wrapped).AcquirePayload(new ReceiverPayload(priority, receiver_, command_.release()));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	108 return wrapped.release();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	109 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	110 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	111 };
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	112
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	113
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	114
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	115 void OracleScheduler::ClearQueue(Queue& queue)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	116 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	117 for (Queue::iterator it = queue.begin(); it != queue.end(); ++it)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	118 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	119 assert(it->second != NULL);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	120 delete it->second;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	121
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	122 totalProcessed_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	123 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	124
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	125 queue.clear();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	126 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	127
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	128
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	129 void OracleScheduler::RemoveReceiverFromQueue(Queue& queue,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	130 boost::shared_ptr<IObserver> receiver)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	131 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	132 if (!receiver)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	133 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	134 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	135 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	136
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	137 Queue tmp;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	138
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	139 for (Queue::iterator it = queue.begin(); it != queue.end(); ++it)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	140 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	141 assert(it->second != NULL);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	142
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	143 if (!(it->second->IsSameReceiver(receiver)))
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	144 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	145 // This promise is still active
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	146 tmp.insert(std::make_pair(it->first, it->second));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	147 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	148 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	149 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	150 delete it->second;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	151
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	152 totalProcessed_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	153 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	154 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	155
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	156 queue = tmp;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	157 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	158
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	159
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	160 void OracleScheduler::CheckInvariants() const
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	161 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	162 #ifndef NDEBUG
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	163 /*char buf[1024];
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	164 sprintf(buf, "active: %d %d %d ; pending: %lu %lu %lu",
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	165 activeHighPriorityCommands_, activeStandardPriorityCommands_, activeLowPriorityCommands_,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	166 highPriorityQueue_.size(), standardPriorityQueue_.size(), lowPriorityQueue_.size());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	167 LOG(INFO) << buf;*/
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	168
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	169 assert(activeHighPriorityCommands_ <= maxHighPriorityCommands_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	170 assert(activeStandardPriorityCommands_ <= maxStandardPriorityCommands_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	171 assert(activeLowPriorityCommands_ <= maxLowPriorityCommands_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	172 assert(totalProcessed_ <= totalScheduled_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	173
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	174 for (Queue::const_iterator it = standardPriorityQueue_.begin(); it != standardPriorityQueue_.end(); ++it)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	175 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	176 assert(it->first > PRIORITY_HIGH &&
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	177 it->first < PRIORITY_LOW);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	178 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	179
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	180 for (Queue::const_iterator it = highPriorityQueue_.begin(); it != highPriorityQueue_.end(); ++it)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	181 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	182 assert(it->first <= PRIORITY_HIGH);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	183 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	184
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	185 for (Queue::const_iterator it = lowPriorityQueue_.begin(); it != lowPriorityQueue_.end(); ++it)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	186 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	187 assert(it->first >= PRIORITY_LOW);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	188 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	189 #endif
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	190 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	191
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	192
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	193 void OracleScheduler::SpawnFromQueue(Queue& queue,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	194 Priority priority)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	195 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	196 CheckInvariants();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	197
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	198 Queue::iterator item = queue.begin();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	199 assert(item != queue.end());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	200
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	201 std::auto_ptr<ScheduledCommand> command(dynamic_cast<ScheduledCommand*>(item->second));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	202 queue.erase(item);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	203
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	204 if (command.get() != NULL)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	205 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	206 /**
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	207 * Only schedule the command for execution in the oracle, if its
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	208 * receiver has not been destroyed yet.
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	209 **/
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	210 boost::shared_ptr<IObserver> observer(command->GetReceiver().lock());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	211 if (observer)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	212 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	213 if (oracle_.Schedule(GetSharedObserver(), command->WrapCommand(priority)))
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	214 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	215 /**
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	216 * Executing this code if "Schedule()" returned "false"
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	217 * above, will result in a memory leak within
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	218 * "OracleScheduler", as the scheduler believes that some
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	219 * command is still active (i.e. pending to be executed by
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	220 * the oracle), hereby stalling the scheduler during its
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	221 * destruction, and not freeing the
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	222 * "shared_ptr<OracleScheduler>" of the Stone context (check
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	223 * out "sjo-playground/WebViewer/Backend/Leak")
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	224 **/
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	225
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	226 switch (priority)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	227 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	228 case Priority_High:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	229 activeHighPriorityCommands_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	230 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	231
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	232 case Priority_Standard:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	233 activeStandardPriorityCommands_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	234 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	235
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	236 case Priority_Low:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	237 activeLowPriorityCommands_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	238 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	239
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	240 default:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	241 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	242 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	243 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	244 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	245 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	246 totalProcessed_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	247 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	248 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	249 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	250 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	251 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	252 LOG(ERROR) << "NULL command, should never happen";
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	253 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	254
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	255 CheckInvariants();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	256 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	257
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	258
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	259 void OracleScheduler::SpawnCommands()
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	260 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	261 // Send as many commands as possible to the oracle
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	262 while (!highPriorityQueue_.empty())
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	263 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	264 if (activeHighPriorityCommands_ < maxHighPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	265 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	266 // First fill the high-priority lane
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	267 SpawnFromQueue(highPriorityQueue_, Priority_High);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	268 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	269 else if (activeStandardPriorityCommands_ < maxStandardPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	270 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	271 // There remain too many high-priority commands for the
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	272 // high-priority lane, schedule them to the standard-priority lanes
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	273 SpawnFromQueue(highPriorityQueue_, Priority_Standard);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	274 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	275 else if (activeLowPriorityCommands_ < maxLowPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	276 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	277 SpawnFromQueue(highPriorityQueue_, Priority_Low);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	278 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	279 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	280 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	281 return; // No slot available
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	282 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	283 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	284
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	285 while (!standardPriorityQueue_.empty())
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	286 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	287 if (activeStandardPriorityCommands_ < maxStandardPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	288 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	289 SpawnFromQueue(standardPriorityQueue_, Priority_Standard);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	290 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	291 else if (activeLowPriorityCommands_ < maxLowPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	292 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	293 SpawnFromQueue(standardPriorityQueue_, Priority_Low);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	294 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	295 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	296 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	297 return;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	298 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	299 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	300
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	301 while (!lowPriorityQueue_.empty())
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	302 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	303 if (activeLowPriorityCommands_ < maxLowPriorityCommands_)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	304 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	305 SpawnFromQueue(lowPriorityQueue_, Priority_Low);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	306 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	307 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	308 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	309 return;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	310 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	311 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	312 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	313
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	314
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	315 void OracleScheduler::RemoveActiveCommand(const ReceiverPayload& payload)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	316 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	317 CheckInvariants();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	318
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	319 totalProcessed_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	320
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	321 switch (payload.GetActivePriority())
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	322 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	323 case Priority_High:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	324 assert(activeHighPriorityCommands_ > 0);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	325 activeHighPriorityCommands_ --;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	326 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	327
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	328 case Priority_Standard:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	329 assert(activeStandardPriorityCommands_ > 0);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	330 activeStandardPriorityCommands_ --;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	331 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	332
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	333 case Priority_Low:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	334 assert(activeLowPriorityCommands_ > 0);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	335 activeLowPriorityCommands_ --;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	336 break;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	337
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	338 default:
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	339 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	340 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	341
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	342 SpawnCommands();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	343
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	344 CheckInvariants();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	345 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	346
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	347
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	348 void OracleScheduler::Handle(const GetOrthancImageCommand::SuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	349 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	350 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	351 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	352
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	353 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	354
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	355 GetOrthancImageCommand::SuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	356 dynamic_cast<const GetOrthancImageCommand&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	357 message.GetImage(), message.GetMimeType());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	358 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	359 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	360
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	361
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	362 void OracleScheduler::Handle(const GetOrthancWebViewerJpegCommand::SuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	363 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	364 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	365 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	366
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	367 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	368
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	369 GetOrthancWebViewerJpegCommand::SuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	370 dynamic_cast<const GetOrthancWebViewerJpegCommand&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	371 message.GetImage());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	372 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	373 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	374
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	375
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	376 void OracleScheduler::Handle(const HttpCommand::SuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	377 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	378 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	379 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	380
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	381 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	382
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	383 HttpCommand::SuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	384 dynamic_cast<const HttpCommand&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	385 message.GetAnswerHeaders(), message.GetAnswer());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	386 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	387 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	388
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	389
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	390 void OracleScheduler::Handle(const OrthancRestApiCommand::SuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	391 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	392 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	393 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	394
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	395 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	396
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	397 OrthancRestApiCommand::SuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	398 dynamic_cast<const OrthancRestApiCommand&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	399 message.GetAnswerHeaders(), message.GetAnswer());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	400 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	401 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	402
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	403
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	404 #if ORTHANC_ENABLE_DCMTK == 1
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	405 void OracleScheduler::Handle(const ParseDicomSuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	406 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	407 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	408 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	409
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	410 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	411
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	412 ParseDicomSuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	413 dynamic_cast<const OracleCommandBase&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	414 message.GetDicom(), message.GetFileSize(), message.HasPixelData());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	415 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	416 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	417 #endif
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	418
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	419
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	420 void OracleScheduler::Handle(const ReadFileCommand::SuccessMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	421 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	422 assert(message.GetOrigin().HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	423 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(message.GetOrigin().GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	424
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	425 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	426
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	427 ReadFileCommand::SuccessMessage bis(
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	428 dynamic_cast<const ReadFileCommand&>(payload.GetOriginalCommand()),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	429 message.GetContent());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	430 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	431 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	432
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	433
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	434 void OracleScheduler::Handle(const OracleCommandExceptionMessage& message)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	435 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	436 const OracleCommandBase& command = dynamic_cast<const OracleCommandBase&>(message.GetOrigin());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	437
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	438 assert(command.HasPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	439 const ReceiverPayload& payload = dynamic_cast<const ReceiverPayload&>(command.GetPayload());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	440
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	441 RemoveActiveCommand(payload);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	442
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	443 OracleCommandExceptionMessage bis(payload.GetOriginalCommand(), message.GetException());
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	444 emitter_.EmitMessage(payload.GetOriginalReceiver(), bis);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	445 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	446
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	447
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	448 OracleScheduler::OracleScheduler(IOracle& oracle,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	449 IMessageEmitter& emitter,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	450 unsigned int maxHighPriority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	451 unsigned int maxStandardPriority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	452 unsigned int maxLowPriority) :
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	453 oracle_(oracle),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	454 emitter_(emitter),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	455 maxHighPriorityCommands_(maxHighPriority),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	456 maxStandardPriorityCommands_(maxStandardPriority),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	457 maxLowPriorityCommands_(maxLowPriority),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	458 activeHighPriorityCommands_(0),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	459 activeStandardPriorityCommands_(0),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	460 activeLowPriorityCommands_(0),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	461 totalScheduled_(0),
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	462 totalProcessed_(0)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	463 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	464 assert(PRIORITY_HIGH < 0 &&
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	465 PRIORITY_LOW > 0);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	466
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	467 if (maxLowPriority <= 0)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	468 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	469 // There must be at least 1 lane available to deal with low-priority commands
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	470 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	471 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	472 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	473
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	474
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	475 boost::shared_ptr<OracleScheduler> OracleScheduler::Create(IOracle& oracle,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	476 IObservable& oracleObservable,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	477 IMessageEmitter& emitter,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	478 unsigned int maxHighPriority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	479 unsigned int maxStandardPriority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	480 unsigned int maxLowPriority)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	481 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	482 boost::shared_ptr<OracleScheduler> scheduler
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	483 (new OracleScheduler(oracle, emitter, maxHighPriority, maxStandardPriority, maxLowPriority));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	484 scheduler->Register<GetOrthancImageCommand::SuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	485 scheduler->Register<GetOrthancWebViewerJpegCommand::SuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	486 scheduler->Register<HttpCommand::SuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	487 scheduler->Register<OrthancRestApiCommand::SuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	488 scheduler->Register<ReadFileCommand::SuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	489 scheduler->Register<OracleCommandExceptionMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	490
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	491 #if ORTHANC_ENABLE_DCMTK == 1
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	492 scheduler->Register<ParseDicomSuccessMessage>(oracleObservable, &OracleScheduler::Handle);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	493 #endif
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	494
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	495 return scheduler;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	496 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	497
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	498
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	499 OracleScheduler::~OracleScheduler()
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	500 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	501 CancelAllRequests();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	502 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	503
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	504
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	505 void OracleScheduler::CancelRequests(boost::shared_ptr<IObserver> receiver)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	506 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	507 RemoveReceiverFromQueue(standardPriorityQueue_, receiver);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	508 RemoveReceiverFromQueue(highPriorityQueue_, receiver);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	509 RemoveReceiverFromQueue(lowPriorityQueue_, receiver);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	510 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	511
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	512
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	513 void OracleScheduler::CancelAllRequests()
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	514 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	515 ClearQueue(standardPriorityQueue_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	516 ClearQueue(highPriorityQueue_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	517 ClearQueue(lowPriorityQueue_);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	518 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	519
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	520
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	521 void OracleScheduler::Schedule(boost::shared_ptr<IObserver> receiver,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	522 int priority,
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	523 IOracleCommand* command /* Takes ownership */)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	524 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	525 std::auto_ptr<ScheduledCommand> pending(new ScheduledCommand(receiver, dynamic_cast<IOracleCommand*>(command)));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	526
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	527 /**
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	528 * Safeguard to remember that a new "Handle()" method and a call
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	529 * to "scheduler->Register()" must be implemented for each
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	530 * possible oracle command.
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	531 **/
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	532 assert(command->GetType() == IOracleCommand::Type_GetOrthancImage \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	533 command->GetType() == IOracleCommand::Type_GetOrthancWebViewerJpeg \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	534 command->GetType() == IOracleCommand::Type_Http \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	535 command->GetType() == IOracleCommand::Type_OrthancRestApi \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	536 command->GetType() == IOracleCommand::Type_ParseDicomFromFile \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	537 command->GetType() == IOracleCommand::Type_ParseDicomFromWado \|\|
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	538 command->GetType() == IOracleCommand::Type_ReadFile);
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	539
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	540 if (priority <= PRIORITY_HIGH)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	541 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	542 highPriorityQueue_.insert(std::make_pair(priority, pending.release()));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	543 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	544 else if (priority >= PRIORITY_LOW)
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	545 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	546 lowPriorityQueue_.insert(std::make_pair(priority, pending.release()));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	547 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	548 else
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	549 {
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	550 standardPriorityQueue_.insert(std::make_pair(priority, pending.release()));
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	551 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	552
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	553 totalScheduled_ ++;
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	554
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	555 SpawnCommands();
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	556 }
c471a0aa137b adding the next generation of loaders Sebastien Jodogne <s.jodogne@gmail.com> parents: diff changeset	557 }

Mercurial > hg > orthanc-stone

annotate Framework/Loaders/OracleScheduler.cpp @ 1271:0ca50d275b9a broker