VC语言6.0程序设计从入门到精通-第71部分
按键盘上方向键 ← 或 → 可快速上下翻页,按键盘上的 Enter 键可回到本书目录页,按键盘上方向键 ↑ 可回到本页顶部!
————未阅读完?加入书签已便下次继续阅读!
CArchive* m_pArchiveOut 输出
BOOL bIsConnected 是否已经连接
为 CChatClientDoc 类添加成员函数,如表 11…12 所示。
表 11…12 CChatClientDoc 类成员函数及其说明
成员函数 说明
BOOL ConnectSocket(LPCTSTR lpszHandle; 连接服务器,3 个参数依次是用户名、服务器地址
LPCTSTR lpszAddress; int nPort) 和端口号
·314 ·
…………………………………………………………Page 326……………………………………………………………
第 11 章 网络编程
续表
成员函数 说明
void ProcessReceive() 当接收到信息时调用此函数,进行信息接收等操作
void SendMsg(CString& strText; int mCode; BOOL 发送信息,3 个参数分别表示信息内容、信息种类
bSendName)
和是否发送用户名
void ReceiveMsg() 接收信息
void DisplayMsg(LPCTSTR lpszText) 显示信息
void UpdateChattersList(CMsg* pMsg) 更新用户列表
virtual void DeleteContents() 断开时所作的工作
下面开始为各函数编写代码。
首先在构造函数中将各成员变量初始化,代码如下:
CChatClientDoc::CChatClientDoc()
{
// TODO: add one…time construction code here
bIsConnected = FALSE;
m_pSocket = NULL;
m_pFile = NULL;
m_pArchiveOut = NULL;
m_pArchiveIn = NULL;
}
接着为“连接”菜单项编写响应函数,代码如下:
void CChatClientDoc::OnConnect()
{
// TODO: Add your mand handler code here
CLogDlg dlg;
if(dlg。DoModal() == IDOK)
{
if(ConnectSocket(dlg。m_Name; dlg。m_Server; dlg。m_Port))
bIsConnected = TRUE;
else
AfxMessageBox(〃连接服务器失败!〃);
}
}
当连接后,应该使“连接”菜单项无效,为消息 UPDATA_MAND_UI 添加响应函
数如下:
void CChatClientDoc::OnUpdateConnect(CCmdUI* pCmdUI)
{
// TODO: Add your mand update UI handler code here
pCmdUI…》Enable(!bIsConnected);
}
·315 ·
…………………………………………………………Page 327……………………………………………………………
Visual C++ 6。0 程序设计从入门到精通
为“断开”菜单项添加响应函数,代码如下:
void CChatClientDoc::OnDisconnect()
{
// TODO: Add your mand handler code here
DeleteContents();
bIsConnected = FALSE;
}
当断开后,应当使“断开”菜单项无效,为消息 UPDATA_MAND_UI 添加如下响
应函数:
void CChatClientDoc::OnUpdateDisconnect(CCmdUI* pCmdUI)
{
// TODO: Add your mand update UI handler code here
pCmdUI…》Enable(bIsConnected);
}
为 ConnectSocket()函数编写如下代码:
BOOL CChatClientDoc::ConnectSocket(LPCTSTR lpszHandle; LPCTSTR lpszAddress; int nPort)
{
m_strName = lpszHandle;
m_pSocket = new CChatSocket(this);
//创建 socket 失败
if (!m_pSocket…》Create())
{
delete m_pSocket;
m_pSocket = NULL;
AfxMessageBox(〃创建 socket 失败!〃);
return FALSE;
}
//连接失败
if(!m_pSocket…》Connect(lpszAddress; nPort))
{
delete m_pSocket;
m_pSocket = NULL;
return FALSE;
}
m_pFile = new CSocketFile(m_pSocket);
//用于数据接收和发送
m_pArchiveIn = new CArchive(m_pFile;CArchive::load);
m_pArchiveOut = new CArchive(m_pFile;CArchive::store);
//将用户名发送给服务器
SendMsg(m_strName; SENDING_NICKNAME; false);
·316 ·
…………………………………………………………Page 328……………………………………………………………
第 11 章 网络编程
CString strTemp;
strTemp。Format(〃进入聊天室〃);
//作为普通消息发送给服务器
SendMsg(strTemp; NORMAL_MESSAGE; true);
return TRUE;
}
为 SendMsg()函数编写如下代码:
void CChatClientDoc::SendMsg(CString& strText; int mCode; BOOL bSendHandle)
{
if (m_pArchiveOut != NULL)
{
CMsg msg;
msg。code = mCode;
msg。m_strText = (bSendHandle ? m_strName + _T(〃: 〃) + strText : strText);
TRY
{
msg。Serialize(*m_pArchiveOut);
m_pArchiveOut…》Flush();
}
CATCH(CFileException; e)
{
m_pArchiveOut…》Abort();
delete m_pArchiveOut;
m_pArchiveOut = NULL;
CString strTemp;
strTemp。Format(〃发送失败〃);
DisplayMsg(strTemp);
}
END_CATCH
}
}
为 ProcessReceive() 函数编写如下代码:
void CChatClientDoc::ProcessReceive()
{
do
{
ReceiveMsg();
if (m_pSocket == NULL)
return;
·317 ·
…………………………………………………………Page 329……………………………………………………………
Visual C++ 6。0 程序设计从入门到精通
}
while(!m_pArchiveIn…》IsBufferEmpty());
}
为 ReceiveMsg() 函数编写如下代码:
void CChatClientDoc::ReceiveMsg()
{
CMsg msg;
TRY
{
msg。Serialize(*m_pArchiveIn);
if(msg。code == SENDING_CHATTERS_LIST)
{
//更新用户列表
UpdateChattersList(&msg);
return;
}
if(msg。code == USED_NAME)
{
//如果是用户名已存在信息则关闭连接,退出
AfxMessageBox(msg。m_strText);
msg。m_bClose = TRUE;
OnDisconnect();
return;
}
//显示消息
DisplayMsg(msg。m_strText);
}
CATCH(CFileException; e)
{
msg。m_bClose = TRUE;
m_pArchiveOut…》Abort();
CString strTemp;
strTemp。Format(〃接收数据失败〃);
DisplayMsg(strTemp);
}
END_CATCH
//如果连接已关闭则删除各对象
if (msg。m_bClose)
{
delete m_pArchiveIn;
·318 ·
…………………………………………………………Page 330……………………………………………………………
第 11 章 网络编程
m_pArchiveIn = NULL;
delete m_pArchiveOut;
m_pArchiveOut = NULL;
delete m_pFile;
m_pFile = NULL;
delete m_pSocket;
m_pSocket = NULL;
}
}
为 DisplayMsg() 函数编写如下代码:
void CChatClientDoc::DisplayMsg(LPCTSTR lpszText)
{
for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
{
CView* pView = GetNextView(pos);
CMessageView* pChatView = DYNAMIC_DOWNCAST(CMessageView; pView);
if (pChatView != NULL)
pChatView…》ShowMessage(lpszText);
}
}
为 UpdateChattersList() 函数编写如下代码:
void CChatClientDoc::UpdateChattersList(CMsg* pMsg)
{
CChattersView* pChattersView;
for(POSITION pos=GetFirstViewPosition();pos!=NULL;)
{
CView* pView = GetNextView(pos);
pChattersView = DYNAMIC_DOWNCAST(CChattersView; pView);
//首先清空用户列表视图
if (pChattersView != NULL)
pChattersView…》ClearChattersList();
}
CString strTemp = pMsg…》