聊天室(C++客户端+Pyhton服务器)2.基本功能添加

2020-12-13 02:18

阅读:478

标签:文件句柄   cli   ber   文件   父窗口   ppa   inline   mode   recv   

根据之前的框架添加新的功能

登录

点击相关按钮

// 登录按钮的响应
void CMainDialog::OnBnClickedLogin()
{
// 1. 获取用户输入的数据
UpdateData(TRUE);
?
// 2. 判断输入的数据是否为空
if (m_UserName.IsEmpty() || m_PassWord.IsEmpty())
{
MessageBox(L"请输入用户名和密码");
return;
}
?
// 3. 填充需要发送的数据内容
SEND_INFO SendInfo = { TYPE::LOGIN, GetSafeHwnd() };
memcpy(SendInfo.LoginInfo.UserName, m_UserName.GetBuffer(), m_UserName.GetLength() * 2);
memcpy(SendInfo.LoginInfo.PassWord, m_PassWord.GetBuffer(), m_PassWord.GetLength() * 2);
?
// 4. 将数据通过自定义消息发送到服务器
SendMessage(UM_SEND_MSG, (WPARAM)& SendInfo, sizeof(SEND_INFO));
}

发送的结构体

// 1. 登录使用的的结构体
using LOGIN_INFO = _REGISTER_INFO;
using PLOGIN_INFO = _REGISTER_INFO*;

 

python接收到后处理的方法

# 实例方式:用于处理登录消息
   def on_login(self, client, message):
       # 解包获取发送来的数据
       hwnd, username, password = struct.unpack("i64s64s", message[:132])
       # 将用户名和密码进行解码
       username = username.decode("UTF16").strip("\x00")
       password = password.decode("UTF16").strip("\x00")
       # 判断用户和密码是否存在于数据库中
       if self.mysql.select("SELECT * FROM user_table WHERE user=‘%s‘ AND pswd=MD5(‘%s‘);" %(username, password)):
           # 如果用户名和密码匹配,返回结果 => 类型 + 句柄 + 是否成功 + 服务器返回的信息
           client.send(struct.pack("iii40s", Type.LOGIN.value, hwnd, 1, "登录成功".encode("UTF16")))
           # 将登录成功的用户添加到【在线用户】字典中
           print(username, "登录了聊天室")
           self.dict_client[username] = client
       else:
           # 失败的信息
           client.send(struct.pack("iii40s", Type.LOGIN.value, hwnd, 0, "用户名或密码不匹配".encode("UTF16")))
?
 

客户端接接受到服务端发来的消息做的反应

主窗口响应,转发

// 响应登录
case TYPE::LOGIN:
{
// 对于注册是否成功的消息,wParam 没有用,lParam 指向 RECV_STATE
::SendMessage(RecvInfo->hWnd, UM_RECV_LOGIN, NULL, (LPARAM)& RecvInfo->RecvState);
break;
}

转发到的地方响应

// 是否登录成功
afx_msg LRESULT CMainDialog::OnUmRecvLogin(WPARAM wParam, LPARAM lParam)
{
// 将 lParam 转换成对应的的结构
PRECV_STATE RecvState = (PRECV_STATE)lParam;
?
// 判断是否登录成功
if (RecvState->IsSuccess)
{
// 弹出用户话框(好友、群组)
CUserDialog* dialog = new CUserDialog;
dialog->Create(IDD_USERDIALOG, this);
// [ 对话框必须是非模态的,否则会导致消息的阻塞 ]
// [ 对话框必须是保存在堆空间中,否则离开作用域会销毁 ]
return dialog->ShowWindow(SW_SHOWNORMAL);
}
?
// 不成功执行这里
MessageBox(RecvState->MsgInfo);
}

 

 

登录之后进到一个新的界面

获取父窗口句柄

//构造函数内设置主窗口
CUserDialog::CUserDialog(CWnd* pParent /*=nullptr*/)
: CDialogEx(IDD_USERDIALOG, pParent)
, m_AddEdit(_T(""))
{
// 保存父窗口
MainDialog = (CMainDialog*)AfxGetMainWnd();
}
?

 

设置标题

初始化树控件

// 初始化对话框(获取用户和群组)
BOOL CUserDialog::OnInitDialog()
{
CDialogEx::OnInitDialog();
?
// 隐藏父窗口
MainDialog->ShowWindow(SW_HIDE);
SetWindowText(MainDialog->m_UserName + L"主窗口");
?
// 添加好友分组和群组的节点
m_FrdNode = m_TreeCtrl.InsertItem(L"好友列表");
m_GrpNode = m_TreeCtrl.InsertItem(L"群组列表");
?
return TRUE;
}

 

 

 

C++客户端添加好友功能

客户端向服务器发送要添加的消息

获取要添加的名字

打包发送给服务器

/ 添加好友按钮的响应
void CUserDialog::OnBnClickedAddfrd()
{
// 1. 获取自己的名字和目标名称
UpdateData(TRUE);
CString MyName = MainDialog->m_UserName;
?
// 2. 自己不能添加自己为好友
if (MyName == m_AddEdit)
{
MessageBox(L"无法添加自己为好友!");
return;
}
?
// 3. 组合消息
SEND_INFO SendInfo = { TYPE::ADDFRD, GetSafeHwnd() };
memcpy(SendInfo.AddFrdInfo.User1, MyName.GetBuffer(), MyName.GetLength() * 2);
memcpy(SendInfo.AddFrdInfo.User2, m_AddEdit.GetBuffer(), m_AddEdit.GetLength() * 2);
?
// 4. 通过主窗口发送消息      
::SendMessage(MainDialog->GetSafeHwnd(), UM_SEND_MSG, (WPARAM)& SendInfo, sizeof(SendInfo));
}

 

打包前得先弄信息结构体

// 2. 添加好友的消息
typedef struct _ADDFRD_INFO
{
WCHAR User1[32];
WCHAR User2[32];
} ADDFRD_INFO, * PADDFRD_INFO;
?

 

 

服务端接收到消息之后的处理

添加类型

添加方法

定义方法

添加新的表单friend_table

user1

user2

发送回客户端

 

# 添加好友的消息
   def on_addfrd(self, client, message):
       # 解包获取发送来的数据
       hwnd, user1, user2 = struct.unpack("i64s64s", message[:132])
       # 获取双方的用户名
       user1 = user1.decode("UTF16").strip("\x00")
       user2 = user2.decode("UTF16").strip("\x00")
       # 是否有这样一个用户
       if not self.mysql.select("SELECT * FROM user_table WHERE user=‘%s‘;" % user2):
           client.send(struct.pack("iii40s", Type.ADDFRD.value, hwnd, 0, "用户不存在".encode("UTF16")))
       # 是否已经是我的好友
       elif self.mysql.select("SELECT * FROM friend_table WHERE user1=‘%s‘ AND user2=‘%s‘;" % (user1, user2)):
           client.send(struct.pack("iii40s", Type.ADDFRD.value, hwnd, 0, "不能重复添加好友".encode("UTF16")))
       # 添加成功的情况
       else:
           # 向数据库中添加一组信息 (我,好友)
           self.mysql.exec("INSERT INTO friend_table VALUE(‘%s‘, ‘%s‘);" % (user1, user2))
           # 在返回的消息处填写添加的好友的名称
           client.send(struct.pack("iii40s", Type.ADDFRD.value, hwnd, 1, user2.encode("UTF16")))
?
 
客户端接收到服务器的反馈的反应

要用非模态的对话框,否则信息会堵塞

 

主窗口响应,转发

// 响应加好友
case TYPE::ADDFRD:
{
// 对于注册是否成功的消息,wParam 没有用,lParam 指向 RECV_STATE
::SendMessage(RecvInfo->hWnd, UM_RECV_ADDFRD, NULL, (LPARAM)& RecvInfo->RecvState);
break;
}

转发到的地方

// 查看是否成功
afx_msg LRESULT CUserDialog::OnUmRecvAddfrd(WPARAM wParam, LPARAM lParam)
{
// 将 lParam 转换成对应的的结构
PRECV_STATE RecvState = (PRECV_STATE)lParam;
?
// 判断是否添加成功
if (RecvState->IsSuccess)
{
// 如果成功执行直接添加到列表
m_TreeCtrl.InsertItem(RecvState->MsgInfo, m_FrdNode);
return 0;
}
?
// 不成功执行这里
MessageBox(RecvState->MsgInfo);
return 0;
}

 

更新好友列表

打开好友的时候会自动更新好友列表

初始化的时候发送请求

// 发消息给服务器更新好友列表
SEND_INFO SendInfo = { TYPE::UPDATEFRD, GetSafeHwnd() };
memcpy(SendInfo.UpdateFrdInfo.User, MainDialog->m_UserName.GetBuffer(),
MainDialog->m_UserName.GetLength() * 2);
?
// 通过主窗口向服务器发送消息
::SendMessage(MainDialog->GetSafeHwnd(), UM_SEND_MSG, (WPARAM)& SendInfo, sizeof(SendInfo));
?

相关的结构体

// 3. 更新好友列表
typedef struct _UPDATEFRD_INFO
{
WCHAR User[32];
} UPDATEFRD_INFO, * PUPDATEFRD_INFO;
?

 

服务器接收到处理之后发回来

# 更新好友的消息
   def on_updatefrd(self, client, message):
       # 解包获取发送来的数据
       hwnd, user = struct.unpack("i64s", message[:68])
       # 获取目标用户名
       user = user.decode("UTF16").strip("\x00")
       # 查询目标用户的所有好友名称
       friends = self.mysql.select("SELECT user2 FROM friend_table WHERE user1=‘%s‘;" % user)
       # 通过一个循环,将每进一个用户名发送到客户端
       for name in friends:   # (("1"),("2"),("3"))
           client.send(struct.pack("ii64s", Type.UPDATEFRD.value, hwnd, name[0].encode("UTF16")))
           time.sleep(0.1)
?
 

 

客户端接受到消息之后处理

主窗口响应,转发

// 响应更新好友
case TYPE::UPDATEFRD:
{
// 对于注册是否成功的消息,wParam 没有用,lParam 指向 RECV_STATE
::SendMessage(RecvInfo->hWnd, UM_RECV_UPDATEFRD, NULL, (LPARAM)& RecvInfo->RecvState);
break;
}

转发到的地方

// 响应更新好友的消息
afx_msg LRESULT CUserDialog::OnUmRecvUpdatefrd(WPARAM wParam, LPARAM lParam)
{
// 转换成对应的结构体
PUPDATEFRD_INFO UserInfo = (PUPDATEFRD_INFO)lParam;
?
// 【添加到树控件的好友节点中,收到的消息前两个字节会是 ff fe】
m_TreeCtrl.InsertItem(&UserInfo->User[1], m_FrdNode);
?
return 0;
}

 

 

 

好友私聊

私聊窗口的界面

客户端的话,双击树控件的好友的时候要能弹出聊天窗口,

所以需要响应树控件按钮

 

字典#include

using namespace std:

判断是不是好友窗口,群窗口同理

窗口可以双击打开,

// 响应窗口聊天
void CUserDialog::OnDblclkTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
// 1. 获取当前点击的这一个子节点的内容
HTREEITEM CurrentNode = m_TreeCtrl.GetSelectedItem();
CString Name = m_TreeCtrl.GetItemText(CurrentNode);
?
// 2. 判断节点是不是[好友列表]或者[群组列表]
if (Name.Compare(L"好友列表") && Name.Compare(L"群组列表"))
{
// 3.1 判断窗口是否已经存在
if (MainDialog->WindowsMap.find(Name) != MainDialog->WindowsMap.end())
{
// 存在则显示,确保一个窗口只会被打开一次
MainDialog->WindowsMap[Name]->ShowWindow(SW_SHOWNORMAL);
}
// 3.2 判断是不是好友聊天窗口
else if (m_TreeCtrl.GetParentItem(CurrentNode) == m_FrdNode)
{
// 3.2.1 创建用户聊天窗口
CFrdDialog* dialog = new CFrdDialog(Name);
dialog->Create(IDD_DIALOG2, this);
// 3.2.2 设置用户聊天窗口的窗口名为点击的文字
dialog->SetWindowText(Name);
dialog->ShowWindow(SW_SHOWNORMAL);
// 3.2.3 将新创建的窗口添加到字典中
MainDialog->WindowsMap[Name] = dialog;
}
// 3.3 判断是不是群组聊天窗口
else if (m_TreeCtrl.GetParentItem(CurrentNode) == m_GrpNode)
{
// 3.3.1 创建群组聊天窗口
CGrpDialog* dialog = new CGrpDialog(Name);
dialog->Create(IDD_DIALOG3, this);
// 3.3.2 设置群组聊天窗口的窗口名为点击的文字
dialog->SetWindowText(Name);
dialog->ShowWindow(SW_SHOWNORMAL);
// 3.3.3 将新创建的窗口添加到字典中
MainDialog->WindowsMap[Name] = dialog;


评论


亲,登录后才可以留言!