GolangWebSocket通信教程详解

本文详细介绍了如何使用 Golang 和强大的 `gorilla/websocket` 库实现 WebSocket 通信，并提供了完整的示例代码。首先，文章指导读者安装必要的依赖，然后分别展示了服务器端和客户端的实现方式。服务器端代码监听 `/ws` 路径，将 HTTP 连接升级为 WebSocket 连接，并实现消息的接收与回显。客户端代码则连接到本地服务器，发送预设消息并接收服务器的回显。最后，文章提供了清晰的运行步骤，帮助读者快速验证双向通信是否成功。WebSocket 作为一种在单个 TCP 连接上进行全双工通信的协议，非常适合构建实时应用，本文的示例可作为 Golang WebSocket 开发的快速入门指南。

答案：使用Golang和gorilla/websocket库可实现WebSocket通信。1. 安装依赖：go get github.com/gorilla/websocket；2. 编写服务器端代码，监听/ws路径，升级HTTP连接为WebSocket，接收并回显消息；3. 编写客户端代码，连接本地服务器，发送"Hello, WebSocket!"并接收回显；4. 先运行服务器，再运行客户端，验证双向通信成功。

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，非常适合实时应用，比如聊天、通知推送等。Golang 提供了简洁高效的语言特性来实现 WebSocket 服务端和客户端。下面是一个基于 gorilla/websocket 库的完整示例，展示如何用 Go 编写 WebSocket 服务器与客户端。

1. 安装依赖

go get github.com/gorilla/websocket

2. WebSocket 服务器端实现

server.go

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("升级 WebSocket 失败:", err)
        return
    }
    defer conn.Close()

    for {
        // 读取消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Print("读取消息失败:", err)
            break
        }

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

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

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

3. WebSocket 客户端实现

client.go

package main

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

func main() {
    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
    log.Printf("连接到 %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("连接失败:", err)
    }
    defer conn.Close()

    // 发送消息
    message := []byte("Hello, WebSocket!")
    err = conn.WriteMessage(websocket.TextMessage, message)
    if err != nil {
        log.Fatal("发送消息失败:", err)
    }

    // 读取回显
    _, msg, err := conn.ReadMessage()
    if err != nil {
        log.Fatal("读取消息失败:", err)
    }

    log.Printf("收到回显: %s", msg)
}

4. 运行步骤

• 运行服务器：go run server.go
• 另起终端运行客户端：go run client.go
• 服务器应输出收到的消息，客户端输出回显内容

基本上就这些。这个例子展示了最基础的 WebSocket 双向通信流程。你可以在此基础上扩展：广播消息、维护连接池、心跳检测等。不复杂但容易忽略。

今天关于《GolangWebSocket通信教程详解》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！