如何保持 golang.org/x/net/websocket 打开？

Question

我天真地使用_, err := ws.Read(msg)在我的代码中保持 Web 套接字打开。我不认为它可靠地工作。在其他代码中，我注意到人们在做sleep loop。保持 Web 套接字打开和稳定的正确方法是什么？我假设底层库会进行乒乓/乒乓？更新：我非常确信client.go是罪魁祸首，因为在服务器上看到断开连接后它似乎没有重拨。我制作了一个视频来演示这个问题。

Answer 1

golang.org/x/net/websocket 没有实现 RFC 6455 中的 ping-pong，但是 gorilla websocket实现了。Gorilla websocket 的最小示例

c := time.Tick(pingInterval)

go func(){

       for  _ := range c {

           if err := conn.WriteControl(Ping...); err != nil {

                 handle error 

            }

      }

   }()

          -----

conn.SetPongHandler(func(string) error { return conn.SetReadDeadline(time.Now().Add(pingInterval)) })

go func(){

      for {

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

                handle error

           }

       }

    }()

Answer 2

我误读了你原来的问题。

不，你马上就做。您基本上需要阻止处理程序返回以保持 websocket 连接活动。如果您不关心该消息，只需将其丢弃并且不对其进行任何处理。

一种常见的方式人们做的是有一个专门的Read和Write够程，是这样的：

func fishHandler(ws *websocket.Conn) {

    id := ws.RemoteAddr().String() + "-" + ws.Request().RemoteAddr + "-" + ws.Request().UserAgent()

    sockets[id] = ws

    go writePump(ws)

    readPump(ws)

}

func readPump(ws *websocket.Conn) {

    defer ws.Close()

    msg := make([]byte, 512)

    _, err := ws.Read(msg)

    if err != nil {

        return

    }

}

func writePump(ws *websocket.Conn) {

    // Hand the write messages here

}