#pragma once



// 网关客户端类，负责处理和服务器数据包的接收和发送
// 采取阻塞方式实现
class CCustomGateClient : protected CSendPacketPool
{
public:
	typedef struct tagGateMsg
	{
		GATEMSGHDR		header;
		int				nBuffLen;			// 数据buff的长度
		char*			data;
	}GateMsg;


	// ctor and dtor
	CCustomGateClient();
	virtual ~CCustomGateClient();

	// 启动客户端
	bool Startup();
	// 停止客户端
	void Stop();
	// 判读是否连接上服务器
	bool connected() const { return m_bConnected; }
	// 增加发送数据
	void PutMessage(int nIdent, 
					int nSessionIdx, 
					SOCKET nSocket, 
					int nServerIdx, 
					unsigned long long param, 
					char *pBuffer, 
					int nBufSize);

	// 获取客户端名称
	inline LPCTSTR GetClientName() const { return m_szClientName; };
	// 设置客户端名称
	void SetClientName(LPCTSTR sClientName);
	// 获取远程服务器地址
	inline LPCTSTR GetServerHost() const { return m_szSrvHost; }
	// 设置远程服务器地址（支持域名）
	void SetServerHost(LPCTSTR sHost);
	// 获取远程服务器端口
	inline INT_PTR GetServerPort() const { return m_nSrvPort; }
	// 设置远程服务器端口
	void SetServerPort(const INT_PTR nPort);
	// Dump数据包情况
	void Dump();
	
	//覆盖例行执行函数
	void SingleRun();	

protected:	
	// 连接到服务器
	virtual void OnConnected();
	virtual void OnClosed();
	// 连接到服务器成功后发送注册客户端的消息
	virtual void SendRegisteClient();
	// 当长时间没有通信后发送保持连接的消息
	virtual void SendKeepAlive();
	// 接收到网关数据
	//virtual void ProcessRecvBuffers(const GateMsg* msgData) = 0;
	virtual void ProcessRecvBuffers(const GATEMSGHDR* pHeader, LPCSTR pData, size_t nLen) = 0;
	// 关闭套接字
	void Close();
	// 创建套接字
	bool CreateSocket();
	// 连接到服务器
	bool ConnectToServerImpl();	
	// 发送指定长度的数据
	bool SendPointedData(const char* pData, size_t nLen);
	bool ReadPointedData(char* pOutBuff, size_t nLen);
	// 分配发送给服务器的数据包
	CDataPacket* AllocPacket();
	// 分配接收到的数据包
	CDataPacket* AllocRecvPacket();

private:
	// 初始化套接字库
	bool InitSockLib();
	// 卸载套接字库
	void UnintSocketLib();
	// 工作线程函数
#ifdef _MSC_VER
	static DWORD WINAPI staticConnectThreadRountine(LPVOID pParam);
#else
	static void* staticConnectThreadRountine(void* pParam);
#endif
	
	void ConnectToServerRoutine();
	// 发送数据线程
#ifdef _MSC_VER
	static DWORD WINAPI staticSendDataThreadRountine(LPVOID pParam);
#else
	static void* staticSendDataThreadRountine(void* pParam);
#endif
	void SendDataRoutine();
	// 接收数据线程
#ifdef _MSC_VER
	static DWORD WINAPI staticRecvDataThreadRountine(LPVOID pParam);
#else
	static void* staticRecvDataThreadRountine(void* pParam);
#endif
	void RecvDataRoutine();
	void ProcessRecvDataImpl();
	void DestroyPacketList(CQueueList<CDataPacket*>& pkgList);
private:
	SOCKET					m_socket;
	TCHAR					m_szSrvHost[256];		// 服务器地址
	INT_PTR					m_nSrvPort;				// 服务器端口
	TCHAR					m_szClientName[256];	// 客户端名称
	bool					m_bConnected;			// 是否已经连接上
	volatile LONG			m_bStoped;				// 是否标记为停止
#ifdef _MSC_VER
	HANDLE					m_hConnectThread;		// 连接服务器线程
	HANDLE					m_hSendDataThread;		// 发送数据线程
	HANDLE					m_hRecvDataThread;		// 接收数据线程
#else
	pthread_t				m_hConnectThread;
	pthread_t				m_hSendDataThread;
	pthread_t				m_hRecvDataThread;
#endif
	CQueueList<CDataPacket*>m_SendMsgList;			// 发送消息队列
	CCSLock					m_SendMsgLock;			// 发送消息锁	
	CQueueList<CDataPacket*>m_FreeMsgList;			// 空闲消息队列
	CCSLock					m_FreeMsgLock;			// 空闲消息锁	
	GateMsg					m_RecvMsg;				// 当前接受的消息数据
	CBufferAllocator		m_Allocator;			// 内存块管理器
	// 接收数据不直接派发上去，而是经过这个缓冲，保证接收数据的及时性
	CQueueList<CDataPacket*>m_RecvMsgList;			// 接受消息队列
	CCSLock					m_RecvMsgLock;			// 接受消息锁
	CQueueList<CDataPacket*>m_RecvFreeMsgList;		// 空闲消息队列
	CCSLock					m_RecvFreeMsgLock;		// 空闲消息锁	
	CDataPacket				m_SendBigPacket;		// 发送给服务器的大数据包缓存

	int nPacketCount;
};