WebSocket 连接的正确关闭

**WebSocket 连接的正确关闭** WebSocket 连接关闭时，需要正确地处理双向通信。本文讨论了编写关闭函数的注意事项，并介绍了 WebSocket 端点的关闭行为。该函数发送一个结束帧，并期望收到一个相同的响应帧。如果未收到响应帧，则函数将返回错误。测试用例表明，该函数可以正常工作，但需要注意端点可能会回显关闭消息。在正常关闭场景中，应用程序应期望在发送关闭消息后收到关闭消息。如果对端在发送关闭消息之前发送了数据消息，则应用程序应读取并丢弃数据消息，直到返回错误。

我写了一个连接关闭函数。它发送一个结束帧并期望得到相同的响应。

func tryclosenormally(wsconn *websocket.conn) error {
    closenormalclosure := websocket.formatclosemessage(websocket.closenormalclosure, "")
    defer wsconn.close()
    if err := wsconn.writecontrol(websocket.closemessage, closenormalclosure, time.now().add(time.second)); err != nil {
        return err
    }
    if err := wsconn.setreaddeadline(time.now().add(time.second)); err != nil {
        return err
    }
    _, _, err := wsconn.readmessage()
    if websocket.iscloseerror(err, websocket.closenormalclosure) {
        return nil
    } else {
        return errors.new("websocket doesn't send a close frame in response")
    }
}

我为此函数编写了一个测试。

func TestTryCloseNormally(t *testing.T) {
    done := make(chan struct{})
    exit := make(chan struct{})
    ctx := context.Background()

    ln, err := net.Listen("tcp", "localhost:")
    require.Nil(t, err)
    handler := HandlerFunc(func(conn *websocket.Conn) {
        for {
            _, _, err := conn.ReadMessage()
            if err != nil {
                require.True(t, websocket.IsCloseError(err, websocket.CloseNormalClosure), err.Error())
                return
            }
        }
    })

    s, err := makeServer(ctx, handler)
    require.Nil(t, err)
    go func() {
        require.Nil(t, s.Run(ctx, exit, ln))
        close(done)
    }()

    wsConn, _, err := websocket.DefaultDialer.Dial(addr+strconv.Itoa(ln.Addr().(*net.TCPAddr).Port), nil) //nolint:bodyclose
    require.Nil(t, err)
    require.Nil(t, wsConn.WriteMessage(websocket.BinaryMessage, []byte{'o', 'k'}))
    require.Nil(t, TryCloseNormally(wsConn))
    close(exit)

    <-done
}

令我惊讶的是，它工作正常。 readmessage() 读取结束帧。但在测试中，我什么也没写。

1. 这种情况是否发生在 gorilla/websocket 级别？
2. 我写的函数正确吗？也许读取响应帧也发生在大猩猩级别。

正确答案

该函数基本正确。

websocket 端点会回显关闭消息，除非端点已自行发送关闭消息。有关更多详细信息，请参阅 websocket rfc 中的 Closing Handshake。

在正常关闭场景中，应用程序应该期望在发送关闭消息后收到关闭消息。

为了处理对端在发送关闭消息之前发送了数据消息的情况，读取并丢弃数据消息，直到返回错误。

func TryCloseNormally(wsConn *websocket.Conn) error {
    defer wsConn.Close()
    closeNormalClosure := websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")
    if err := wsConn.WriteControl(websocket.CloseMessage, closeNormalClosure, time.Now().Add(time.Second)); err != nil {
        return err
    }
    if err := wsConn.SetReadDeadline(time.Now().Add(time.Second)); err != nil {
        return err
    }
    for {
        _, _, err := wsConn.ReadMessage()
        if websocket.IsCloseError(err, websocket.CloseNormalClosure) {
            return nil
        }
        if err != nil {
            return err
        }
    }
}

今天关于《WebSocket 连接的正确关闭》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！