探索 Golang 云原生游戏服务器开发，根据官方示例实战 Gorilla WebSocket 的用法

......
func main() {
  ......
  // 应用一运行，就初始化 `Hub` 管理工作
  hub := newHub()
  // 开个 goroutine，后台运行监听三个 channel
  // register：注册客户端 channel
  // unregister：注销客户端 channel
  // broadcast：广播客户端 channel
  go hub.run()
  http.HandleFunc("/", serveHome)
  http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
    serveWs(hub, w, r)
  })
  .....
}
......
// serveWs 处理来自每一个客户端的 "/ws" 请求。
func serveWs(hub *Hub, w http.ResponseWriter, r *http.Request) {
  // 升级这个请求为 `websocket` 协议
  conn, err := upgrader.Upgrade(w, r, nil)
  if err != nil {
    log.Println(err)
    return
  }
  // 初始化当前的客户端实例，并与 `hub` 中心管理勾搭上，
  client := &Client{hub: hub, conn: conn, send: make(chan []byte, 256)}
  client.hub.register <- client
  // 通过在新的goroutines中完成所有工作，允许调用者引用内存的集合。
  // 其实对当前 `websocket` 连接的 `I/O` 操作
  // 写操作（发消息到客户端）-> 这里 `Hub` 会统一处理
  go client.writePump()
  // 读操作（对消息到客户端）-> 读完当前连接立即发 -> 交由 `Hub` 分发消息到所有连接
  go client.readPump()
}

func (h *Hub) run() {
  for {
    select {
    // 注册 channel
    case client := <-h.register:
      // 键值对操作，没啥好说的
      h.clients[client] = true
    // 注销 channel
    case client := <-h.unregister:
      // 键值对操作，没啥好说的
      if _, ok := h.clients[client]; ok {
        delete(h.clients, client)
        close(client.send)
      }
    // 广播 channel
    case message := <-h.broadcast:
      for client := range h.clients {
        select {
        // 直接送入各个连接的 send channel
        case client.send <- message:
        // 卡住，这里直接踢掉
        default:
          close(client.send)
          delete(h.clients, client)
        }
      }
    }
  }
}

// readPump 从 Websocket 连接用泵将消息输送到 hub。
// 应用程序在每个连接 goroutine 中运行 readPump。
// 应用程序通过执行此 goroutine 中的所有读取来确保连接上最多有一个 reader。
func (c *Client) readPump() {
  defer func() {
    c.hub.unregister <- c
    c.conn.Close()
  }()
  // SetReadLimit 设置从对等方读取的消息的最大大小。如果消息超出限制，则连接会将关闭消息发送给对等方，然后将ErrReadLimit返回给应用程序。
  c.conn.SetReadLimit(maxMessageSize)
  // SetReadDeadline 设置基础网络连接上的读取期限。读取超时后，websocket 连接状态已损坏，以后所有读取将返回错误。参数值为零表示读取不会超时。
  c.conn.SetReadDeadline(time.Now().Add(pongWait))
  // SetPongHandler 为从 peer 接收到的 pong 消息设置处理程序。处理程序的参数是 PONG 消息应用程序数据。默认的 pong 处理程序不执行任何操作。
  // handler函数从 NextReader、ReadMessage 和 message reader Read方法处被调用。
  c.conn.SetPongHandler(func(string) error { c.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil })
  for {
    // 读取消息
    _, message, err := c.conn.ReadMessage()
    if err != nil {
      // 错误处理
      if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
        log.Printf("error: %v", err)
      }
      break
    }
    // 整理 message 内容
    message = bytes.TrimSpace(bytes.Replace(message, newline, space, -1))
    // 广播
    c.hub.broadcast <- message
  }
}
// writePump 将消息从 hub pump到 websocket 连接。
// 为每个连接启动运行 writePump 的 goroutine。
// 通过执行这个 goroutine 中的所有写操作，应用程序确保连接最多只有一个 writer。
func (c *Client) writePump() {
  ticker := time.NewTicker(pingPeriod)
  defer func() {
    ticker.Stop()
    c.conn.Close()
  }()
  for {
    select {
    // 写消息到当前的 websocket 连接
    case message, ok := <-c.send:
      c.conn.SetWriteDeadline(time.Now().Add(writeWait))
      if !ok {
        // hub 关闭这个 channel
        c.conn.WriteMessage(websocket.CloseMessage, []byte{})
        return
      }
      // NextWriter 为要发送的下一条消息返回一个写入器。写入器的Close方法将完整的消息刷新到网络。
      w, err := c.conn.NextWriter(websocket.TextMessage)
      if err != nil {
        return
      }
      w.Write(message)
      // 将排队聊天消息添加到当前的 websocket 消息中。
      n := len(c.send)
      for i := 0; i < n; i++ {
        w.Write(newline)
        w.Write(<-c.send)
      }
      if err := w.Close(); err != nil {
        return
      }
    // 定时检测下客户端的状态
    case <-ticker.C:
      c.conn.SetWriteDeadline(time.Now().Add(writeWait))
      if err := c.conn.WriteMessage(websocket.PingMessage, nil); err != nil {
        return
      }
    }
  }
}

docker build -f Dockerfile.dev -t cloud-native-game-server:dev .

DEMO=2-gorilla-websocket-chat docker-compose up demo
#docker-compose down

DEMO=2-gorilla-websocket-chat docker-compose up demo-debug