Go语言实现WebSocket（手把手教你构建实时网络通信应用）

什么是 WebSocket？

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。与传统的 HTTP 请求-响应模式不同，WebSocket 允许服务器主动向客户端推送数据，实现真正的双向实时通信。

准备工作

在开始编码前，请确保你已安装 Go 语言环境（建议 1.16+ 版本）。我们还将使用 Gorilla WebSocket 库，这是 Go 社区中最流行的 WebSocket 实现。

首先，在你的项目目录下初始化 Go 模块：

go mod init websocket-demo

然后安装 Gorilla WebSocket 包：

go get github.com/gorilla/websocket

编写 WebSocket 服务器

创建一个 main.go 文件，写入以下代码：

package mainimport (    "log"    "net/http"    "github.com/gorilla/websocket")// 定义升级器，用于将 HTTP 连接升级为 WebSocket 连接var upgrader = websocket.Upgrader{    CheckOrigin: func(r *http.Request) bool {        return true // 允许所有跨域请求（生产环境请谨慎配置）    },}func echo(w http.ResponseWriter, r *http.Request) {    // 升级 HTTP 连接为 WebSocket    conn, err := upgrader.Upgrade(w, r, nil)    if err != nil {        log.Print("升级失败: ", err)        return    }    defer conn.Close()    // 循环读取消息并回显    for {        messageType, p, err := conn.ReadMessage()        if err != nil {            log.Println("读取消息出错:", err)            break        }        log.Printf("收到消息: %s", p)        // 将消息原样返回给客户端        if err := conn.WriteMessage(messageType, p); err != nil {            log.Println("发送消息出错:", err)            break        }    }}func main() {    http.HandleFunc("/ws", echo)    log.Println("WebSocket 服务器启动，监听 :8080/ws")    log.Fatal(http.ListenAndServe(":8080", nil))}

这段代码实现了一个简单的“回显”服务器：任何客户端发送的消息都会被原样返回。

测试 WebSocket 服务器

运行服务器：

go run main.go

你可以使用浏览器开发者工具或在线 WebSocket 测试工具（如 websocket.org/echo）连接 ws://localhost:8080/ws 并发送消息进行测试。

编写简易 WebSocket 客户端（可选）

你也可以用 Go 写一个命令行客户端来测试：

package mainimport (    "bufio"    "log"    "os"    "github.com/gorilla/websocket")func main() {    conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)    if err != nil {        log.Fatal("连接失败:", err)    }    defer conn.Close()    go func() {        for {            _, msg, err := conn.ReadMessage()            if err != nil {                log.Println("接收消息失败:", err)                return            }            log.Printf("收到服务器消息: %s\n", msg)        }    }()    scanner := bufio.NewScanner(os.Stdin)    for scanner.Scan() {        text := scanner.Text()        if err := conn.WriteMessage(websocket.TextMessage, []byte(text)); err != nil {            log.Println("发送失败:", err)            break        }    }}

运行客户端后，输入任意文本，服务器会将其回显回来。

小结

通过本教程，你已经掌握了如何使用 Go语言 和 Gorilla WebSocket 库构建一个基础的 WebSocket 服务。这为你开发更复杂的 实时通信 应用（如聊天系统、协同编辑、实时监控面板等）打下了坚实基础。

记住，实际项目中还需考虑连接管理、并发安全、错误处理、心跳机制等高级话题。但万丈高楼平地起，你现在已迈出了关键的第一步！

关键词回顾：Go语言、WebSocket、网络编程、实时通信