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
|
#include "Session.hpp"
#include <algorithm>
using namespace Epistmool::Server;
namespace {
constexpr int kInvalidSessionIndex = -1;
} // namespace
SessionId SessionId::makeInvalid()
{
return SessionId{
.index = kInvalidSessionIndex,
};
}
bool SessionId::isInvalid() const
{
return index == kInvalidSessionIndex;
}
SessionId Session::getSessionId() const
{
return mSessionId;
}
const std::vector<ConnectionId>& Session::getConnections() const
{
return mConnections;
}
Session::Session(SessionId id)
: mSessionId{ id }
{
}
// TODO delete session if isn't active for N period of time
SessionManager::SessionManager(QObject* parent)
: QObject(parent)
{
}
Session* SessionManager::findOrCreateSession(SessionId id)
{
auto it = mSessions.find(id.index);
if (it != mSessions.end()) {
return &it->second;
} else {
SessionId newId{ mNextIndex++ };
auto [it, _] = mSessions.insert({ newId.index, Session(newId) });
return &it->second;
}
}
Session* SessionManager::findSession(SessionId id)
{
auto it = mSessions.find(id.index);
if (it != mSessions.end()) {
return &it->second;
} else {
return nullptr;
}
}
void SessionManager::addConnection(SessionId id, ConnectionId connId)
{
// Add session -> connection links
auto session = findSession(id);
if (session) {
auto location = std::lower_bound(
session->mConnections.begin(),
session->mConnections.end(),
connId,
[](const ConnectionId& a, const ConnectionId& b) -> bool {
return a.index < b.index;
});
session->mConnections.insert(location, connId);
}
// Add connection -> session links
mConnection2SessionMap.insert(connId.index, id.index);
}
void SessionManager::dropConnection(ConnectionId connId)
{
// Remove session -> connection links
auto it = mConnection2SessionMap.find(connId.index);
while (it != mConnection2SessionMap.end() && it.key() == connId.index) {
auto& connList = mSessions.at(it.value()).mConnections;
auto range = std::equal_range(
connList.begin(),
connList.end(),
connId,
[](const ConnectionId& a, const ConnectionId& b) -> bool {
return a.index < b.index;
});
connList.erase(range.first, range.second);
++it;
}
// Remove connection -> sesion links
mConnection2SessionMap.remove(connId.index);
}
|
