GolangWebSocket教程：客户端与服务端实现

本文详细介绍了如何使用 Golang 语言和流行的 `gorilla/websocket` 库构建 WebSocket 客户端和服务端。WebSocket 是一种在单个 TCP 连接上实现全双工通信的协议，尤其适用于实时应用场景。教程首先指导读者安装必要的依赖，然后分别展示了服务端和客户端的代码实现，服务端监听指定路径，升级 HTTP 连接为 WebSocket，并实现消息的接收和回显。客户端则连接服务端，发送消息并接收服务器的回显。最后，教程提供了清晰的运行步骤，帮助开发者快速验证 WebSocket 双向通信的实现。通过本文，开发者可以快速掌握 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学习网公众号，带你了解更多关于的知识点！