1聊天室服务器端

package mainimport (    "fmt"    "net"    "strings"    "time")//定义的此结构体为全局map的value值,包括每一个用户的姓名,ip地址和私人管道type client struct {    name string    addr string    C    chan string}/*这个函数是将私人管道中的内容发送给用户,配合全局管道Message使用可以实现广播的功能,单独使用可以实现私聊的功能*/func writemsg2client(clinet client, conn net.Conn) {    for m := range clinet.C {        conn.Write([]byte(m + "\n"))    }}//这只是一个封装好用来统一(发送信息格式)的小函数,不用在意func makemsg(name string, addr string, s string) string {    return "[" + addr + "]" + name + ":" + s}//每一个进入聊天室的用户都将启动一个handleconn的go程来处理事件func handleconn(conn net.Conn) {    defer conn.Close()    /*用户连接进来以后要初始化全局map,把自己的信息加入到字典里,相当于进到聊天室里之前要登      记一下个人信息,注意姓名初始为ip地址。*/    addr := conn.RemoteAddr().String()    fmt.Printf("用户%s进入了房间\n", addr)    client := client{addr, addr, make(chan string)}    //在这里启动子go程,功能上面已经提及,具体就是会写信息给自己连接的客户端    go writemsg2client(client, conn)    onlinemap[addr] = client    //登录进来一切准备就绪后就给所有人广播上线信息啦    Message <- makemsg(client.name, addr, "login")    //下面这三个变量服务于下面一些小功能    var haschat = make(chan bool)    var ifquit = make(chan bool)    var flag bool    //从这单独开启一个go程来读取用户输入的信息    go func() {        buf := make([]byte, 4096)        for {            n, _ := conn.Read(buf)            if n == 0 {                fmt.Printf("%s离开了房间\n", client.name)                ifquit <- true                return            }            //改名功能的实现            if string(buf[:7]) == "Rename|" {                client.name = strings.Split(string(buf[:n]), "|")[1]                onlinemap[addr] = client                conn.Write([]byte("rename success\n"))            } else if string(buf[:n-1]) == "/who" {                //查询在线用户信息的功能                for _, s := range onlinemap {                    conn.Write([]byte(s.name + "online\n"))                }            } else if string(buf[:2]) == "m|" && strings.Count(string(buf[:n]), "|") == 2 {                /*私聊功能的实现,其实私聊功能就是跳过了往全局Message里传输信息,                  改为直接向私人管道里传输信息*/                flag = true                slice := strings.Split(string(buf[:n]), "|")                fmt.Println(slice[1])                for _, a := range onlinemap {                    //遍历所有在线用户,向指定的用户管道中发送信息                    if a.name == slice[1] {                        flag = false                        a.C <- makemsg(client.name, addr, slice[2])                        conn.Write([]byte("send success"))                    }                }                if flag {                    conn.Write([]byte("no such man or not online"))                }            } else {                Message <- makemsg(client.name, addr, string(buf[:n]))            }            haschat <- true        }    }()    for {        select {        case <-haschat:        //超时强踢        case <-time.After(time.Minute * 100):            delete(onlinemap, addr)            Message <- makemsg(client.name, addr, "out time to leave")            close(client.C)            return        case <-ifquit:            //退出处理            delete(onlinemap, addr)            Message <- makemsg(client.name, addr, "out time to leave")            close(client.C)            return        }    }}//这个函数用来将全局Message中的内容全部塞到私人管道C里,实现上下线广播和群聊的功能func Manager() {    for {        msg := <-Message        for _, s := range onlinemap {            s.C <- msg        }    }}var Message = make(chan string)var onlinemap map[string]client = make(map[string]client)//主函数func main() {    listener, _ := net.Listen("tcp", "127.0.0.1:9876")    defer listener.Close()    //提前开启全局Message的go程,防止被阻塞    go Manager()    for {        conn, err := listener.Accept()        if err != nil {            fmt.Println("accept err", err)            continue        }        //每一个连接进来的用户都会被分配进入一个子go程,用来处理上面我们提到的各种功能        go handleconn(conn)    }}/*备注1、  listener, _ := net.Listen("tcp", "127.0.0.1:9876")监听启动2、  go Manager()开启全局Message的go程,防止被阻塞,没有消息便被阻塞,有消息便会被唤起,消息发送完毕后重新等待消息,有消息变发送没消息便阻塞等待(Message 是一个字符串channel )。接收到消息后,遍历所有在线人员,并把消息发送给client的私人通道。func Manager() {    for {        msg := <-Message        for _, s := range onlinemap {            s.C <- msg        }    }}3、私人通道消息处理这个函数是将私人管道中的内容发送给用户,配合全局管道Message使用可以实现广播的功能。单独使用可以实现私聊的功能(m|客户端连接ip加端口|发送消息)(m|127.0.0.1:59700|hello)。这个函数也是等待消息,收到消息后被唤醒执行,消息执行完毕后等待新消息,没有阻塞,有就处理func writemsg2client(clinet client, conn net.Conn) {    for m := range clinet.C {        conn.Write([]byte(m + "\n"))    }}*/

2、聊天室客户端

package mainimport (    "bufio"    "fmt"    "net"    "os"    "strings")func readFromServer(conn net.Conn) {    buf := make([]byte, 4096)    for {        n, err := conn.Read(buf)        if err != nil {            fmt.Println(err)            os.Exit(1)        }        defer conn.Close()        fmt.Println("接收到消息:", string(buf[:n]))        fmt.Println("请输入要发送的消息:")    }}func main() {    conn, err := net.Dial("tcp", "127.0.0.1:9876")    if err != nil {        fmt.Println(err)        return    }    defer conn.Close()    go readFromServer(conn)    //fmt.Println("请输入要发送的消息:")    for {        //strs :=""        // fmt.Scanln(&strs) 空格有问题        //strs := make([]byte, 4096)        //n, err := os.Stdin.Read(strs)        str, err := bufio.NewReader(os.Stdin).ReadString('\n')        if err != nil {            fmt.Println(err)        }        str = strings.TrimSpace(str)        //fmt.Println("发送前", , "展示")        //fmt.Println("a", str, "b")        if str == "Q" {            fmt.Println("接收到退出命令,退出客户端")            break        }        conn.Write([]byte(str))    }}