Golang Fiber + Socket.IO：WebSocket陷入无限连接-断开循环，断开时无法正确标记访客离线

嘿，我目前在用Golang Fiber搭的Web服务器结合Socket.IO来处理WebSocket连接，现在遇到个头疼的问题：我想在Socket断开连接时把数据库里对应访客的Online字段设为false，但现在WebSocket一连接就立刻断开，还陷入了无限循环，根本没法正常处理离线标记的逻辑。

我当前的Socket.IO处理代码片段 #

socketio.On(socketio.EventConnect, func(ep *socketio.EventPayload) {
    guest_id := ep.Kws.GetStringAttribute("guest_id")
    guest, err := ge... // 这里是我后续查询访客信息、标记在线的逻辑，但代码没写完就出问题了
}

我折腾了好一阵都没搞定，下面是我梳理的排查方向和解决思路，应该能帮到同样踩坑的人：

一、先搞定无限连接循环的问题 #

这种循环大概率是「连接成功后触发错误导致断开→客户端自动重连→再次触发错误断开」的死循环，得先把这个破了：

• 给连接事件加完整的错误捕获：别让未处理的错误搞崩连接，比如获取不到guest_id、数据库查询失败时，主动优雅断开并打日志，别让连接异常中断触发客户端重连：

  socketio.On(socketio.EventConnect, func(ep *socketio.EventPayload) {
      guest_id := ep.Kws.GetStringAttribute("guest_id")
      // 先校验guest_id是否存在
      if guest_id == "" {
          log.Println("连接失败：未传递合法的guest_id")
          ep.Kws.Close()
          return
      }
      // 查询访客信息
      guest, err := getGuestByID(guest_id)
      if err != nil {
          log.Printf("查询访客%s信息失败：%v", guest_id, err)
          ep.Kws.Close()
          return
      }
      // 标记访客在线的数据库操作
      err = updateGuestOnlineStatus(guest.ID, true)
      if err != nil {
          log.Printf("标记访客%s在线失败：%v", guest_id, err)
          ep.Kws.Close()
          return
      }
      // 把guest_id存在Socket属性里，方便断开时直接取
      ep.Kws.SetAttribute("guest_id", guest_id)
      log.Printf("访客%s已连接并标记为在线", guest_id)
  })
  

• 检查客户端重连逻辑：如果客户端配置了自动重连，那得确保只有合法连接才保留，非法连接断开后让客户端停止重连（或者在客户端加判断，比如收到错误提示后不再重连）。

二、正确处理断开事件，实现离线标记 #

解决了循环问题，接下来要确保断开时能正确标记离线，核心是监听EventDisconnect事件，并且能拿到访客ID：

• 监听断开事件并处理：从之前存在Socket属性里的guest_id取数据，执行离线标记：

  socketio.On(socketio.EventDisconnect, func(ep *socketio.EventPayload) {
      // 从Socket属性里获取guest_id
      guest_id, ok := ep.Kws.GetAttribute("guest_id").(string)
      if !ok || guest_id == "" {
          log.Println("断开连接时无法获取有效访客ID")
          return
      }
      // 执行标记离线的数据库操作
      err := updateGuestOnlineStatus(guest_id, false)
      if err != nil {
          log.Printf("标记访客%s离线失败：%v", guest_id, err)
          return
      }
      log.Printf("访客%s已断开连接并标记为离线", guest_id)
  })
  

• 补充心跳机制：如果是网络中断这类异常断开，EventDisconnect可能不会及时触发，建议加个心跳检测：定期给客户端发心跳包，超过一定时间没回应就主动标记访客为离线，同时清理无效连接。

三、额外排查点 #

• 检查Fiber和Socket.IO的版本是否兼容，版本不匹配很容易出现各种连接异常；
• 打开服务器的详细日志，看看连接、断开时有没有未捕获的panic或错误，这些往往是问题的关键；
• 测试客户端传递的guest_id是否正确，比如有没有在连接时通过query参数或属性正确传递到服务器。

备注：内容来源于stack exchange，提问作者Vidy45agarvk