在尝试使用 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)
    }
}

对于这个客户：

    
    
WebSocket Echo Test
    

        

            Message: 
        
    
    Send Message

但出现以下错误：

正确答案

我在使用 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学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！