并发聊天服务器编写

并发聊天服务器

package main

import (
	"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 + ": " + msg
	return
}

// 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}
	//把结构体添加到map
	onlineMap[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 <- true
				fmt.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 = name
				onlineMap[cliAddr] = cli
				conn.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") //广播谁下线了
			return
		case <-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