使用Golang怎么实现并发聊天室?针对这个问题,这篇文章详细介绍了相对应的分析和解答,希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
package main
import (
"net"
"fmt"
"strings"
"time"
)
//定义的此结构体为全局map的value值,包括每一个用户的姓名,ip地址和私人管道
type client struct {
name string
addr string
C chan string
}
/*这个函数是将私人管道中的内容发送给用户,配合全局管道Message使用可以实现广播的功能,
单独使用可以实现私聊的功能*/
func writemsg2client(clinet client,conn net.Conn) {
for m := range clinet.C {
conn.Write([]byte(m + "\n"))
}
}
//这只是一个封装好用来统一(发送信息格式)的小函数,不用在意
func makemsg(name string, addr string, s string) string {
return "[" + addr + "]" + name + s
}
//每一个进入聊天室的用户都将启动一个handleconn的go程来处理事件
func handleconn(conn net.Conn) {
defer conn.Close()
/*用户连接进来以后要初始化全局map,把自己的信息加入到字典里,相当于进到聊天室里之前要登
记一下个人信息,注意姓名初始为ip地址。*/
addr := conn.RemoteAddr().String()
fmt.Printf("用户%s进入了房间\n", addr)
client := client{addr, addr, make(chan string)}
//在这里启动子go程,功能上面已经提及
go writemsg2client(client,conn)
onlinemap[addr] = client
//登录进来一切准备就绪后就给所有人广播上线信息啦
Message <- makemsg(client.name, addr, "login")
//下面这三个变量服务于下面一些小功能
var haschat=make(chan bool)
var ifquit=make(chan bool)
var flag bool
//从这单独开启一个go程来读取用户输入的信息
go func() {
buf:=make([]byte,4096)
for {
n,_:=conn.Read(buf)
if n==0 {
fmt.Printf("%s离开了房间\n",client.name)
ifquit<-true
return
}
//改名功能的实现
if string(buf[:7])=="Rename|" {
client.name=strings.Split(string(buf[:n-1]),"|")[1]
onlinemap[addr]=client
conn.Write([]byte("rename success\n"))
}else if string(buf[:n-1])=="/who"{
//查询在线用户信息的功能
for _,s:=range onlinemap{
conn.Write([]byte(s.name+"online\n"))
}
}else if string(buf[:2])=="m|"&&strings.Count(string(buf[:n]),"|")==2 {
/*私聊功能的实现,其实私聊功能就是跳过了往全局Message里传输信息,
改为直接向私人管道里传输信息*/
flag=false
slice:=strings.Split(string(buf[:n-1]),"|")
for _,a:=range onlinemap{
//遍历所有在线用户,向指定的用户管道中发送信息
if a.name==slice[1]{
flag=true
a.C<-makemsg(client.name,addr,slice[2])
conn.Write([]byte("send success"))
}
}
if flag {
conn.Write([]byte("no such man or not online"))
}
} else {
Message<-makemsg(client.name,addr,string(buf[:n-1]))
}
haschat<-true
}
}()
for {
select {
case <-haschat:
//超时强踢
case <-time.After(time.Minute*3):
delete(onlinemap,addr)
Message<-makemsg(client.name,addr,"out time to leave")
close(client.C)
return
case <-ifquit:
//退出处理
delete(onlinemap,addr)
Message<-makemsg(client.name,addr,"out time to leave")
close(client.C)
return
}
}
}
//这个函数用来将全局Message中的内容全部塞到私人管道C里,实现上下线广播和群聊的功能
func Manager() {
for {
msg := <-Message
for _, s := range onlinemap {
s.C <- msg
}
}
}
var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)
//主函数
func main() {
listener, _ := net.Listen("tcp", "127.0.0.1:6666")
defer listener.Close()
//提前开启全局Message的go程,防止被阻塞
go Manager()
for {
conn, err := listener.Accept()
if err != nil {
fmt.Println("accept err", err)
continue
}
//每一个连接进来的用户都会被分配进入一个子go程,用来处理上面我们提到的各种功能
go handleconn(conn)
}
}
关于使用Golang怎么实现并发聊天室问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注亿速云行业资讯频道了解更多相关知识。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。