51 lines
1.3 KiB
C++
51 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include "message_broker.h"
|
|
#include "connection_pool.h"
|
|
#include <uWebSockets/App.h>
|
|
#include <unordered_map>
|
|
#include <string>
|
|
#include <atomic>
|
|
#include <thread>
|
|
#include <shared_mutex>
|
|
#include <nlohmann/json.hpp>
|
|
#include <queue>
|
|
#include <condition_variable>
|
|
|
|
struct WebSocketUserData {
|
|
std::string userId;
|
|
bool pongReceived = true;
|
|
};
|
|
|
|
class WebSocketServer {
|
|
public:
|
|
WebSocketServer(int port, ConnectionPool &pool, MessageBroker &broker);
|
|
void run();
|
|
void stop();
|
|
|
|
private:
|
|
void startServer();
|
|
void processMessageQueue();
|
|
void pingClients();
|
|
void handleBrokerMessage(const std::string &message);
|
|
void handleWebSocketMessage(uWS::WebSocket<false, true, WebSocketUserData> *ws, std::string_view message, uWS::OpCode opCode);
|
|
void handleWebSocketClose(uWS::WebSocket<false, true, WebSocketUserData> *ws);
|
|
std::string getUserIdFromFalukantUserId(int &userId);
|
|
|
|
int port;
|
|
ConnectionPool &pool;
|
|
MessageBroker &broker;
|
|
|
|
std::atomic<bool> running{false};
|
|
std::thread serverThread;
|
|
std::thread messageProcessingThread;
|
|
std::thread pingThread;
|
|
|
|
std::unordered_map<std::string, uWS::WebSocket<false, true, WebSocketUserData> *> connections;
|
|
std::shared_mutex connectionsMutex;
|
|
|
|
std::queue<std::string> messageQueue;
|
|
std::mutex queueMutex;
|
|
std::condition_variable queueCV;
|
|
};
|