#pragma once

#ifndef _HMNETWORK
#include "IOCPController.h"
#include "NetPacket.h"
#define _HMNETWORK
#endif

class AFX_EXT_CLASS CNetJoiner
{
public:
	CNetJoiner()
	{
		m_pCtx				= NULL;
		Reset();
	}
	virtual ~CNetJoiner()
	{
		Reset();
	}
	
	void					Reset()
	{
		m_nModuleNo			= -1;
		if (m_pCtx)
			m_pCtx->Reset();
		m_pCtx				= NULL;
		m_ulIP				= -1;

		m_bUseHeartbeat		= FALSE;
		m_dwRecvHeartbeat	= 0;
	}

	WSABUF*					GetSendBuffer()				{ return m_pCtx ? m_pCtx->sendContext->GetBuffer() : NULL; }
	pPerSocketContext		GetSocketContext()			{ return m_pCtx; }
	ULONG					GetClientIP()						{ return m_ulIP; }
	int						GetModuleNo()				{ return m_nModuleNo; }
	
	void					SetSocketContext(pPerSocketContext pCtx)	{ m_pCtx = pCtx; }
	void					SetClientIP(ULONG ulIP)		{ m_ulIP = ulIP; }
	void					SetModuleNo(int nModuleNo, BOOL bUseHeartbeat)
	{
		m_nModuleNo = nModuleNo;
		m_bUseHeartbeat = bUseHeartbeat;
	}
	void					SetUseHeartbeat(BOOL bUse)	{ m_bUseHeartbeat = bUse; }
	void					SetRecvHeartbeat()			{ m_dwRecvHeartbeat = GetTickCount(); }
	DWORD					GetRecvHeartbeat()			{ return m_bUseHeartbeat ? m_dwRecvHeartbeat : 0; }

protected:
	BOOL					m_bUseHeartbeat;
	DWORD					m_dwRecvHeartbeat;

	int						m_nModuleNo;
	pPerSocketContext		m_pCtx;
	ULONG					m_ulIP;
};