fix

src/websocket_server.h

@@ -0,0 +1,50 @@
+#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;
+};