package main;
import (
"net"
"fmt"
"bufio"
)
//里面的代码部分参考cmu440课程
//https://github.com/cmu440/p0
//广播服务器接口
type MultiEchoServer interface {
//开始
Start(port int) error;
//停止
Close();
//当前客户端连接数
Count() int;
}
//广播服务器
type multiEchoServer struct {
lis *net.TCPListener;
//当前客户端ID
curClientId int;
//所有客户端
clients map[int]*client;
//广播消息
broadcastMsg chan []byte;
}
//客户端
type client struct {
//ID
id int;
//连接
conn net.Conn;
//接收消息
recvMsg chan []byte;
//发送消息
sendMsg chan []byte;
//接收消息是否关闭
isRecvMsgClose chan bool;
//发送消息是否关闭
isSendMsgClose chan bool;
//服务器
mes *multiEchoServer;
}
//返回一个广播服务器
func New() *multiEchoServer {
return &multiEchoServer{
curClientId: 0,clients: make(map[int]*client),broadcastMsg: make(chan []byte,1),};
}
//启动服务器
func (m *multiEchoServer) Start(port int) error {
//获取tcp地址
addr,err := net.ResolveTCPAddr("tcp",fmt.Sprintf(":%d",port));
if err != nil {
return err;
}
//监听端口
m.lis,err = net.ListenTCP("tcp",addr);
if err != nil {
return err;
}
//启一个goroutine处理广播
go m.BroadcastLoop();
//启一个goroutine处理客户端来的连接
go func() {
for {
conn,err := m.lis.Accept();
if err != nil {
continue;
}
cli := &client{
id: m.curClientId,conn: conn,recvMsg: make(chan []byte,sendMsg: make(chan []byte,isRecvMsgClose: make(chan bool,isSendMsgClose: make(chan bool,mes: m,}
//加客户端加入到服务器clients中
clis := m.clients;
clis[m.curClientId] = cli;
m.clients = clis;
m.curClientId++;
//启两个goroutine分别处理客户端的接收与发送消息
go cli.RecvLoop();
go cli.SendLoop();
}
}();
return nil;
}
//停止服务器
func (m *multiEchoServer) Close() {
m.lis.Close();
//循环关闭客户端
for _,client := range m.clients {
client.conn.Close();
//这里只需给一个发送消息就好了
client.isRecvMsgClose <- true;
}
}
//返回当前客户端连接数
func (m *multiEchoServer) Count() int {
return len(m.clients);
}
//处理广播
func (m *multiEchoServer) BroadcastLoop() {
for {
select {
case data := <-m.broadcastMsg:
{
//遍历所有客户端,循环发送消息
for _,client := range m.clients {
client.sendMsg <- data;
}
break;
}
}
}
}
//删除客户端
func (m *multiEchoServer) DelClient(c *client) error {
c.conn.Close();
clis := m.clients;
delete(clis,c.id);
m.clients = clis;
return nil;
}
//处理客户端接收消息
func (c *client) RecvLoop() {
defer func() {
fmt.Println(c.conn.RemoteAddr().String() + " RecvLoop exit");
}();
for {
read := bufio.NewReader(c.conn);
data,err := read.ReadBytes('\n');
if err != nil {
c.isSendMsgClose <- true;
//这里直接返回,如果不直接返回
//当客户端退出时,这里会运行2次,导致c.isSendMsgClose<-true执行2次造成阻塞
//没有机会运行后面的select,那么一直无法返回,不能回收。
return;
}
select {
//接收消息是否关闭
case <-c.isRecvMsgClose:
{
c.isSendMsgClose <- true;
return;
}
//广播消息
case c.mes.broadcastMsg <- data:
{
break;
}
}
}
}
//处理客户端发送消息
func (c *client) SendLoop() {
defer func() {
fmt.Println(c.conn.RemoteAddr().String() + " SendLoop exit");
}();
for {
select {
//发送消息关闭,则把客户端从服务中删除
case <-c.isSendMsgClose:
{
c.mes.DelClient(c);
return;
}
//向客户写入要发送的消息
case data := <-c.sendMsg:
{
_,err := c.conn.Write(data);
if err != nil {
return;
}
}
}
}
}
func main() {
mes := New();
mes.Start(8888);
//循环
select {};
}
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。