首页 > Golang > Go问答

Websocket迅速关闭

来源：stackoverflow

时间：2024-03-18 10:00:31 392浏览收藏

推广推荐

支持 PC / 移动端，安全直达

在尝试使用 Go 标准库的 Websocket 时遇到困难，转而使用 Gorilla Websocket 包。本示例代码展示了如何使用 Gorilla Websocket 包建立一个简单的回显服务器，它可以接收和回显客户端发送的消息。该服务器在端口 1234 上侦听，并使用 HTTP 升级处理 Websocket 连接。代码包括了处理消息接收和发送的逻辑，以及处理连接关闭的逻辑。

问题内容

我正在尝试这个 websocket 示例：

package main

import (
    "fmt"
    "log"
    "net/http"

    "golang.org/x/net/websocket"
)

func echo(ws *websocket.conn) {
    var err error

    for {
        var reply string

        if err = websocket.message.receive(ws, &reply); err != nil {
            fmt.println("can't receive")
            break
        }

        fmt.println("received back from client: " + reply)

        msg := "received:  " + reply
        fmt.println("sending to client: " + msg)

        if err = websocket.message.send(ws, msg); err != nil {
            fmt.println("can't send")
            break
        }
    }
}

func main() {
    http.handle("/", websocket.handler(echo))

    if err := http.listenandserve(":1234", nil); err != nil {
        log.fatal("listenandserve:", err)
    }
}

对于这个客户：

<html>
<head></head>
<body>
    <script type="text/javascript">
        var sock = null;
        var wsuri = "ws://localhost:1234";

        window.onload = function() {

            console.log("onload");

            sock = new WebSocket(wsuri);

            sock.onopen = function() {
                console.log("connected to " + wsuri);
            }

            sock.onclose = function(e) {
                console.log("connection closed (" + e.code + ")");
            }

            sock.onmessage = function(e) {
                console.log("message received: " + e.data);
            }
        };

        function send() {
            var msg = document.getElementById('message').value;
            sock.send(msg);
        };
    </script>
    <h1>WebSocket Echo Test</h1>
    <form>
        <p>
            Message: &lt;input id=&quot;message&quot; type=&quot;text&quot; value=&quot;Hello, world!&quot;&gt;
        </p>
    </form>
    <button onclick="send();">Send Message</button>
</body>
</html>

但出现以下错误：

正确答案

我在使用 std lib webscokets 时遇到了麻烦，并且（像大多数 go 用户一样）使用 gorilla websocket 包。试试这个代码（基于我的工作代码）：

package main

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

func echo(w http.ResponseWriter, r *http.Request) {
    conn, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)
    if err != nil {
        http.Error(w, "Could not open websocket connection", http.StatusBadRequest)
        return
    }
    for {
        messType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("read:", err)
            break
        }
        log.Printf("recv: %s", message)
        err = conn.WriteMessage(messType, message)
        if err != nil {
            log.Println("write:", err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/", echo)

    if err := http.ListenAndServe(":1234", nil); err != nil {
        log.Fatal("ListenAndServe:", err)
    }
}

我想这样你会得到更多的快乐。

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

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