Go语言编写简单聊天室
package main
import (
"fmt"
"net"
"strings"
"time"
)
type Client struct {
c chan string
Name string
Addr string
}
//创建全局map,存储在线用户
var onlineMap map[string]Client
//创建全局channel,传递用户消息
var message = make(chan string)
func MakeMsg(client Client,msg string) (buf string){
buf = "[" + client.Addr + "]" + client.Name + ": " + msg
return
}
func WriteMsgToClient(conn net.Conn,client Client) {
//监听用户的channel是的有消息
for msg := range client.c {
_, err := conn.Write([]byte(msg + "\n"))
if err != nil {
fmt.Println("Write err------>",err)
continue
}
}
}
func HandlerConn(conn net.Conn) {
defer conn.Close()
//创建channel,判断用户是否活跃
hasData := make(chan bool)
//获取用户addr, ip+port
addr := conn.RemoteAddr().String()
//创建用户
client := Client{make(chan string),addr,addr}
//将新连接用户添加到在线用户的map中
onlineMap[addr] = client
//创建用来给当前用户发送消息的go程
go WriteMsgToClient(conn,client)
//发送用户上线消息到全局channel中
//message <- "[" + addr + "]" + client.Name + " login"
message <- MakeMsg(client,"login")
//创建一个channel,判断用户退出状态
quit := make(chan bool)
//创建一个匿名go程,专门处理用户发送的消息
go func() {
buf := make([]byte, 4096)
for {
n, err := conn.Read(buf)
if n == 0 {
quit <- true
fmt.Printf("检测到用户%s退出\n",client.Name)
return
}
if err != nil {
fmt.Println("conn Read Err------>",err)
return
}
msg := string(buf[:n-1])
//获取在线用户列表
if msg == "who" && len(msg) == 3 {
_,_ = conn.Write([]byte("online user list:\n"))
for _,user := range onlineMap {
userInfo := user.Addr + ":" + user.Name + "\n"
_,_ = conn.Write([]byte(userInfo))
}
//判断用户修改用户名命令
} else if len(msg) >= 8 && msg[:6] == "rename"{
newname := strings.Split(msg,"|")[1]
client.Name = newname //修改用户名
onlineMap[addr] = client//更新用户列表
conn.Write([]byte("rename successeful!\n"))
}else {
//将读到的用户消息广播给所有在线用户
message <- MakeMsg(client,msg)
}
hasData <- true
}
}()
for {
select {
case <- quit:
delete(onlineMap,client.Addr)//将用户从列表中删除
message <- MakeMsg(client,"logout")//写入用户退出消息到全局channel
return
case <- hasData:
//目的是重置下面的case的计时器
case <- time.After(time.Second * 10):
delete(onlineMap,client.Addr)//将用户从列表中删除
message <- MakeMsg(client,"logout")//写入用户退出消息到全局channel
return
}
}
}
func Manager () {
//初始化map
onlineMap = make(map[string]Client)
//循环监听message,是否有数据,转储到msg,无数据就阻塞
for {
msg := <- message
//循环发送消息给在线用户
for _, client := range onlineMap {
client.c <- msg
}
}
}
func main() {
//创建监听
listener, err := net.Listen("tcp","127.0.0.1:8008")
if err != nil {
fmt.Println("Listen err------>",err)
return
}
defer listener.Close()
//创建管理者goroutine,管理map和全局channel
go Manager()
//循环监听客户端连接
for {
conn, err := listener.Accept()
if err != nil {
fmt.Println("Accept err------>",err)
return
}
//启动go程处理客户端
go HandlerConn(conn)
}
}
网友评论