99 lines
3.6 KiB
C
99 lines
3.6 KiB
C
|
#ifndef _CUSTOM_WORK_SOCKET_H_
|
|||
|
|
#define _CUSTOM_WORK_SOCKET_H_
|
|||
|
|
|
|||
|
|
/************************************************************
|
|||
|
|
*
|
|||
|
|
* 工作套接字连接模型类
|
|||
|
|
*
|
|||
|
|
* 数据的发送使用队列堆积的形式，而不是使用缓冲区堆积
|
|||
|
|
* 的方式，所有提交的待发送的数据将被保存到一个列表中排队
|
|||
|
|
* 发送，在添加发送数据的时候，可以指定是否优先发送以便将
|
|||
|
|
* 数据插入到待发队列的头部而优先发送。
|
|||
|
|
*
|
|||
|
|
* 凡是当链接断开的时候，所有已经提交到发送队列中的数据包均
|
|||
|
|
* 不会被销毁或丢弃，会在链接建立后继续发送。因此，如果希望在连
|
|||
|
|
* 接断开后立刻清空发送队列，则子类必须覆盖Disconnected函数或
|
|||
|
|
* 在OnDisconnected通知函数中调用clearSendList。
|
|||
|
|
*
|
|||
|
|
***********************************************************/
|
|||
|
|
|
|||
|
|
using namespace wylib::time::tick64;
|
|||
|
|
using namespace wylib::container;
|
|||
|
|
using namespace wylib::sync::lock;
|
|||
|
|
|
|||
|
|
class CCustomWorkSocket :
|
|||
|
|
public wylib::inet::socket::CCustomSocket,
|
|||
|
|
public CSendPacketPool
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
typedef CCustomSocket Inherited;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
CCustomWorkSocket();
|
|||
|
|
virtual ~CCustomWorkSocket();
|
|||
|
|
|
|||
|
|
//发送内部消息
|
|||
|
|
VOID PostInternalMessage(UINT uMsg, UINT64 uParam1, UINT64 uParam2, UINT64 uParam3,UINT64 uParam4=0);
|
|||
|
|
//获取连接的时间
|
|||
|
|
TICKCOUNT getConnectTick(){ return m_dwConnectTick;}
|
|||
|
|
|
|||
|
|
//初始化网络库（调用WSAStartup）,返回0表示成功
|
|||
|
|
static INT InitSocketLib();
|
|||
|
|
//卸载网络库（调用WSACleanup）
|
|||
|
|
static VOID UnintSocketLib();
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
/**** 以下函数集为子类可能需要覆盖的函数 ****/
|
|||
|
|
//处理接受到的服务器数据包
|
|||
|
|
virtual VOID ProcessRecvBuffers(PDATABUFFER pDataBuffer);
|
|||
|
|
//进行内部消息的处理分派
|
|||
|
|
virtual VOID DispatchInternalMessage(UINT uMsg, UINT64 uParam1, UINT64 uParam2, UINT64 uParam3,UINT64 uParam4);
|
|||
|
|
//每次逻辑循环的例行逻辑处理函数
|
|||
|
|
virtual VOID OnRun();
|
|||
|
|
//例行执行函数
|
|||
|
|
virtual VOID SingleRun();
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
//交换接收、处理的缓冲包
|
|||
|
|
VOID SwapRecvProcessBuffers();
|
|||
|
|
//清除所有待发数据
|
|||
|
|
VOID ClearSendBuffers();
|
|||
|
|
//添加待发送数据
|
|||
|
|
//★★应当尽量使用allocSendPacket以及flushSendPacket进行数据发送★★
|
|||
|
|
VOID AppendSendBuffer(LPCVOID lpData, const size_t dwSize);
|
|||
|
|
//读取套接字数据
|
|||
|
|
VOID ReadSocket();
|
|||
|
|
//发送数据队列中的数据
|
|||
|
|
VOID SendSocketBuffers();
|
|||
|
|
//处理内部消息列表
|
|||
|
|
VOID ProcessInternalMessages();
|
|||
|
|
//获取接收数据缓冲，仅为子类提供
|
|||
|
|
inline const PDATABUFFER getRecvBuffer(){ return m_pRecvBuffer; }
|
|||
|
|
//获取接受处理缓冲，仅为子类提供
|
|||
|
|
inline const PDATABUFFER getProcRecvBuffer(){ return m_pProcRecvBuffer; }
|
|||
|
|
//获取内部消息队列中的消息数量，仅为子类提供
|
|||
|
|
inline INT_PTR getInternalMessageCount(){ return m_ItnMsgList.count(); }
|
|||
|
|
inline void SetSendDataInSingleRun(bool bFlag) { m_bSendData = bFlag; }
|
|||
|
|
protected:
|
|||
|
|
//覆盖父类断开连接的处理函数
|
|||
|
|
VOID Disconnected();
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
TICKCOUNT m_dwMsgTick; //上次通信的时间
|
|||
|
|
CBufferAllocator m_Allocator; //内存池
|
|||
|
|
TICKCOUNT m_dwConnectTick; //连接建立的时间
|
|||
|
|
TICKCOUNT m_dwFreeMemoryTime; //清理内存的时间
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
DATABUFFER m_RecvBuffers[2]; //网络数据接收和处理缓冲器
|
|||
|
|
PDATABUFFER m_pRecvBuffer; //接收数据缓冲区
|
|||
|
|
PDATABUFFER m_pProcRecvBuffer; //处理接收到的数据缓冲区
|
|||
|
|
CQueueList<PAPPINTERNALMSG> m_ItnMsgList; //内部消息列表
|
|||
|
|
CCSLock m_ItnMsgListLock; //内部消息列表锁
|
|||
|
|
bool m_bSendData; // 标记在SingleRun是否发送数据
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|