go语言聊天室服务器

package main

import (
    "fmt"
    "net"
    "strings"
    "time"
)

type Client struct {
    C    chan string // 用户发送数据的管道
    Name string      // 用户名
    Addr string      // 网络地址
}

var onlineMap map[string]Client

var messaage = make(chan string)

// 新开一个协程，转发消息，只要消息来了，遍历map给每个成员都发送此消息
func Manager() {
    // 给MAP分配空间
    onlineMap = make(map[string]Client)
    for {
        msg := <-messaage // 没有消息这里阻塞
        //遍历map给每个成员都发送此消息
        for _, cli := range onlineMap {
            cli.C <- msg
        }
    }
}
func WriteMsgToClient(cli Client, conn net.Conn) {
    for msg := range cli.C { //当前客户发送消息
        conn.Write([]byte(msg + "\n"))
    }
}

func MakeMsg(cli Client, msg string) (buf string) {
    buf = "[" + cli.Addr + "]" + cli.Name + ": " + msg
    return
}
func HandleConn(conn net.Conn) { // 处理用户连接
    defer conn.Close()
    // 获取客户端的网络地址
    cliAddr := conn.RemoteAddr().String()

    // 创建一个结构体
    cli := Client{make(chan string), cliAddr, cliAddr}
    // 把结构体添加到map
    onlineMap[cliAddr] = cli
    // 新开一协程，专门给当前客户发送消息
    go WriteMsgToClient(cli, conn)
    // 广播某个在线
    //messaage <- "[" + cli.Addr + "]" + cli.Name + ": login"
    messaage <- MakeMsg(cli, "login")

    //提示我是谁
    cli.C <- MakeMsg(cli, "I am here")

    isQuit := make(chan bool) //对方是否是主动退出

    hasData := make(chan bool) // 对方是否有数局发送

    go func() {
        buf := make([]byte, 2048)
        for {
            n, err := conn.Read(buf)
            if n == 0 { //对方断开或者出问题
                isQuit <- true
                fmt.Println("conn.Read err = ", err)
                return
            }
            msg := string(buf[:n-1]) //通过windows nc 测试，多一个换行
            if len(msg) == 3 && msg == "who" {
                //遍历map，给当前用户发送所有成员
                conn.Write([]byte("user list:\n"))
                for _, tmp := range onlineMap {
                    msg = tmp.Addr + ":" + tmp.Name + "\n"
                    conn.Write([]byte(msg))
                }
            } else if len(msg) >= 8 && msg[:6] == "rename" {
                //rename mike
                name := strings.Split(msg, ",")[1]
                cli.Name = name
                onlineMap[cliAddr] = cli
                conn.Write([]byte("rename ok"))
            } else {

                //转发此内容
                messaage <- MakeMsg(cli, msg)
            }
            hasData <- true

        }
    }()

    for {
        //通过select检测channel的流动
        select {
        case <-isQuit:
            delete(onlineMap, cliAddr)            //当前用户从map移除
            messaage <- MakeMsg(cli, "login out") //广播谁下线了
            return
        case <-hasData:
        case <-time.After(10 * time.Second): //60s
            delete(onlineMap, cliAddr)                     //当前用户从map移除
            messaage <- MakeMsg(cli, "time out leave out") //广播谁下线了
            return
        }
    }
}
func main() {
    //监听端口
    listenner, err := net.Listen("tcp", ":8000")//也可自己制定IP地址
    if err != nil {
        fmt.Println("net.Listen err = ", err)
        return
    }

    defer listenner.Close()

    // 新开一个协程，转发消息，只要消息来了，变量MAp给每个成员都发送此消息
    go Manager()
    // 主协程，循环阻塞等待用户连接
    for {
        conn, err := listenner.Accept()
        if err != nil {
            fmt.Println("listenner.Accept err = ", err)
            continue
        }
        go HandleConn(conn) // 处理用户连接
    }
}

使用nc连接测试