在 github.com/gorilla/websocket 中，您可以使用 PingHandler 函数来实现 WebSocket 的 Ping 机制。以下是一个示例代码：
package main
import (
"log"
"net/http"
"time"
"github.com/gorilla/websocket"
)
var (
upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
pingPeriod = 10 * time.Second
)
func pingHandler(conn *websocket.Conn) {
ticker := time.NewTicker(pingPeriod)
defer ticker.Stop()
for {
select {
case <-ticker.C:
err := conn.WriteControl(websocket.PingMessage, nil, time.Now().Add(time.Second))
if err != nil {
log.Println("Failed to send ping:", err)
return
}
}
}
}
func websocketHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Failed to upgrade connection:", err)
return
}
defer conn.Close()
go pingHandler(conn)
for {
// 处理其他的消息逻辑，如读取和处理客户端发送的消息等
}
}
func main() {
http.HandleFunc("/ws", websocketHandler)
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
在这个例子中，pingHandler 函数会定期地向客户端发送 Ping 消息。您可以根据自己的需求调整 pingPeriod 来设置 Ping 的时间间隔。请注意，此代码仅提供了实现 Ping 机制的基本示例，您可能需要根据自己的应用场景进行修改和完善。
内容由零声教学AI助手提供，问题来源于学员提问
