Golang如何使用WebSocket发送消息_Golang WebSocket消息发送方法

使用gorilla/websocket库可实现在Go中发送WebSocket消息。1. 通过go get github.com/gorilla/websocket安装库；2. 使用websocket.DefaultDialer.Dial建立连接；3. 调用conn.WriteMessage发送文本或二进制消息；4. 服务端用websocket.Upgrader升级HTTP连接并处理读写；5. 建议封装心跳与重连机制提升稳定性。

在Go语言中使用WebSocket发送消息，通常借助第三方库gorilla/websocket，它是目前最流行且稳定的WebSocket实现。下面介绍如何建立连接并发送消息。

1. 安装 gorilla/websocket 库

go get github.com/gorilla/websocket

2. 建立WebSocket连接并发送消息

package main

import (
    "fmt"
    "log"
    "net/url"
    "os"

    "github.com/gorilla/websocket"
)

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

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

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

    fmt.Println("已发送消息:", message)

    // 可选：接收回显消息
    _, received, err := conn.ReadMessage()
    if err != nil {
        log.Println("读取消息失败:", err)
    } else {
        fmt.Println("收到回复:", string(received))
    }
}

3. 服务端接收并响应消息

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

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

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

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

4. 消息类型说明

• websocket.TextMessage：文本消息，内容为UTF-8编码字符串
• websocket.BinaryMessage：二进制消息，适合传输字节流（如图片、协议数据）

conn.WriteMessage(websocket.TextMessage, []byte("文本"))
conn.WriteMessage(websocket.BinaryMessage, []byte{0x01, 0x02, 0x03})

Whimsical

Whimsical推出的AI思维导图工具

下载

基本上就这些。只要建立好连接，调用 WriteMessage 即可发送消息。实际项目中建议封装连接管理、心跳机制和错误重连逻辑来提升稳定性。