Go语言WebSocket：构建高级实时应用

引言：超越Go语言的基础WebSocket

Go语言的基础WebSocket示例通常只展示简单的回声连接。在实际的生产项目中——从交易终端到多人在线游戏和通知系统——需要更多功能：房间管理、Graceful Shutdown、扩展性和正确的错误处理。本文将使用gorilla/websocket库，深入讲解构建Go实时应用的高级模式。

我们将基于Hub（中央调度器）和Client（独立连接）构建架构，添加房间支持、容错机制，并展示如何正确关闭服务器。

Hub和Client架构：实时服务器的基础

为什么需要Hub？

每个WebSocket客户端都是一个独立的goroutine。直接从HTTP处理器管理它们很危险：会导致数据竞争和内存泄漏。Hub模式通过作为注册、注销和消息广播的统一通道来解决这个问题。

Hub的实现

package main

import (    "log"    "sync")

type Client struct {    hub  *Hub    send chan []byte    room string}

type Hub struct {    clients    map[*Client]bool    broadcast  chan Message    register   chan *Client    unregister chan *Client    mu         sync.RWMutex}

type Message struct {    Room    string    Data    []byte    Sender  *Client}

func NewHub() *Hub {    return &Hub{        clients:    make(map[*Client]bool),        broadcast:  make(chan Message, 256),        register:   make(chan *Client),        unregister: make(chan *Client),    }}

func (h *Hub) Run() {    for {        select {        case client := <-h.register:            h.mu.Lock()            h.clients[client] = true            h.mu.Unlock()            log.Printf("客户端已连接。总计：%d", len(h.clients))

        case client := <-h.unregister:            h.mu.Lock()            if _, ok := h.clients[client]; ok {                delete(h.clients, client)                close(client.send)            }            h.mu.Unlock()

        case message := <-h.broadcast:            h.mu.RLock()            for client := range h.clients {                // 仅发送到指定房间                if client.room == message.Room || message.Room == "" {                    select {                    case client.send <- message.Data:                    default:                        // 如果通道已满——断开客户端连接                        h.mu.RUnlock()                        h.unregister <- client                        h.mu.RLock()                    }                }            }            h.mu.RUnlock()        }    }}

关键点：

• 通道——register、unregister、broadcast——安全地同步goroutine。
• RWMutex——允许同时读取客户端列表，但阻止写入。
• 基于房间的路由——消息仅发送给同一房间的客户端。

房间管理和订阅

房间（Rooms）是客户端的逻辑分组。例如，在聊天应用中，每个对话都有自己的房间。在我们的Hub中，房间由Client的room字段定义。让我们看看客户端如何订阅房间。

带WebSocket升级的HTTP处理器

import (    "net/http"    "github.com/gorilla/websocket")

var upgrader = websocket.Upgrader{    ReadBufferSize:  1024,    WriteBufferSize: 1024,    CheckOrigin: func(r *http.Request) bool {        return true // 生产环境应替换为origin检查    },}

func serveWs(hub *Hub, w http.ResponseWriter, r *http.Request) {    conn, err := upgrader.Upgrade(w, r, nil)    if err !=