106 lines
2.3 KiB
C++
106 lines
2.3 KiB
C++
#ifndef _WORK_SOCKET_H_
|
||
#define _WORK_SOCKET_H_
|
||
|
||
#include "os_def.h"
|
||
#define flushProtoPacket(x) FlushProtoPacket(x, __FILE__, __LINE__)
|
||
using namespace tick64;
|
||
using namespace container;
|
||
using namespace lock;
|
||
using namespace net;
|
||
|
||
class WorkSocket :
|
||
public BaseSocket,
|
||
public SendPacketPool
|
||
{
|
||
public:
|
||
const static size_t MAX_DATA_LEN = 32 * 1024 * 1024; // 每个网络数据包加上包头的最大长度
|
||
public:
|
||
WorkSocket(const char* name);
|
||
virtual ~WorkSocket();
|
||
virtual void SingleRun();
|
||
|
||
//数据包填充完毕后需要调用flushProtoPacket提交到发送队列
|
||
DataPacket& allocProtoPacket(uint16_t cmd);
|
||
DataPacket& AllocProtoPacket(int nIdent, int nSessionIdx, SOCKET nSocket, int nServerIdx);
|
||
|
||
void FlushProtoPacket(DataPacket& packet, const char* fn, int line);
|
||
|
||
//发送内部消息
|
||
void PostMsg(UINT uMsg, size_t para1, size_t para2, size_t para3, size_t para4);
|
||
|
||
//初始化网络库（调用WSAStartup）,返回0表示成功
|
||
static int InitSocketLib();
|
||
//卸载网络库（调用WSACleanup）
|
||
static void UnintSocketLib();
|
||
|
||
// 是否有未处理的数据
|
||
bool HasRemainData();
|
||
// 是否自动发送数据包
|
||
void SetAutoSend(bool flag)
|
||
{
|
||
auto_send_ = flag;
|
||
}
|
||
|
||
// 发送数据
|
||
void SendBuffers();
|
||
|
||
// 线程在每一次loop是否空转
|
||
inline bool GetBusy()
|
||
{
|
||
return busy_;
|
||
}
|
||
protected:
|
||
virtual void OnRecv();
|
||
|
||
/**** 以下函数集为子类可能需要覆盖的函数 ****/
|
||
virtual bool PreProcess(DataPacketReader*)
|
||
{
|
||
return true;
|
||
}
|
||
|
||
virtual void OnRecvSysMsg(unsigned int msg, size_t p1, size_t p2, size_t p3, size_t p4);
|
||
virtual void OnRun();
|
||
virtual void OnRecv(const UInt16 cmd, char* buf, int size);
|
||
|
||
void ProcessRecvData(DataPacket* data_buf);
|
||
|
||
protected:
|
||
void ProcessSysMsg();
|
||
void ClearSendBuffers();
|
||
int ReadSocket();
|
||
inline int GetMsgCount()
|
||
{
|
||
return msg_list_.count();
|
||
}
|
||
void Disconnected();
|
||
void SendKeepAlive();
|
||
void SetRecvSize(int size)
|
||
{
|
||
recv_size_ = size;
|
||
}
|
||
inline void SleepX()
|
||
{
|
||
Sleep(1);
|
||
}
|
||
inline void SetBusy(bool busy)
|
||
{
|
||
busy_ = busy_ || busy;
|
||
}
|
||
|
||
protected:
|
||
time_t last_msg_t_; //上次通信的时间
|
||
QueueList<APPINTERNALMSG> msg_list_; //内部消息列表
|
||
bool busy_;
|
||
private:
|
||
DataPacket* proc_pack_;
|
||
BaseAllocator allocator_;
|
||
|
||
Mutex msg_lock_; //内部消息列表锁
|
||
size_t recv_size_;
|
||
bool auto_send_;
|
||
};
|
||
|
||
|
||
#endif
|
||
|