56 lines
1.7 KiB
C++
56 lines
1.7 KiB
C++
#pragma once
|
|
|
|
// WebSocketClient.h
|
|
#include "NetworkInterface.h"
|
|
#include <queue>
|
|
#include <boost/beast/core.hpp>
|
|
#include <boost/beast/websocket.hpp>
|
|
#include <boost/asio/connect.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
|
|
namespace ZL {
|
|
|
|
class WebSocketClient : public INetworkClient {
|
|
private:
|
|
// Переиспользуем io_context из TaskManager
|
|
boost::asio::io_context& ioc_;
|
|
|
|
// Объекты переехали в члены класса
|
|
std::unique_ptr<boost::beast::websocket::stream<boost::beast::tcp_stream>> ws_;
|
|
boost::beast::flat_buffer buffer_;
|
|
|
|
std::queue<std::string> messageQueue;
|
|
std::mutex queueMutex; // Защита для messageQueue
|
|
|
|
std::queue<std::shared_ptr<std::string>> writeQueue_;
|
|
bool isWriting_ = false;
|
|
std::mutex writeMutex_; // Отдельный мьютекс для очереди записи
|
|
|
|
bool connected = false;
|
|
int clientId = -1;
|
|
|
|
std::unordered_map<int, ClientStateInterval> remotePlayers;
|
|
std::mutex playersMutex;
|
|
|
|
void startAsyncRead();
|
|
void processIncomingMessage(const std::string& msg);
|
|
|
|
public:
|
|
explicit WebSocketClient(boost::asio::io_context& ioc) : ioc_(ioc) {}
|
|
|
|
void Connect(const std::string& host, uint16_t port) override;
|
|
|
|
void Poll() override;
|
|
|
|
void Send(const std::string& message) override;
|
|
void doWrite();
|
|
|
|
bool IsConnected() const override { return connected; }
|
|
int GetClientId() const { return clientId; }
|
|
|
|
std::unordered_map<int, ClientStateInterval> getRemotePlayers() override {
|
|
std::lock_guard<std::mutex> lock(playersMutex);
|
|
return remotePlayers;
|
|
}
|
|
};
|
|
} |