• 博客园logo
  • 会员
  • 周边
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录

0x1cc4

Youth means limitless possibilities. 年轻就是无限的可能。
  • 订阅
  • 管理

公告

View Post

go+gin webscoket 怎么实现心跳

在Go+Gin中实现WebSocket的心跳,你需要一个WebSocket库,如gorilla/websocket。下面的例子展示了如何实现心跳机制:

package main

import (
    "net/http"
    "time"

    "github.com/gin-gonic/gin"
    "github.com/gorilla/websocket"   
)

// 升级器,用于将HTTP连接升级为WebSocket连接
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func websocketEndpoint(c *gin.Context) {
    // 升级HTTP连接为WebSocket
    conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
    if err != nil {
        http.Error(c.Writer, "Could not open websocket connection", http.StatusBadRequest)
    }

    go func() {
        // 每30秒发送一个ping消息到客户端来检测连接是否正常
        for {
            err := conn.WriteMessage(websocket.PingMessage, []byte{})
            if err != nil {
                return
            }

            time.Sleep(30 * time.Second)
        }
    }()

    for {
        // 处理来自客户端的消息
        _, _, err := conn.ReadMessage()
        if err != nil {
            conn.Close()
            break
        }
        // ...
    }
}

func main() {
    router := gin.Default()

    router.GET("/ws", websocketEndpoint)

    router.Run(":8080")
}

以上代码实现了一个简单的WebSocket服务端,客户端需要通过/ws路径来连接此WebSocket服务。
这个服务器会不间断地每30秒发送一次ping消息到每个连接的客户端,作为心跳。如果写入消息(发送心跳)失败,那么就终止这个心跳发送协程,因为这可能意味着客户端已经断开连接。
并且这个服务器也会不停地去读取来自客户端的信息,如果读取失败,那么会关闭连接,这也可能是因为客户端已经断开连接。
不过注意,这个例子假设客户端会对心跳ping进行正确的回应(发送一个pong回来)。如果客户端不这么做的话,可能需要另外的机制来检测连接是否终止。

本文来自博客园,作者:0x1cc4,转载请注明原文链接:https://www.cnblogs.com/0x1cc4/p/18087819

posted on 2024-03-21 17:14  0x1cc4  阅读(428)  评论(0)    收藏  举报

刷新页面返回顶部
 
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3