咔叽游戏

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
查看: 647|回复: 0

[Golang] GO使用socket和channel实现简单控制台聊天室

[复制链接]
  • TA的每日心情
    无聊
    2019-6-2 14:11
  • 签到天数: 4 天

    [LV.2]圆转纯熟

    发表于 2022-1-10 11:07:33 | 显示全部楼层 |阅读模式
    使用socket和channel,实现简单控制台聊天室
    这里使用socket和channel,演示在GO中如何编写一个简单网络程序
    功能分析

    聊天室主要功能:用户可以加入/离开聊天室;每个用户发送的消息,广播给所有人
    聊天室分为客户端和服务端,客户端负责发送消息和打印服务器消息,服务器负责接收客户端消息,并广播给所有人
    客户端可以使用telnet程序
    服务端是需要实现的。需要实现的功能,
      如何保存多个客户端的连接,管理连接的接入与断开如何接收和广播客户端消息
    实现思路

    通过功能分析,拆分为聊天室结构体和客户端结构体
    聊天室结构体负责管理当前接入的客户端和广播消息
    客户端结构体负责管理socket连接和需要接收与发送的数据
    客户端连接/断开时通知聊天室;客户端发送的消息实际是转发给聊天室,然后聊天室再广播出去
    完整代码
    1. package main
    2. import (
    3.         "bufio"
    4.         "fmt"
    5.         "log"
    6.         "net"
    7. )
    8. type Client struct {
    9.         id      string
    10.         conn    *net.Conn
    11.         message chan string
    12. }
    13. type Hub struct {
    14.         clients  map[*Client]bool
    15.         entering chan *Client
    16.         leaving  chan *Client
    17.         messages chan string
    18. }
    19. func main() {
    20.         hub := &Hub{
    21.                 clients:  make(map[*Client]bool),
    22.                 entering: make(chan *Client),
    23.                 leaving:  make(chan *Client),
    24.                 messages: make(chan string),
    25.         }
    26.         listener, err := net.Listen("tcp", ":8000")
    27.         if err != nil {
    28.                 log.Fatal(err)
    29.         }
    30.         go hub.broadcaster()
    31.         for {
    32.                 conn, err := listener.Accept()
    33.                 if err != nil {
    34.                         log.Println(err)
    35.                         continue
    36.                 }
    37.                 go hub.handleConn(conn)
    38.         }
    39. }
    40. func (hub *Hub) broadcaster() {
    41.         for {
    42.                 select {
    43.                 case msg := <-hub.messages:
    44.                         for cli := range hub.clients {
    45.                                 cli.message <- msg
    46.                         }
    47.                 case cli := <-hub.entering:
    48.                         hub.clients[cli] = true
    49.                 case cli := <-hub.leaving:
    50.                         delete(hub.clients, cli)
    51.                 }
    52.         }
    53. }
    54. func (hub *Hub) handleConn(conn net.Conn) {
    55.         defer conn.Close()
    56.         ch := make(chan string)
    57.         who := conn.RemoteAddr().String()
    58.         client := &Client{who, &conn, ch}
    59.         go hub.writeLoop(client)
    60.         ch <- "welcome " + client.id
    61.         hub.messages <- client.id + " join chat"
    62.         hub.entering <- client
    63.         hub.readLoop(client)
    64.         hub.messages <- client.id + " has left"
    65.         hub.leaving <- client
    66. }
    67. func (hub *Hub) writeLoop(client *Client) {
    68.         for msg := range client.message {
    69.                 fmt.Fprintf(*client.conn, "%s\n", msg)
    70.         }
    71. }
    72. func (hub *Hub) readLoop(client *Client) {
    73.         input := bufio.NewScanner(*client.conn)
    74.         for input.Scan() {
    75.                 hub.messages <- client.id + ": " + input.Text()
    76.         }
    77. }
    复制代码
    分析

    实现的关键是封装了客户端通信channel,无论是远程发送过来的消息还是聊天室广播的消息,都通过这个channel传递,且这个channel是绑定客户端的
    参考链接中,直接使用channel来定义客户端type client chan<- string,其实更能表达这一点
    为了容易理解,这里将channel封装为客户端的一个通信管道,客户端还可以有别的属性,例如:id、连接和超时时间等
    参考: Go 网络编程示例
    到此这篇关于GO实现简单控制台聊天室的文章就介绍到这了,更多相关go控制台聊天室内容请搜索咔叽论坛以前的文章或继续浏览下面的相关文章希望大家以后多多支持咔叽论坛!

    原文地址:https://www.jb51.net/article/232439.htm

    QQ|免责声明|小黑屋|手机版|Archiver|咔叽游戏

    GMT+8, 2024-3-29 05:04

    Powered by Discuz! X3.4

    © 2001-2023 Discuz! Team.

    快速回复 返回顶部 返回列表