35 lines
952 B
C++
35 lines
952 B
C++
#pragma once
|
||
#include "IOCPUDPClient.h"
|
||
#include "ikcp.h"
|
||
#include <thread>
|
||
#include <atomic>
|
||
|
||
class IOCPKCPClient : public IOCPUDPClient
|
||
{
|
||
public:
|
||
IOCPKCPClient(State& bExit, bool exit_while_disconnect = false);
|
||
virtual ~IOCPKCPClient();
|
||
|
||
virtual BOOL ConnectServer(const char* szServerIP, unsigned short uPort) override;
|
||
|
||
// 重写接收函数：输入UDP数据给KCP，输出KCP层解包后的数据
|
||
virtual int ReceiveData(char* buffer, int bufSize, int flags) override;
|
||
|
||
virtual bool ProcessRecvData(CBuffer* m_CompressedBuffer, char* szBuffer, int len, int flag) override;
|
||
|
||
// 重写发送函数：将应用数据通过KCP发送
|
||
virtual int SendTo(const char* buf, int len, int flags) override;
|
||
|
||
private:
|
||
// KCP发送数据的回调函数，负责调用UDP的sendto
|
||
static int kcpOutput(const char* buf, int len, struct IKCPCB* kcp, void* user);
|
||
|
||
// 定时调用ikcp_update的线程函数
|
||
void KCPUpdateLoop();
|
||
|
||
private:
|
||
ikcpcb* kcp_;
|
||
std::thread updateThread_;
|
||
std::atomic<bool> running_;
|
||
};
|