46 lines
1.2 KiB
C++
46 lines
1.2 KiB
C++
#include "core/CNShared.hpp"
|
|
|
|
static std::unordered_map<int64_t, LoginMetadata*> logins;
|
|
static std::mutex mtx;
|
|
|
|
void CNShared::storeLoginMetadata(int64_t sk, LoginMetadata *lm) {
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
// take ownership of connection data
|
|
logins[sk] = lm;
|
|
}
|
|
|
|
LoginMetadata* CNShared::getLoginMetadata(int64_t sk) {
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
// fail if the key isn't found
|
|
if (logins.find(sk) == logins.end())
|
|
return nullptr;
|
|
|
|
// transfer ownership of connection data to shard
|
|
LoginMetadata *lm = logins[sk];
|
|
logins.erase(sk);
|
|
|
|
return lm;
|
|
}
|
|
|
|
void CNShared::pruneLoginMetadata(CNServer *serv, time_t currTime) {
|
|
std::lock_guard<std::mutex> lock(mtx);
|
|
|
|
auto it = logins.begin();
|
|
while (it != logins.end()) {
|
|
auto& sk = it->first;
|
|
auto& lm = it->second;
|
|
|
|
if (currTime > lm->timestamp + CNSHARED_TIMEOUT) {
|
|
std::cout << "[WARN] Pruning hung connection attempt" << std::endl;
|
|
|
|
// deallocate object and remove map entry
|
|
delete logins[sk];
|
|
it = logins.erase(it); // skip the invalidated iterator
|
|
} else {
|
|
it++;
|
|
}
|
|
}
|
|
}
|