当前位置：首页 > 文章列表 > Golang > Go教程 > GolangWebSocket教程：客户端与服务器实现

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

2025-09-27 23:12:35 0浏览收藏

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Golang WebSocket客户端与服务器教程》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

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

GolangWebSocket客户端与服务器示例

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

1. 安装依赖

使用 gorilla/websocket 库，先安装：

go get github.com/gorilla/websocket

2. WebSocket 服务器端实现

服务器监听 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)
}