Go语言WebSocket：从零构建实时应用

使用Go实现WebSocket：构建实时应用

现代Web应用程序越来越需要客户端和服务器之间的即时数据传输。传统的HTTP请求及其"请求-响应"循环并不适用于对速度要求高的场景：在线聊天、股票行情、多人在线游戏或实时通知。此时WebSocket协议派上了用场——它提供持久化的双向连接。本文将讲解如何使用Go（Golang）实现WebSocket，从基础服务器到完整的实时应用程序。

什么是WebSocket，为什么选择Go？

WebSocket是基于TCP的协议，允许在客户端（浏览器）和服务器之间建立持久连接。握手完成后，数据无需额外的HTTP头部即可双向传输，从而显著降低延迟。

为什么选择Go？ Go语言因其内置的并发支持（goroutine和channel）、高性能以及简洁的网络编程代码，非常适合用于WebSocket服务器。与Node.js不同，Go不需要外部事件循环，而且其编译型特性在速度上更有优势。

基于Go的基础WebSocket服务器

在Go中使用WebSocket，我们将采用流行的库 gorilla/websocket。它提供了便捷的API并处理了协议细节。

安装该库：

go get github.com/gorilla/websocket

现在编写一个简单的回声服务器，它将收到的消息原样返回给客户端。

package main

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

var upgrader = websocket.Upgrader{    CheckOrigin: func(r *http.Request) bool {        return true // 开发用。生产环境中请检查Origin。    },}

func echoHandler(w http.ResponseWriter, r *http.Request) {    conn, err := upgrader.Upgrade(w, r, nil)    if err != nil {        log.Print("升级失败:", err)        return    }    defer conn.Close()

    for {        messageType, message, err := conn.ReadMessage()        if err != nil {            log.Println("读取错误:", err)            break        }        log.Printf("已接收: %s", message)

        if err := conn.WriteMessage(messageType, message); err != nil {            log.Println("写入错误:", err)            break        }    }}

func main() {    http.HandleFunc("/ws", echoHandler)    log.Println("服务器已在 :8080 启动")    log.Fatal(http.ListenAndServe(":8080", nil))}

代码说明：

• Upgrader — 用于将HTTP连接"升级"为WebSocket的对象。
• CheckOrigin — 检查请求来源的函数。生产环境中请务必配置此项。
• conn.ReadMessage() 和 conn.WriteMessage() — 读/写消息的主要方法。
• for 循环保持连接打开，直到出现错误。

基于WebSocket的实时聊天架构

回声服务器是基础，但对于实时应用程序（如聊天），需要更复杂的逻辑。下面介绍一种架构，其中多个客户端可以通过服务器交换消息。

核心组件：

• Hub — 中央管理器，存储活跃连接的列表。
• Client — 表示单个WebSocket连接的结构体。
• 通道（channels） — 用于在goroutine之间安全地传输数据。

package main

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

var upgrader = websocket.Upgrader{    CheckOrigin: func(r *http.Request) bool { return true },}

// Clien