如何优雅地重新连接websocket?

我正在使用 HTTP 和 Websockets 构建一个单页面应用程序。用户提交表单,我将响应流式传输给客户端。下面是一个片段客户端。


var html = `<!DOCTYPE html>


<meta charset="utf-8">

<head>

</head>

<body>


<script>

var ws = new WebSocket("ws://localhost:8000/ws")

ws.onmessage = function(e) {

  document.getElementById("output").innerHTML += e.data + "<br>"

}


function submitFunction() {

  document.getElementById("output").innerHTML += ""

  return false

}

</script>


<form

enctype="multipart/x-www-form-urlencoded"

action="http://localhost:8000/"

method="post"

>`

这是服务器。如果请求不是 POST,我会编写/渲染 html (parseAndExecute),以建立新的 websocket 连接。如果请求是 POST(来自表单),那么我开始处理并最终写入 websocket。


func (c *Config) ServeHtml(w http.ResponseWriter, r *http.Request) {

    if r.Method == http.MethodPost {

        //process

        channel <- data

    }


    c.parseAndExecute(w)

}


func (sh *SocketHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {

    ws, err := upgrader.Upgrade(w, r, nil)

    if err != nil {

        w.Write([]byte(fmt.Sprintf("", err)))

        return

    }

    //defer ws.Close()


    // discard received messages

    go func(c *websocket.Conn) {

        for {

            if _, _, err := c.NextReader(); err != nil {

                c.Close()

                break

            }

        }

    }(ws)


    data <- channel


仅当我不刷新页面时,一切都会按我的预期进行。如果我不刷新,我可以继续提交表单并看到不同的输出逐行出现。澄清一下,它实际上只有在页面已经启动时才有效,因此parseAndExecute永远不会被调用。该函数解析并执行 html/template 创建一个新的 websocket 客户端。


任何页面刷新或最初浏览 localhost:8000 都会websocket: close sent在服务器上引起。


我不知道如何解决这个问题。服务器是否需要优雅地处理断开连接并允许重新连接?或者客户需要做些什么?似乎服务器应该升级任何连接,/ws因此创建多少个新的 websocket 客户端并不重要,但显然我的理解是错误的。


我不会关闭服务器上的 websocket 连接,因为只要程序运行,它就应该保持连接状态。当用户停止该程序时,我认为它会自动关闭。


慕姐4208626
浏览 139回答 1
1回答

慕后森

Websocket 服务器应用程序必须通过关闭连接并释放与该连接关联的资源来处理连接上的错误。当包含的页面刷新时,浏览器会关闭 websocket 连接。浏览器关闭连接后,服务器最终会收到读取或写入错误。连接关闭是服务器可能遇到的几种错误之一。服务器应用程序应以相同的方式处理连接上的所有错误:关闭连接并释放与该连接关联的资源。典型的应用程序设计是让客户端在页面加载时进行连接,并在发生错误后重新连接(使用退避)。服务器假设客户端会随着时间的推移而连接和断开连接。可以通过添加一个通过退避重新连接的 onerror 处理程序来改进 JS 代码。根据应用程序的不同,您可能还希望显示指示连接状态的 UI。Go代码并不会在所有场景下关闭连接。正在运行的命令是与连接关联的资源。应用程序不会在连接错误时终止该程序。以下是一些修复:defer ws.Close()升级成功后添加。ws.Close()删除对from的其他直接调用SocketHandler.ServeHTTP。这确保了ws.Close()在所有场景中都被调用。从读写泵退出时终止该命令。命令启动后将读取泵移至。回来就杀掉。go func(c *websocket.Conn, cmd *exec.Command) {    defer c.Close()    defer cmd.Process.Kill()    for {        if _, _, err := c.NextReader(); err != nil {            break        }    }}(ws, cmd)从写泵退出时终止命令:s := bufio.NewScanner(io.MultiReader(stdout, stderr))for s.Scan() {    err := ws.WriteMessage(1, s.Bytes())    if err != nil {        break    }}cmd.Process.Kill()我还没有运行或测试过这段代码。有些细节可能是错误的,但这概述了关闭连接和释放资源的一般方法。
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go