Gorilla WebSocket 库 for Go | Golang WebSocket

Gorilla WebSocket: 用于 Go 语言实时通信的强大工具

在现代 Web 开发中，对实时数据交换的需求变得越来越重要。无论是聊天、在线游戏、服务器监控还是股票行情——WebSocket 都是客户端与服务器之间双向通信的事实标准。对于 Go (Golang) 语言来说，最流行且最稳定的 WebSocket 库之一是 Gorilla WebSocket。

gorilla/websocket 库属于 Gorilla Toolkit 生态系统——这是一组用于 Go 语言 Web 开发的高质量包。它提供了创建 WebSocket 服务器和客户端的完整功能集，支持所有主要协议 (RFC 6455)，并且使用简单。

Gorilla WebSocket 的主要特性

完全支持 WebSocket 协议 — 处理文本和二进制帧、消息分片、控制帧 (ping/pong)。
便捷的 API — 通过 ReadMessage 和 WriteMessage 函数提供极简接口。
支持 TLS/SSL — 安全的 WebSocket 连接 (wss://)。
灵活的配置 — 可以设置超时、缓冲区大小、检查请求来源。
与 HTTP 兼容 — 易于与任何 Go HTTP 服务器 (net/http) 集成。
支持上下文 — 支持 context.Context 用于取消操作。

安装库

要安装 Gorilla WebSocket，请在终端中执行以下命令：

go get github.com/gorilla/websocket

之后，将导入添加到您的 Go 文件中：

import "github.com/gorilla/websocket"

WebSocket 服务器基础示例

让我们看一个最简单的回声服务器，它接受 WebSocket 连接并将接收到的消息发送回去。

package main

import (    "log"    "net/http"

    "github.com/gorilla/websocket")

var upgrader = websocket.Upgrader{    CheckOrigin: func(r *http.Request) bool {        return true // 允许所有来源（用于开发）    },}

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

    for {        messageType, message, err := conn.ReadMessage()        if err != nil {            log.Println("Read error:", err)            break        }        log.Printf("Received: %s", message)

        if err := conn.WriteMessage(messageType, message); err != nil {            log.Println("Write error:", err)            break        }    }}

func main() {    http.HandleFunc("/echo", echoHandler)    log.Println("Server started on :8080")    log.Fatal(http.ListenAndServe(":8080", nil))}

代码详细解释

Upgrader — 这是一个结构体，它将普通的 HTTP 请求转换为 WebSocket 连接。在示例中，我们允许来自任何来源的请求 (CheckOrigin 字段)。在生产环境中，应该检查特定的域名。
Upgrade — 主要方法，它接收 http.ResponseWriter 和 *http.Request 并返回 *websocket.Conn。
ReadMessage — 阻塞调用，等待来自客户端的新消息。返回消息类型（文本或二进制数据）和数据本身。
WriteMessage — 将消息发送回客户端。在这种情况下，我们只是返回收到的相同内容。