mychatsocket.cpp

关于使用VC编程的代码

// MyChatSocket.cpp : implementation file
//

#include "stdafx.h"
#include "MyChatClt.h"
#include "MyChatSocket.h"
#include "MyChatCltDoc.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CMyChatSocket

CMyChatSocket::CMyChatSocket()
{
	m_pDoc = NULL;

}

CMyChatSocket::CMyChatSocket(CMyChatCltDoc* pDoc)
{
	m_pDoc = pDoc;
}

CMyChatSocket::~CMyChatSocket()
{
	Finish_Comm();
}


// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(CMyChatSocket, CSocket)
	//{{AFX_MSG_MAP(CMyChatSocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0

/////////////////////////////////////////////////////////////////////////////
// CMyChatSocket member functions

void CMyChatSocket::OnReceive(int nErrorCode) 
{
	ASSERT(m_pDoc != NULL);
	m_pDoc->OnReadPendingData();
	CSocket::OnReceive(nErrorCode);
}

void CMyChatSocket::Init()
{
	m_pFile = new CSocketFile(this);
	m_pArIn = new CArchive(m_pFile, CArchive::load);
	m_pArOut = new CArchive(m_pFile, CArchive::store);
}

CString CMyChatSocket::ReceiveData()
{
	CString str;
	*m_pArIn >> str;
	return str;
}

void CMyChatSocket::SendData(CString strMsg)
{
	*m_pArOut << m_strClientID + ":" + strMsg;
	m_pArOut->Flush();
//	AfxMessageBox("enter Client:sendData");
}

void CMyChatSocket::Finish_Comm()
{
	if(m_pArIn != NULL)
	{
		m_pArIn->Close();
		delete m_pArIn;
		m_pArIn = NULL;
	}
	if(m_pArOut != NULL)
	{
		m_pArOut->Close();
		delete m_pArOut;
		m_pArOut = NULL;
	}
	if(m_pFile != NULL)
	{
		m_pFile->Close();
		delete m_pFile;
		m_pFile = NULL;
	}
}