在使用 code.google.com/p/go.net/websocket 构建的 WebSocket 服务器中,客户端可以接收来自服务器的通知。然而,由于长时间不活动触发的超时机制,客户端与服务器的连接可能会提前终止。
为了解决这个问题,WebSocket 协议包含了“乒乓”心跳机制。这种机制允许客户端和服务器相互发送 keep-alive 消息,防止连接被断开。
不幸的是,code.google.com/p/go.net/websocket 包没有原生支持此乒乓协议。作为一种解决方法,可以实现如下所示的自定义 ping-pong 处理程序:
import ( "time" "github.com/gorilla/websocket" ) func keepAlive(c *websocket.Conn, timeout time.Duration) { lastResponse := time.Now() c.SetPongHandler(func(msg string) error { lastResponse = time.Now() return nil }) go func() { for { err := c.WriteMessage(websocket.PingMessage, []byte("keepalive")) if err != nil { return } time.Sleep(timeout / 2) if time.Since(lastResponse) > timeout { c.Close() return } } }() }
通过将此处理程序嵌入到 WebSocket 服务器中,您可以定期向客户端发送 ping 消息并接收 pong 响应作为回报。这可以确保即使在不活动期间连接也保持活动状态。
以上是如何在 Go 中保持 WebSocket 连接处于活动状态?的详细内容。更多信息请关注PHP中文网其他相关文章!