Golang实现WebSocket收发教程详解

**Golang实现WebSocket数据收发教程：快速构建实时通信应用** 想用Go语言构建实时通信应用？本文将带你快速上手Golang WebSocket开发！利用强大的`gorilla/websocket`库，轻松实现客户端与服务端之间的双向数据传输。本文详细讲解如何使用`go get`命令安装依赖，并通过`Upgrader`实例将HTTP连接升级为WebSocket连接。通过示例代码，你将学会如何搭建一个简单的WebSocket服务端，实现消息的接收与回显。同时，提供JavaScript客户端测试代码，方便你快速验证连接。本文还重点强调了跨域处理、消息读写以及连接关闭等关键点，助你避免常见错误，构建稳定可靠的WebSocket应用。无论你是构建实时聊天、通知推送还是其他实时互动应用，本教程都能为你提供坚实的基础。

Go语言通过gorilla/websocket库实现WebSocket通信，首先使用go get安装依赖，然后创建Upgrader实例将HTTP连接升级为WebSocket，示例代码展示了服务端接收并回显消息的过程，客户端可用JavaScript测试连接，关键点包括允许跨域、读写消息及连接关闭，适用于实时通信场景。

Go语言实现WebSocket通信主要依赖第三方库 gorilla/websocket，这是目前最常用且稳定的选择。下面介绍如何用Golang搭建一个简单的WebSocket服务端，完成数据的接收与发送。

1. 安装 gorilla/websocket

go get github.com/gorilla/websocket

2. 创建WebSocket服务端

使用 gorilla/websocket 提供的 Upgrader 将HTTP连接升级为WebSocket连接。

示例代码：

package main

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

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域连接，生产环境应做更严格校验
    },
}

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

    // 循环读取消息
    for {
        msgType, message, err := conn.ReadMessage()
        if err != nil {
            log.Printf("读取失败: %v", err)
            break
        }

        log.Printf("收到消息: %s", message)

        // 回显消息
        if err := conn.WriteMessage(msgType, message); err != nil {
            log.Printf("发送失败: %v", err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/ws", wsHandler)
    log.Println("服务器启动，地址: :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

3. 客户端测试（JavaScript）

可以用浏览器控制台测试连接：

const ws = new WebSocket("ws://localhost:8080/ws");
ws.onopen = () => {
    ws.send("你好，Golang WebSocket！");
};
ws.onmessage = (event) => {
    console.log("收到:", event.data);
};
ws.onclose = () => {
    console.log("连接关闭");
};

4. 关键点说明

• Upgrader.CheckOrigin：默认拒绝跨域请求，开发时可设为true，上线前应限制合法来源。
• ReadMessage()：阻塞读取客户端消息，返回消息类型（text/binary）和内容。
• WriteMessage()：向客户端发送数据，保持与读取相同的消息类型更稳妥。
• defer conn.Close()：确保连接关闭，防止资源泄漏。

基本上就这些。通过 gorilla/websocket 可以轻松实现全双工通信，适合实时聊天、通知推送等场景。不复杂但容易忽略错误处理和连接管理，建议结合 context 或 hub 模式做更完善的控制。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang实现WebSocket收发教程详解》文章吧，也可关注golang学习网公众号了解相关技术文章。