blob: fce3e04e30e2b42ed555ab82ef10692a0ec5fcac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#include "drape_frontend/message_queue.hpp"
#include "base/assert.hpp"
#include "base/stl_add.hpp"
#include "std/chrono.hpp"
namespace df
{
MessageQueue::MessageQueue()
: m_isWaiting(false)
{
}
MessageQueue::~MessageQueue()
{
CancelWaitImpl();
ClearQuery();
}
drape_ptr<Message> MessageQueue::PopMessage(bool waitForMessage)
{
unique_lock<mutex> lock(m_mutex);
if (waitForMessage && m_messages.empty())
{
m_isWaiting = true;
m_condition.wait(lock, [this]() { return !m_isWaiting; });
m_isWaiting = false;
}
if (m_messages.empty())
return nullptr;
drape_ptr<Message> msg = move(m_messages.front().first);
m_messages.pop_front();
return msg;
}
void MessageQueue::PushMessage(drape_ptr<Message> && message, MessagePriority priority)
{
lock_guard<mutex> lock(m_mutex);
switch (priority)
{
case MessagePriority::Normal:
{
m_messages.emplace_back(move(message), priority);
break;
}
case MessagePriority::High:
{
auto iter = m_messages.begin();
while (iter != m_messages.end() && iter->second > MessagePriority::High) { iter++; }
m_messages.emplace(iter, move(message), priority);
break;
}
case MessagePriority::UberHighSingleton:
{
bool found = false;
auto iter = m_messages.begin();
while (iter != m_messages.end() && iter->second == MessagePriority::UberHighSingleton)
{
if (iter->first->GetType() == message->GetType())
{
found = true;
break;
}
iter++;
}
if (!found)
m_messages.emplace_front(move(message), priority);
break;
}
default:
ASSERT(false, ("Unknown message priority type"));
}
CancelWaitImpl();
}
void MessageQueue::FilterMessages(TFilterMessageFn needFilterMessageFn)
{
ASSERT(needFilterMessageFn != nullptr, ());
lock_guard<mutex> lock(m_mutex);
for (auto it = m_messages.begin(); it != m_messages.end(); )
{
if (needFilterMessageFn(make_ref(it->first)))
it = m_messages.erase(it);
else
++it;
}
}
#ifdef DEBUG_MESSAGE_QUEUE
bool MessageQueue::IsEmpty() const
{
lock_guard<mutex> lock(m_mutex);
return m_messages.empty();
}
size_t MessageQueue::GetSize() const
{
lock_guard<mutex> lock(m_mutex);
return m_messages.size();
}
#endif
void MessageQueue::CancelWait()
{
lock_guard<mutex> lock(m_mutex);
CancelWaitImpl();
}
void MessageQueue::CancelWaitImpl()
{
if (m_isWaiting)
{
m_isWaiting = false;
m_condition.notify_all();
}
}
void MessageQueue::ClearQuery()
{
m_messages.clear();
}
} // namespace df
|
