1 user.go

  1. // 用户处理消息的业务
  2. func (this *User) DoMessage(msg string) {
  3.     if msg == "who" {
  4.         // 查询当前在线用户
  5.         this.server.mapLock.Lock()
  6.         for userName, _ := range this.server.OnlineMap {
  7.             this.SendMsg(userName + "在线\n")
  8.         }
  9.         this.server.mapLock.Unlock()
  10.     } else if len(msg) > 7 && msg[:7] == "rename|" {
  11.         // 更改自己的用户名, 消息格式: rename|张三
  12.         newName := strings.Split(msg, "|")[1]
  13.         // 判断name是否存在
  14.         _, err := this.server.OnlineMap[newName]
  15.         if err {
  16.             this.SendMsg("失败, 该用户名已被其它人使用\n")
  17.         } else {
  18.             this.server.mapLock.Lock()
  19.             delete(this.server.OnlineMap, this.Name)
  20.             this.server.OnlineMap[newName] = this
  21.             this.server.mapLock.Unlock()
  23.             this.Name = newName
  24.             this.SendMsg("您已成功更新为用户名:" + this.Name + "\n")
  25.         }
  26.     } else if len(msg) > 4 && msg[:3] == "to|" {
  27.         // 私聊, 消息格式: to|用户名|消息内容
  28.         arr := strings.Split(msg, "|")
  29.         // 获取对方的用户名
  30.         remoteName := arr[1]
  31.         if remoteName == "" {
  32.             this.SendMsg("消息格式不正确, to|用户名|消息内容")
  33.             return
  34.         }
  35.         // 根据用户名, 获取对方的User对象
  36.         remoteUser, ok := this.server.OnlineMap[remoteName]
  37.         if !ok {
  38.             this.SendMsg("该用户名不存在")
  39.             return
  40.         }
  41.         // 获取消息内容
  42.         content := arr[2]
  43.         if content == "" {
  44.             this.SendMsg("无消息内容, 请重发")
  45.             return
  46.         }
  47.         remoteUser.SendMsg(this.Name + ":" + content)
  48.     } else {
  49.         this.server.BroadCast(this, msg)
  50.     }
  51. }