加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
user.go 2.62 KB
一键复制 编辑 原始数据 按行查看 历史
Venns 提交于 2022-07-25 17:39 . client update username
package main
import (
"net"
"strings"
)
type User struct {
Name string
Address string
Channel chan string
conn net.Conn
server *Server
}
// 构造方法
func NewUser(conn net.Conn, server *Server) *User {
userAddr := conn.RemoteAddr().String()
user := &User{
Name: userAddr,
Address: userAddr,
Channel: make(chan string),
conn: conn,
server: server,
}
// 启动监听
go user.ListenMessage()
return user
}
// 用户上线业务
func (this *User) Online() {
// 用户上线,将用户加入到onlineMap中
this.server.mapLock.Lock()
this.server.OnlineMap[this.Name] = this
this.server.mapLock.Unlock()
// 广播用户上线
this.server.BroadCast(this, "login")
}
// 用户下线业务
func (this *User) Offline() {
// 用户下线,将用户从onlineMap中删除
this.server.mapLock.Lock()
delete(this.server.OnlineMap, this.Name)
this.server.mapLock.Unlock()
// 广播用户上线
this.server.BroadCast(this, "logout")
}
// 给当前客户端发生消息
func (this *User) SendMsg(msg string) {
this.conn.Write([]byte(msg + "\n"))
}
// 用户处理消息的业务
func (this *User) DoMessage(msg string) {
if msg == "who" {
// 查询当前用户都有哪些
this.server.mapLock.Lock()
for _, user := range this.server.OnlineMap {
onlineMsg := "[" + user.Address + "]" + user.Name + ": online"
this.SendMsg(onlineMsg)
}
} else if len(msg) > 7 && msg[:7] == "rename|" {
newName := strings.Split(msg, "|")[1]
// 判断name是否存在
_, ok := this.server.OnlineMap[newName]
if ok {
this.SendMsg(newName + " already exists")
} else {
this.server.mapLock.Lock()
delete(this.server.OnlineMap, this.Name)
this.server.OnlineMap[newName] = this
this.server.mapLock.Unlock()
this.Name = newName
this.SendMsg("Successfully modified")
}
} else if len(msg) > 4 && msg[:3] == "to|" {
// 消息格式: to|username|msg
// 1 获取对方的用户名
remoteName := strings.Split(msg, "|")[1]
if remoteName == "" {
this.SendMsg("msg format error,format:\" to|username|msg\"")
return
}
// 2 根据用户名 得到对方的user对象
remoteUser, ok := this.server.OnlineMap[remoteName]
if !ok {
this.SendMsg("user not exits")
return
}
// 3 获取消息内容,通过对方的user对象将消息内容发生过去
content := strings.Split(msg, "|")[2]
if content == "" {
this.SendMsg("msg is null")
return
}
remoteUser.SendMsg(this.Name + "said:" + content)
} else {
this.server.BroadCast(this, msg)
}
}
// 监听channel
func (this *User) ListenMessage() {
for true {
msg := <-this.Channel
this.conn.Write([]byte(msg + "\n"))
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化