1 | // |
---|---|
2 | // TimedNotificationQueue.cpp |
3 | // |
4 | // Library: Foundation |
5 | // Package: Notifications |
6 | // Module: TimedNotificationQueue |
7 | // |
8 | // Copyright (c) 2009, Applied Informatics Software Engineering GmbH. |
9 | // and Contributors. |
10 | // |
11 | // SPDX-License-Identifier: BSL-1.0 |
12 | // |
13 | |
14 | |
15 | #include "Poco/TimedNotificationQueue.h" |
16 | #include "Poco/Notification.h" |
17 | #include <limits> |
18 | |
19 | |
20 | namespace Poco { |
21 | |
22 | |
23 | TimedNotificationQueue::TimedNotificationQueue() |
24 | { |
25 | } |
26 | |
27 | |
28 | TimedNotificationQueue::~TimedNotificationQueue() |
29 | { |
30 | try |
31 | { |
32 | clear(); |
33 | } |
34 | catch (...) |
35 | { |
36 | poco_unexpected(); |
37 | } |
38 | } |
39 | |
40 | |
41 | void TimedNotificationQueue::enqueueNotification(Notification::Ptr pNotification, Timestamp timestamp) |
42 | { |
43 | poco_check_ptr (pNotification); |
44 | |
45 | Timestamp tsNow; |
46 | Clock clock; |
47 | Timestamp::TimeDiff diff = timestamp - tsNow; |
48 | clock += diff; |
49 | |
50 | FastMutex::ScopedLock lock(_mutex); |
51 | _nfQueue.insert(NfQueue::value_type(clock, pNotification)); |
52 | _nfAvailable.set(); |
53 | } |
54 | |
55 | |
56 | void TimedNotificationQueue::enqueueNotification(Notification::Ptr pNotification, Clock clock) |
57 | { |
58 | poco_check_ptr (pNotification); |
59 | |
60 | FastMutex::ScopedLock lock(_mutex); |
61 | _nfQueue.insert(NfQueue::value_type(clock, pNotification)); |
62 | _nfAvailable.set(); |
63 | } |
64 | |
65 | |
66 | Notification* TimedNotificationQueue::dequeueNotification() |
67 | { |
68 | FastMutex::ScopedLock lock(_mutex); |
69 | |
70 | NfQueue::iterator it = _nfQueue.begin(); |
71 | if (it != _nfQueue.end()) |
72 | { |
73 | Clock::ClockDiff sleep = -it->first.elapsed(); |
74 | if (sleep <= 0) |
75 | { |
76 | Notification::Ptr pNf = it->second; |
77 | _nfQueue.erase(it); |
78 | return pNf.duplicate(); |
79 | } |
80 | } |
81 | return 0; |
82 | } |
83 | |
84 | |
85 | Notification* TimedNotificationQueue::waitDequeueNotification() |
86 | { |
87 | for (;;) |
88 | { |
89 | _mutex.lock(); |
90 | NfQueue::iterator it = _nfQueue.begin(); |
91 | if (it != _nfQueue.end()) |
92 | { |
93 | _mutex.unlock(); |
94 | Clock::ClockDiff sleep = -it->first.elapsed(); |
95 | if (sleep <= 0) |
96 | { |
97 | return dequeueOne(it).duplicate(); |
98 | } |
99 | else if (!wait(sleep)) |
100 | { |
101 | return dequeueOne(it).duplicate(); |
102 | } |
103 | else continue; |
104 | } |
105 | else |
106 | { |
107 | _mutex.unlock(); |
108 | } |
109 | _nfAvailable.wait(); |
110 | } |
111 | } |
112 | |
113 | |
114 | Notification* TimedNotificationQueue::waitDequeueNotification(long milliseconds) |
115 | { |
116 | while (milliseconds >= 0) |
117 | { |
118 | _mutex.lock(); |
119 | NfQueue::iterator it = _nfQueue.begin(); |
120 | if (it != _nfQueue.end()) |
121 | { |
122 | _mutex.unlock(); |
123 | Clock now; |
124 | Clock::ClockDiff sleep = it->first - now; |
125 | if (sleep <= 0) |
126 | { |
127 | return dequeueOne(it).duplicate(); |
128 | } |
129 | else if (sleep <= 1000*Clock::ClockDiff(milliseconds)) |
130 | { |
131 | if (!wait(sleep)) |
132 | { |
133 | return dequeueOne(it).duplicate(); |
134 | } |
135 | else |
136 | { |
137 | milliseconds -= static_cast<long>((now.elapsed() + 999)/1000); |
138 | continue; |
139 | } |
140 | } |
141 | } |
142 | else |
143 | { |
144 | _mutex.unlock(); |
145 | } |
146 | if (milliseconds > 0) |
147 | { |
148 | Clock now; |
149 | _nfAvailable.tryWait(milliseconds); |
150 | milliseconds -= static_cast<long>((now.elapsed() + 999)/1000); |
151 | } |
152 | else return 0; |
153 | } |
154 | return 0; |
155 | } |
156 | |
157 | |
158 | bool TimedNotificationQueue::wait(Clock::ClockDiff interval) |
159 | { |
160 | const Clock::ClockDiff MAX_SLEEP = 8*60*60*Clock::ClockDiff(1000000); // sleep at most 8 hours at a time |
161 | while (interval > 0) |
162 | { |
163 | Clock now; |
164 | Clock::ClockDiff sleep = interval <= MAX_SLEEP ? interval : MAX_SLEEP; |
165 | if (_nfAvailable.tryWait(static_cast<long>((sleep + 999)/1000))) |
166 | return true; |
167 | interval -= now.elapsed(); |
168 | } |
169 | return false; |
170 | } |
171 | |
172 | |
173 | bool TimedNotificationQueue::empty() const |
174 | { |
175 | FastMutex::ScopedLock lock(_mutex); |
176 | return _nfQueue.empty(); |
177 | } |
178 | |
179 | |
180 | int TimedNotificationQueue::size() const |
181 | { |
182 | FastMutex::ScopedLock lock(_mutex); |
183 | return static_cast<int>(_nfQueue.size()); |
184 | } |
185 | |
186 | |
187 | void TimedNotificationQueue::clear() |
188 | { |
189 | FastMutex::ScopedLock lock(_mutex); |
190 | _nfQueue.clear(); |
191 | } |
192 | |
193 | |
194 | Notification::Ptr TimedNotificationQueue::dequeueOne(NfQueue::iterator& it) |
195 | { |
196 | FastMutex::ScopedLock lock(_mutex); |
197 | Notification::Ptr pNf = it->second; |
198 | _nfQueue.erase(it); |
199 | return pNf; |
200 | } |
201 | |
202 | |
203 | } // namespace Poco |
204 |