blob: 290b8c638ac2996c1ea2f015b90cd0731be326b9 (
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
|
#include "sessionManager.hpp"
#include "server.hpp"
#include "session.hpp"
#include <algorithm>
#include <unistd.h>
#include <utility>
CSessionManager::CSessionManager() = default;
CSessionManager::~CSessionManager() {
for (auto &[thread, session] : m_vSessions)
thread.join();
}
std::pair<std::jthread, std::shared_ptr<CSession>> *CSessionManager::newSession(Hyprutils::OS::CFileDescriptor sockfd) {
std::shared_ptr session = std::make_shared<CSession>(std::move(sockfd));
const auto instance = &m_vSessions.emplace_back(std::jthread(&CSession::run, session), std::move(session));
instance->second->setSelf(instance);
return instance;
}
void CSessionManager::enterLoop() {
while (true)
newSession(Hyprutils::OS::CFileDescriptor{accept(g_pServer->getSocket()->get(), nullptr, nullptr) /* awaits new client/session */});
}
void CSessionManager::broadcast(const std::string &msg, eFormatType type) const {
for (const auto &[thread, session] : m_vSessions)
session->write(msg, type);
}
void CSessionManager::removeSession(std::pair<std::jthread, std::shared_ptr<CSession>> *session) {
auto it = std::ranges::find_if(m_vSessions, [session](const auto &s) { return s.second.get() == session->second.get(); });
if (it != m_vSessions.end()) {
if (it->first.joinable())
it->first.detach(); // .detach the thread since it's removing itself
m_vSessions.erase(it);
}
}
|
