并发聊天服务器

package mainimport ("fmt""net""strings""time"
)// 结构体
type Client struct {C    chan string //用户发送数据的管道Name string      //用户名Addr string      //网络地址
}// 保存在线用户 cliAddr -->client
var onlineMap map[string]Client// 创建一个通信的管道
var message = make(chan string)func MakeMsg(cli Client, msg string) (buf string) {buf = "[" + cli.Addr + "]" + cli.Name + ": " + msgreturn
}// WriteMsgToClient 专门给当前客户端发送信息
func WriteMsgToClient(cli Client, conn net.Conn) {for msg := range cli.C { //给当前客户端发送信息_, err := conn.Write([]byte(msg + "\n"))if err != nil {fmt.Println("conn write err =", err)return}}
}// HandleConn //处理用户链接
func HandleConn(conn net.Conn) {defer conn.Close()//获取客户端的网络地址cliAddr := conn.RemoteAddr().String()//创建一个结构体 创建一个管道 默认用户名和网络地址一样cli := Client{make(chan string), cliAddr, cliAddr}//把结构体添加到maponlineMap[cliAddr] = cli//新开一个协程，专门给当前客户端发送信息go WriteMsgToClient(cli, conn)//广播某个人在线//message <- "[" + cli.Addr + "]" + cli.Name + ": login"message <- MakeMsg(cli, "login")//提示，我是谁cli.C <- MakeMsg(cli, "I am hello")isQuit := make(chan bool) //对方是否主动退出isData := make(chan bool) //对方是否有数据发送//新建一个协程，接收用户发送过来的数据go func() {buf := make([]byte, 2048)for {readSize, err := conn.Read(buf)//if err != nil {//	if err == io.EOF {//		fmt.Println("写入完毕")//	}//	return//}if readSize == 0 { //对象断开，或者出问题isQuit <- truefmt.Println("conn read err = ", err)return}msg := string(buf[:readSize-1])if len(msg) == 3 && msg == "who" { //查询在线用户//遍历map,get当前用户发送所有成员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" {//修改用户名name := strings.Split(msg, "|")[1]cli.Name = nameonlineMap[cliAddr] = cliconn.Write([]byte("rename ok\n"))} else {//转发此内容fmt.Println("成朋")message <- MakeMsg(cli, msg)}isData <- true //代表有数据}}()for true {//通过select检测channel的流动select {case <-isQuit:delete(onlineMap, cliAddr)           //当前用户从map移除message <- MakeMsg(cli, "login out") //广播谁下线了returncase <-isData:case <-time.After(30 * time.Second):delete(onlineMap, cliAddr)message <- MakeMsg(cli, "time out leave out") //超时了return}}}// Manager 新开一个协程 转发消息,只要有消息来了，遍历map，给map每个成员都发送消息
func Manager() {//给map分配空间onlineMap = make(map[string]Client)for true {meg := <-message //没有消息前，这里会阻塞//遍历map，给map每个成员都发送消息for _, client := range onlineMap {client.C <- meg //把消息给这个管道}}
}
func main() {//监听listener, err := net.Listen("tcp", ":8000")if err != nil {fmt.Println("listener err = ", err)return}defer listener.Close()//新开一个协程 转发消息,只要有消息来了，遍历map，给map每个成员都发送消息go Manager()//主协程。循环阻塞等待用户连接for true {conn, err := listener.Accept() //建立连接if err != nil {fmt.Println("conn err = ", err)continue}go HandleConn(conn) //处理用户链接}
}

调试工具netcat