首页  >  文章  >  后端开发  >  如何在 Go 中保持 WebSocket 连接处于活动状态?

如何在 Go 中保持 WebSocket 连接处于活动状态?

Susan Sarandon
Susan Sarandon原创
2024-11-03 19:16:29348浏览

How to Keep WebSocket Connections Alive in Go?

在 Go 中维护 WebSocket 连接

在使用 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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn