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

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《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学习网公众号！