go实现聊天室【简洁版】

go写了一个简洁版的聊天室实现功能
1.广播用户登录
2.广播群发消息
3.修改当前登录用户名
4.广播用户退出

服务端代码

package main

import (
	"log"
	"net"
	"strings"
)
type Client struct {
	C chan string
	Name string
	Addr string
}
var onLineMap map[string]Client
var  Message=make(chan string)
var isQuit=make(chan bool)

func WriteMsg(clnt Client, conn net.Conn){
	for msg:=range clnt.C {
		msg:=msg+"\n"
		conn.Write([]byte(msg))
	}
}

//广播所有用户消息
func tellAllUser(conn net.Conn,clnt Client){
	addr:= conn.RemoteAddr().String()
	buf:=make([]byte,1024)
	for {
		n,err:=conn.Read(buf)
		if n == 0 {
			log.Println("客户端退休")
			isQuit <-true
			return
		}
		if err != nil {
			log.Println("system is errors")
			return
		}
		content:=string(buf[:n])
		if content == "who" && len(content) == 3 {
			conn.Write([]byte("用户列表：\n"))
			for _,item:=range onLineMap {
				userInfo:= item.Addr+"："+item.Name+"\n"
				conn.Write([]byte(userInfo))
			}
		}else if len(content) >=8 && content[:6] == "rename"{
			newName:= strings.Split(content,"|")
			clnt.Name = newName[1]
			onLineMap[addr]=clnt
		}else{
			msg:="客户端消息: "+content
			Message <-msg
		}

	}
}


//处理连接出来的用户
func HandleConnect(conn net.Conn){
	netAddr:=conn.RemoteAddr().String()
	clnt:=Client{
		C:make(chan string),
		Name: netAddr,
		Addr: netAddr,
	}
	//创建专门当前用户发送消息的go协程
	onLineMap[netAddr]=clnt
	go WriteMsg(clnt,conn)
	//发送用户上校消息导全局通道
	msg:="欢迎用户："+netAddr+"登录"
	Message<-msg

	go tellAllUser(conn,clnt)//广播所有用户消息

	for {
		select {
			case <-isQuit:
			delete(onLineMap,netAddr)
			Message <-clnt.Name+"login out"
			return
		}
	}
}
func Manager(){
	onLineMap=make(map[string]Client)
	for {
		msg:=<-Message
		for _,clnt:=range onLineMap {
			clnt.C <-msg
		}
	}
}

func main(){
	listener,err:= net.Listen("tcp","127.0.0.1:8007")
	if err != nil {
		log.Println("open tcp is errors")
	}
	defer listener.Close()
	go Manager()//存储用户消息
	for {
		conn,err:=listener.Accept()
		if err != nil {
			log.Println("Accept is erros")
		}
		go HandleConnect(conn)
	}

}

客户端代码

package main

import (
	"bufio"
	"fmt"
	"io"
	"net"
	"os"
	"strings"
)
func read(conn net.Conn){
	defer conn.Close()
	var buf[1024]byte
	reader:=bufio.NewReader(conn)
	for {
		n,err:=reader.Read(buf[:])
		if err  == io.EOF {
			break
		}
		if err != nil {
			return
		}
		recvStr:= string(buf[:n])
		fmt.Printf("收到服务端数据为:%v\n",recvStr)
	}
}

func main(){
	conn,err:= net.Dial("tcp","127.0.0.1:8007")
	if err != nil {
		fmt.Println("is erro")
	}
	go read(conn)
	reader:= bufio.NewReader(os.Stdin)
	fmt.Printf("%v,\n","请写入内容")
	for {
		text,_:=reader.ReadString('\n')
		text = strings.TrimSpace(text)
		if text == "exit" {
			break
		}
		conn.Write([]byte(text))
	}
	conn.Close()
}