Gorilla WebSocket WriteMessage网络断开无报错，数据丢失如何解决？

这是TCP协议的特性导致的——WriteMessage成功仅代表数据写入了本地TCP发送缓冲区，而非已被对方接收。当路由丢失这类静默断连发生时，TCP不会立刻检测到连接失效，直到保活机制触发或超时。以下是几种可行的解决办法：

1. 设置写超时 #

每次调用WriteMessage前设置写超时，确保数据在指定时间内无法发送时能返回错误：

import "time"

// 发送前设置5秒写超时
err := conn.SetWriteDeadline(time.Now().Add(5 * time.Second))
if err != nil {
    // 处理设置超时错误
    return err
}

// 发送消息
err = conn.WriteMessage(websocket.TextMessage, []byte("your data"))
if err != nil {
    // 捕获发送超时或连接失效错误
    conn.Close()
    return err
}

原理：超时后操作系统会终止未完成的发送操作，Gorilla库会将这个错误返回给调用方。

2. 实现应用层心跳检测 #

主动发送ping帧并等待pong响应，主动验证连接可用性：

import (
    "time"
    "github.com/gorilla/websocket"
)

func heartbeat(conn *websocket.Conn, pingInterval time.Duration) {
    ticker := time.NewTicker(pingInterval)
    defer ticker.Stop()

    for {
        select {
        case <-ticker.C:
            // 设置读超时，等待pong响应
            err := conn.SetReadDeadline(time.Now().Add(pingInterval * 2))
            if err != nil {
                conn.Close()
                return
            }

            // 发送ping帧
            if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
                conn.Close()
                return
            }

            // 等待pong响应
            _, _, err = conn.ReadMessage()
            if err != nil {
                conn.Close()
                return
            }
        }
    }
}

// 启动心跳goroutine
go heartbeat(conn, 30*time.Second)

原理：如果连接已断，发送ping会失败；如果没收到pong，读操作会超时，从而及时检测到连接失效。

3. 启用TCP层保活 #

让操作系统自动检测死连接，通过Gorilla的Dialer配置TCP保活参数：

import (
    "net"
    "time"
    "github.com/gorilla/websocket"
)

dialer := websocket.Dialer{
    NetDialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
        conn, err := net.Dial(network, addr)
        if err != nil {
            return nil, err
        }
        // 转换为TCP连接并设置保活
        tcpConn, ok := conn.(*net.TCPConn)
        if ok {
            tcpConn.SetKeepAlive(true)
            // 每30秒发送一次保活探测
            tcpConn.SetKeepAlivePeriod(30 * time.Second)
        }
        return conn, err
    },
}

// 建立连接
conn, _, err := dialer.Dial("ws://your-server-url", nil)
if err != nil {
    // 处理连接错误
    return err
}

原理：TCP保活会定期发送探测包，若对方无响应，操作系统会标记连接为失效，后续的WriteMessage会返回错误。

4. 关键数据的确认机制 #

对于不能丢失的关键数据，要求服务端收到后返回确认消息，客户端未收到确认则重试或判定连接失效：

// 发送关键数据
err := conn.WriteMessage(websocket.TextMessage, []byte("critical data"))
if err != nil {
    conn.Close()
    return err
}

// 设置读超时等待确认
err = conn.SetReadDeadline(time.Now().Add(10 * time.Second))
if err != nil {
    conn.Close()
    return err
}

// 读取服务端的确认消息
msgType, msg, err := conn.ReadMessage()
if err != nil || msgType != websocket.TextMessage || string(msg) != "ACK" {
    // 未收到确认，处理错误（重发/关闭连接）
    conn.Close()
    return err
}

原理：通过应用层的双向确认，确保数据确实被服务端接收，避免静默丢失。

内容的提问来源于stack exchange，提问作者nerd