当我导航到其他页面时，获取 WebSocket 错误关闭发送

Question

当我尝试将数据从服务器端发送到客户端（仪表板页面）时，我收到错误 websocket: close sent。当我导航到主页并返回仪表板页面时发生错误。 仪表板页面最初一切正常
我的仪表板页面代码

  let socket = new ReconnectingWebSocket("ws://127.0.0.1:8004/wsendpoint");
        console.log("Attempting Connection...");

        socket.onopen = () => {
            console.log("Successfully Connected");
          
        };
        socket.onclose = event => {
            console.log("Socket Closed Connection: ", event);
         
        };
        socket.onerror = error => {
            console.log("Socket Error: ", error);
         
        };
        socket.onmessage = function (event) {
            console.log("message received: " + event.data);
        }

我的服务器端代码（作者和读者

func wsEndpoint(w http.ResponseWriter, r *http.Request) {

    upgrader.CheckOrigin = func(r *http.Request) bool { return true }
    // upgrade this connection to a WebSocket connection
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("upgrade error %s", err)
    }

    defer ws.Close()

    go writer(ws) 
    go reader(ws)
}

Writer，当有新数据从数据库进来时，从通道中获取数据

func writer(conn *websocket.Conn) {

    for {
        singleset := <-singleorder // get data from channel
        jsonString, err := json.Marshal(singleset ) 
        err = conn.WriteMessage(1, []byte(jsonString))
        if err != nil {
            log.Println(err)
        }
    }

Reader，从客户端读取数据

func reader(conn *websocket.Conn) {
    for {
        _, p, err := conn.ReadMessage() //what is message type?
        if err != nil {
            log.Println("there is errors%s", err)
            return
        }
    }
}

我也收到错误 WebSocket 在建立连接之前关闭。 和 reconnecting-websocket.min.js:1 WebSocket 连接到 'ws://127.0.0.1:8004/ wsendpoint' failed: WebSocket 在连接建立之前关闭。

正如您所见，所有代码都非常简单，因为我刚开始学习 websocket 并遵循简单的教程。我试图在网上搜索，但 webworker 对我来说似乎有点复杂，并且看到了 ping pong 方法，但我不确定如果我导航它是否仍然有效。我可以在主页上建立 websocket 这样连接就不会关闭吗？因为我在客户端只有两个页面。

提前感谢有关如何处理这些情况的任何指导！

Answer 1

处理离开页面、页面关闭或错误的基本策略是：客户端在页面加载时创建 websocket 连接。服务器期望客户端连接来来去去，并在客户端连接出错时清理资源。

这就是您收到错误“websocket: close sent”的原因：当用户导航离开页面时，浏览器会向服务器发送一个close message 作为websocket closing handshake 的一部分。 Gorilla 包通过向客户端发送另一个关闭消息来响应关闭消息。从那时起，服务器无法在连接上发送消息。发送关闭消息后，连接从写入方法返回错误。

关闭消息是returned as errors from the websocket read methods。要解决此问题，请修改代码以处理一般错误。无需专门处理结束握手。

这是更新后的代码。 wsEndpoint 函数为读取器创建一个通道，以向写入器发出读取器已完成的信号。删除了defer ws.Close() 语句，因为读取器和写入器 goroutine 将负责关闭连接。

func wsEndpoint(w http.ResponseWriter, r *http.Request) {

    upgrader.CheckOrigin = func(r *http.Request) bool { return true }
    // upgrade this connection to a WebSocket connection
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("upgrade error %s", err)
    }

    done := make(chan struct{})     
    go writer(ws, done) 
    go reader(ws, done)
}

阅读器在返回时关闭连接。它还会关闭done 频道以向写入者发出读取完成的信号。

func reader(conn *websocket.Conn, done chan struct{}) {
    defer conn.Close()
    defer close(done)
    for {
        _, p, err := conn.ReadMessage() //what is message type?
        if err != nil {
            log.Println("there is errors%s", err)
            return
        }
    }
}

编写器在返回时也会关闭连接。当连接关闭时，连接上的读取立即返回一个错误，导致读取 goroutine 完成。作者在done 频道上等待。当阅读器关闭通道时，通道上的接收会产生零值。 writer 在写入错误时返回，而不是像问题中那样永远循环。

func writer(conn *websocket.Conn, done chan struct{}) {
    defer conn.Close()
    for {
        select {
        case <-done:
            // the reader is done, so return
            return
        case singleset := <-singleorder: // get data from channel
            jsonString, err := json.Marshal(singleset ) 
            err = conn.WriteMessage(1, []byte(jsonString))
            if err != nil {
                log.Println(err)
                return
            }
        }
    }
}

应用程序应该期望连接因多种原因而关闭，包括用户离开页面。

很可能发送到通道singleorder 的代码需要知道连接已关闭，但我们在这里看不到该代码。我会留给你弄清楚如何处理。