44 lines
964 B
C++
44 lines
964 B
C++
#pragma once
|
|
|
|
#include <functional>
|
|
|
|
#include "VoidNet/TcpClient.hpp"
|
|
|
|
#include "HLAPI/NetworkMessage.hpp"
|
|
|
|
namespace std::net
|
|
{
|
|
class TcpConnection
|
|
{
|
|
public:
|
|
friend class TcpConnectionHandler;
|
|
|
|
TcpConnection();
|
|
TcpConnection(TcpClient *client);
|
|
|
|
std::shared_ptr<TcpClient> GetClient();
|
|
uint32_t GetID();
|
|
|
|
bool Connect(IPAddress addr);
|
|
|
|
template<typename T>
|
|
void SendMessage(DistributionMode mode, uint32_t destinationId, uint32_t tag, void *data)
|
|
{
|
|
NetworkMessage msg(m_id, mode, destinationId, tag, data, sizeof(T));
|
|
sendMessage(msg);
|
|
}
|
|
|
|
void ReceiveData();
|
|
|
|
std::function<void(uint32_t, DistributionMode, uint32_t, uint32_t, void*)> DataReceivedEvent;
|
|
std::function<void(std::string)> DisconnectedEvent;
|
|
std::function<void(uint32_t, void*)> NewConnectionEvent;
|
|
std::function<void()> OnConnectionEvent;
|
|
|
|
private:
|
|
bool sendMessage(const NetworkMessage &msg);
|
|
|
|
std::shared_ptr<TcpClient> m_client;
|
|
uint32_t m_id;
|
|
};
|
|
} |