go 聊天系统项目-5 客户端发消息
一、前言
敬告:本文不讲解代码，只是把代码展示出来。
该代码之前的代码见
go 聊天系统项目-1
go聊天系统项目-2 redis 验证用户id和密码
go聊天系统项目-3 redis注册用户
go聊天项目4-显示用户列表
注意：本文使用 go mod 管理代码。详情见 go 包相关知识
二、为实现客户端发消息功能修改代码
修改 common message 模块 message.go 代码
const 常量增加如下字段
SmsMesType = "SmsMes"
增加 SmsMes 结构体
// 增加一个SmsMes发送的消息
type SmsMes struct {
Content string `json: "content"`
User //匿名结构体，继承
}
// SmsReMes
client model 模块增加 curUser.go 代码
package model
import (
"net"
"redis/day8/chatroom/common/message"
)
type CurUser struct {
Conn net.Conn
message.User
}
修改 client process 模块 userMgr.go 代码
增加导入
import (
"redis/day8/chatroom/common/message"
)
增加全局变量
var CurUser model.CurUser //我们在用户登录成功后，完成对CurUser初始化
修改common message 模块 smsProcess.go 代码
在 User 机构体中增加如下字段
Sex string `json:"sex"` //性别
如下
type User struct {
UserId int `json:"userId"`
UserPwd string `json:"userPwd"`
UserName string `json:"userName"`
UserStatus int `json:"userStatus"` //用户状态
Sex string `json:"sex"` //性别
}
修改 client process 模块 userProcess.go 代码
在 Login 方法的
var loginResMes message.LoginResMes
err = json.Unmarshal([]byte(mes.Data), &loginResMes)
if loginResMes.Code == 200 {
代码下增加如下代码
//初始化CurUser
CurUser.Conn = conn
CurUser.UserId = userId
CurUser.UserStatus = message.UserOnline
修改 client process 模块 增加 smsProcess.go 代码
package process
import (
"encoding/json"
"fmt"
"redis/day8/chatroom/client/utils"
"redis/day8/chatroom/common/message"
)
type SmsProcess struct {
}
// 发送群聊的消息
func (this *SmsProcess) SendGroupMes(content string) (err error) {
//1 创建一个 Mes
var mes message.Message
mes.Type = message.SmsMesType
//2 创建一个SmsMes实例
var smsMes message.SmsMes
smsMes.Content = content //内容
smsMes.UserId = CurUser.UserId
smsMes.UserStatus = CurUser.UserStatus
//3.序列化 smsMes
data, err := json.Marshal(smsMes)
if err != nil {
fmt.Println("SendGroupMes json.Marshal fail =", err.Error())
return
}
mes.Data = string(data)
//4.对mes再次序列化
data, err = json.Marshal(mes)
if err != nil {
fmt.Println("SendGroupMes json.Marshal fail =", err.Error())
return
}
//5. 将mes发送给服务器
tf := &utils.Transfer{
Conn: CurUser.Conn,
}
//6. 发送
err = tf.WritePkg(data)
if err != nil {
fmt.Println("SendGroupMes err=", err.Error())
return
}
return
}
修改client process 模块 server.go 代码
在 ShowMenu 函数代码
fmt.Println("----------恭喜xxx登录成功--------")
fmt.Println("--------1、显示在线用户列表--------")
fmt.Println("--------2、发送消息--------")
fmt.Println("--------3、信息列表--------")
fmt.Println("--------4、退出系统--------")
var key int
var content string
下增加如下代码
//因为我们总会使用到SmsProcess实例，因此我们将其定义在swtich外部
smsProcess := &SmsProcess{}
将 switch key{case 2:} 中代码修改为如下
fmt.Println("请输入你想对大家说点什么:")
fmt.Scanf("%s\n", &content)
smsProcess.SendGroupMes(content)
修改 server processor 模块 processor.go 代码
在代码
func (this *Processor) serverProcessMes(mes *message.Message) (err error) {
下增加如下代码
//看看是否能接收到客户端发送的群发的消息
fmt.Println("mes=", mes)
测试一下
服务端
./main
服务器[新的结构]在 8889 端口监听......
等待客户端连接服务器......
等待客户端连接服务器......
mes= &{LoginMes {"UserId":100,"UserPwd":"123456","UserName":""}}
&{100 123456 tom 0 } 登录成功
客户端
./client
---------------欢迎登录多人聊天系统-------------------
1、登录聊天室
2、注册用户
3、退出系统
请选择(1-3):
1
登录聊天室
请输入用户的id
100
请输入用户密码
123456
客户端，发送消息的长度=86,消息内容为: {"Type":"LoginMes","Data":"{\"UserId\":100,\"UserPwd\":\"123456\",\"UserName\":\"\"}"}
当前在线用户列表如下:
----------恭喜xxx登录成功--------
--------1、显示在线用户列表--------
--------2、发送消息--------
--------3、信息列表--------
--------4、退出系统--------
客户端%s正在等待读取服务器发送的消息
客户端
2
请输入你想对大家说点什么:
hello,body!
----------恭喜xxx登录成功--------
--------1、显示在线用户列表--------
--------2、发送消息--------
--------3、信息列表--------
--------4、退出系统--------
服务端
mes= &{SmsMes {"Content":"hello,body!","userId":100,"userPwd":"","userName":"","userStatus":0,"sex":""}}
消息类型不存在，无法处理