接下来我们来演示如何实现单聊,既然群聊可以,那么单聊就是在于如何区分用户,然后找到用户发送消息即可.
package main /* * 单聊--一对一聊天 */ import ( "bufio" "encoding/json" "fmt" "math/rand" "net" "strings" "sync" "time" ) //保存连接 var connList sync.Map //消息结构{"from":"aaa","to":"bbb","body":"aaa","cmd":"sendMsg"} type UserMsg struct { From string `json:"from"` To string `json:"to"` Body string `json:"body"` Cmd string `json:"cmd"` } //随机字符串 func randName() string { var letterRunes = []rune("123456789") rand.Seed(time.Now().UnixNano()) b := make([]rune, 6) for i := range b { b[i] = letterRunes[rand.Intn(len(letterRunes))] } return string(b) } func DealConn(conn net.Conn) { // 处理完关闭连接 defer conn.Close() fmt.Println("new conn\n") name := randName() connList.Store(name, conn) conn.Write([]byte("you is : " + name + "\r\n")) // 针对当前连接做发送和接受操作 for { reader := bufio.NewReader(conn) // 读取字符串, 直到碰到回车返回 str, err := reader.ReadString('\n') // 数据读取正确 if err == nil { // 去掉字符串尾部的回车 str = strings.TrimSpace(str) var strStruct UserMsg err := json.Unmarshal([]byte(str), &strStruct) if err != nil { fmt.Printf("msg json error, err:%v\n", err) } else { to := strStruct.To cmd := strStruct.Cmd if cmd == "sendMsg" { toUser, ok := connList.Load(to) if !ok { fmt.Println("to user is not\n") } else { toUserConn, ok := toUser.(net.Conn) if !ok { fmt.Println("to user type wrong\n") } else { toUserConn.Write([]byte(strStruct.From + " say : " + strStruct.Body + "\r\n")) } } } else { fmt.Println("unkonw msg") } } } else { //删除保存的连接 connList.Delete(name) fmt.Println("conn close\n") break } } } func main() { // 建立tcp 服务 listen, err := net.Listen("tcp", "0.0.0.0:1215") if err != nil { fmt.Printf("listen failed, err:%v\n", err) return } for { // 等待客户端建立连接 fmt.Printf("accept conn ...... \n") conn, err := listen.Accept() if err != nil { fmt.Printf("accept failed, err:%v\n", err) continue } // 启动一个单独的 goroutine 去处理连接 go DealConn(conn) } }
这个例子是通过json协议来标识消息来源于那里,要发送给谁,消息类型是什么,消息内容是什么来实现单聊的.