首页 > Golang > Go教程

Golang实现WebSocket通信详解

时间：2025-12-27 17:01:35 375浏览收藏

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《Golang实现WebSocket通信方法解析》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

Go语言通过gorilla/websocket库实现WebSocket通信，1. 服务端监听/ws路径并升级HTTP连接；2. 客户端使用ws协议连接服务器；3. 双方通过ReadMessage和WriteMessage收发数据，支持跨域与消息回显。

Golang如何实现WebSocket通信

Go语言实现WebSocket通信主要依赖第三方库，最常用的是gorilla/websocket。它提供了对WebSocket协议的完整支持，使用简单且性能良好。下面介绍如何用Golang搭建一个基础的WebSocket服务端和客户端。

安装gorilla/websocket库

在项目目录下执行：

go get github.com/gorilla/websocket

编写WebSocket服务端

服务端负责监听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 handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("升级失败:", err)
        return
    }
    defer conn.Close()

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

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

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

编写WebSocket客户端

客户端通过ws或wss连接到服务端，发送和接收数据。

示例代码：

package main

import (
    "log"
    "net/url"
    "os"
    "os/signal"
    "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()

    done := make(chan struct{})

    // 读取协程
    go func() {
        defer close(done)
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("读取错误:", err)
                return
            }
            log.Printf("收到: %s", message)
        }
    }()

    // 发送消息
    ticker := time.NewTicker(time.Second)
    defer ticker.Stop()

    sig := make(chan os.Signal, 1)
    signal.Notify(sig, os.Interrupt)

    for {
        select {
        case

关键点说明

Upgrader.CheckOrigin：默认拒绝跨域请求，开发时可设为返回true，生产环境应做域名校验。

conn.ReadMessage()：返回消息类型（文本或二进制）和内容。

conn.WriteMessage()：发送消息到客户端，类型需匹配。

并发安全：每个连接建议使用独立的goroutine处理读写，避免阻塞。

连接管理：实际项目中可用map或channel池维护活跃连接，实现广播或多用户通信。

基本上就这些。使用gorilla/websocket能快速实现稳定可靠的WebSocket通信，适合聊天、实时通知等场景。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。