首页 > Golang > Go问答

重新建立websocket连接的优雅方法

来源：stackoverflow

时间：2024-02-20 19:45:26 156浏览收藏

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《重新建立websocket连接的优雅方法》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

问题内容

我正在使用 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 连接，因为只要程序运行，它就应该保持连接。当用户停止该程序时，我认为它会自动关闭。

完整的 sockethandler 代码：

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
    }

    // discard received messages
    go func(c *websocket.Conn) {
        for {
            if _, _, err := c.NextReader(); err != nil {
                c.Close()
                break
            }
        }
    }(ws)

    cmd := <-sh.cmdCh

    log.Printf("Executing")
    stdout, err := cmd.StdoutPipe()
    if err != nil {
        w.Write([]byte(err.Error()))
        return
    }
    defer stdout.Close()

    stderr, err := cmd.StderrPipe()
    if err != nil {
        w.Write([]byte(err.Error()))
        return
    }
    defer stderr.Close()

    if err := cmd.Start(); err != nil {
        w.Write([]byte(err.Error()))
        return
    }

    s := bufio.NewScanner(io.MultiReader(stdout, stderr))
    for s.Scan() {
        err := ws.WriteMessage(1, s.Bytes())
        if err != nil {
            log.Printf("Error writing to client: %v", err)
            ws.Close()
        }

    }

    if err := cmd.Wait(); err != nil {
        w.Write([]byte(err.Error()))
        return
    }

    log.Println("Done")
}

解决方案

websocket 服务器应用程序必须通过关闭连接并释放与连接关联的资源来处理连接上的错误。

刷新包含页面时，浏览器会关闭 websocket 连接。浏览器关闭连接后，服务器最终会收到读取或写入错误。

连接关闭是服务器可能遇到的几种错误之一。服务器应用程序应以相同的方式处理连接上的所有错误：关闭连接并释放与该连接关联的资源。

典型的应用程序设计是让客户端在页面加载时进行连接，并在发生错误后重新连接（使用退避）。服务器假设客户端会随着时间的推移而连接和断开连接。

可以通过添加一个通过退避重新连接的 onerror 处理程序来改进 js 代码。根据应用程序的不同，您可能还希望显示指示连接状态的 ui。

go代码不会在所有场景下关闭连接。正在运行的命令是与连接关联的资源。应用程序不会在连接错误时终止该程序。以下是一些修复：

升级成功后添加defer ws.close()。从 sockethandler.servehttp 中删除对 ws.close() 的其他直接调用。这确保了在所有场景中都会调用 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()

我尚未运行或测试此代码。可能有些细节是错误的，但这概述了关闭连接和释放资源的一般方法。

看一下 Gorilla command example。该示例展示了如何将 stdin 和 stdout 泵送到 websocket。该示例还处理更高级的功能，例如使用 ping/pong 检查连接的运行状况。

终于介绍完啦！小伙伴们，这篇关于《重新建立websocket连接的优雅方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！

声明：本文转载于：stackoverflow 如有侵犯，请联系study_golang@163.com删除