今天就跟大家聊聊有关GO使用socket和channel怎么实现简单控制台聊天室,可能很多人都不太了解,为了让大家更加了解,小编给大家总结了以下内容,希望大家根据这篇文章可以有所收获。
使用socket和channel,实现简单控制台聊天室
这里使用socket和channel,演示在GO中如何编写一个简单网络程序
功能分析
聊天室主要功能:用户可以加入/离开聊天室;每个用户发送的消息,广播给所有人
聊天室分为客户端和服务端,客户端负责发送消息和打印服务器消息,服务器负责接收客户端消息,并广播给所有人
客户端可以使用telnet程序
服务端是需要实现的。需要实现的功能,
如何保存多个客户端的连接,管理连接的接入与断开
如何接收和广播客户端消息
实现思路
通过功能分析,拆分为聊天室结构体和客户端结构体
聊天室结构体负责管理当前接入的客户端和广播消息
客户端结构体负责管理socket连接和需要接收与发送的数据
客户端连接/断开时通知聊天室;客户端发送的消息实际是转发给聊天室,然后聊天室再广播出去
完整代码
package mainimport ("bufio""fmt""log""net")type Client struct {id stringconn *net.Connmessage chan string}type Hub struct {clients map[*Client]boolentering chan *Clientleaving chan *Clientmessages chan string}func main() {hub := &Hub{clients: make(map[*Client]bool),entering: make(chan *Client),leaving: make(chan *Client),messages: make(chan string),}listener, err := net.Listen("tcp", ":8000")if err != nil {log.Fatal(err)}go hub.broadcaster()for {conn, err := listener.Accept()if err != nil {log.Println(err)continue}go hub.handleConn(conn)}}func (hub *Hub) broadcaster() {for {select {case msg := <-hub.messages:for cli := range hub.clients {cli.message <- msg}case cli := <-hub.entering:hub.clients[cli] = truecase cli := <-hub.leaving:delete(hub.clients, cli)}}}func (hub *Hub) handleConn(conn net.Conn) {defer conn.Close()ch := make(chan string)who := conn.RemoteAddr().String()client := &Client{who, &conn, ch}go hub.writeLoop(client)ch <- "welcome " + client.idhub.messages <- client.id + " join chat"hub.entering <- clienthub.readLoop(client)hub.messages <- client.id + " has left"hub.leaving <- client}func (hub *Hub) writeLoop(client *Client) {for msg := range client.message {fmt.Fprintf(*client.conn, "%s\n", msg)}}func (hub *Hub) readLoop(client *Client) {input := bufio.NewScanner(*client.conn)for input.Scan() {hub.messages <- client.id + ": " + input.Text()}}
分析
实现的关键是封装了客户端通信channel,无论是远程发送过来的消息还是聊天室广播的消息,都通过这个channel传递,且这个channel是绑定客户端的
参考链接中,直接使用channel来定义客户端type client chan<- string
,其实更能表达这一点
为了容易理解,这里将channel封装为客户端的一个通信管道,客户端还可以有别的属性,例如:id、连接和超时时间等
看完上述内容,你们对GO使用socket和channel怎么实现简单控制台聊天室有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注编程网行业资讯频道,感谢大家的支持。