Mercurial > hg > orthanc
annotate Core/MultiThreading/SharedMessageQueue.cpp @ 789:55dae8c5a6ab
refactoring
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Mon, 05 May 2014 18:43:34 +0200 |
| parents | 0da078f3affc |
| children | a811bdf8b8eb |
| rev | line source |
|---|---|
| 450 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 689 | 3 * Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, |
| 450 | 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 #include "SharedMessageQueue.h" | |
| 34 | |
| 35 namespace Orthanc | |
| 36 { | |
| 37 SharedMessageQueue::SharedMessageQueue(unsigned int maxSize) | |
| 38 { | |
| 39 maxSize_ = maxSize; | |
| 40 } | |
| 41 | |
| 42 | |
| 43 SharedMessageQueue::~SharedMessageQueue() | |
| 44 { | |
| 656 | 45 for (Queue::iterator it = queue_.begin(); it != queue_.end(); ++it) |
| 450 | 46 { |
| 47 delete *it; | |
| 48 } | |
| 49 } | |
| 50 | |
| 51 | |
| 52 void SharedMessageQueue::Enqueue(IDynamicObject* message) | |
| 53 { | |
| 54 boost::mutex::scoped_lock lock(mutex_); | |
| 55 | |
| 56 if (maxSize_ != 0 && queue_.size() > maxSize_) | |
| 57 { | |
| 58 // Too many elements in the queue: First remove the oldest | |
| 59 delete queue_.front(); | |
| 60 queue_.pop_front(); | |
| 61 } | |
| 62 | |
| 63 queue_.push_back(message); | |
| 64 elementAvailable_.notify_one(); | |
| 65 } | |
| 66 | |
| 67 | |
| 68 IDynamicObject* SharedMessageQueue::Dequeue(int32_t millisecondsTimeout) | |
| 69 { | |
| 70 boost::mutex::scoped_lock lock(mutex_); | |
| 71 | |
| 72 // Wait for a message to arrive in the FIFO queue | |
| 73 while (queue_.empty()) | |
| 74 { | |
| 75 if (millisecondsTimeout == 0) | |
| 76 { | |
| 77 elementAvailable_.wait(lock); | |
| 78 } | |
| 79 else | |
| 80 { | |
| 81 bool success = elementAvailable_.timed_wait | |
| 82 (lock, boost::posix_time::milliseconds(millisecondsTimeout)); | |
| 83 if (!success) | |
| 84 { | |
| 85 return NULL; | |
| 86 } | |
| 87 } | |
| 88 } | |
| 89 | |
| 90 std::auto_ptr<IDynamicObject> message(queue_.front()); | |
| 91 queue_.pop_front(); | |
|
723
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
92 |
|
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
93 if (queue_.empty()) |
|
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
94 { |
|
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
95 emptied_.notify_all(); |
|
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
96 } |
| 450 | 97 |
| 98 return message.release(); | |
| 99 } | |
| 452 | 100 |
| 101 | |
| 102 | |
| 103 bool SharedMessageQueue::WaitEmpty(int32_t millisecondsTimeout) | |
| 104 { | |
| 105 boost::mutex::scoped_lock lock(mutex_); | |
| 106 | |
| 107 // Wait for the queue to become empty | |
|
723
0da078f3affc
multithreading tests
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
108 while (!queue_.empty()) |
| 452 | 109 { |
| 110 if (millisecondsTimeout == 0) | |
| 111 { | |
| 112 emptied_.wait(lock); | |
| 113 } | |
| 114 else | |
| 115 { | |
| 116 if (!emptied_.timed_wait | |
| 117 (lock, boost::posix_time::milliseconds(millisecondsTimeout))) | |
| 118 { | |
| 119 return false; | |
| 120 } | |
| 121 } | |
| 122 } | |
| 123 | |
| 124 return true; | |
| 125 } | |
| 450 | 126 } |