go语言socket框架之心跳处理

前言:

本次我们来讲解怎么在自己的框架里面增加心跳处理。已经把所有代码整合了,希望给个星星支持一下 microSocket

心跳处理的必要性:

服务端需要同时处理上千甚至上万的客户端的连接,所以每个连接资源都是很宝贵的,当客户端断开连接的时候服务端应该及时移除该连接。
正常情况下,客户端断开连接的时候,会和服务端进行四次挥手,服务端就会知道这个连接 已经不能用了优雅的退出监听消息。但是总会有意外,比如客户端忽然断网了,没电了,这个时候客户端肯定不可能按照流程和 服务端进行挥手,不知道消息的 服务端还傻傻的在哪儿等着,不知道客户端早就走了
这个时候 心跳包就很完美的解决了此问题。客户端和服务端约定好每隔一段时间 就会发消息,如果服务端每过一段时间没有收到客户端 的心跳消息 就说明 客户端出事了,服务端就删除此连接,确保 资源最大化。
一般心跳包就是 符合该协议的 最小包

实现思路:

一般的go语言框架都是一个连接单独开一个协程 去处理读取超时问题,我承认协程是很廉价,但是总不至于这么浪费吧,当协程数量多到一定程度的时候,协程之间的调度也是一个很大的 消耗,所以我没有采用这种思路。
经过我的苦思冥想终于被我想到了自认为比较好的方法,实现了,一个协程 进行 心跳检测。

  • . 每当一个连接成功接收到消息的时候,就在该连接对象 上设置当前时间戳。来保存最近一次接收消息的时间
  • . 在框架启动的时候就开启一个协程 ,每隔一段时间就遍历 当前所有连接对象 ,如果当前时间 减去 连接对象里的最近接收时间 超过心跳时间,就说明 该连接 已经死了就执行删除

代码实现:

//这是每个连接对象 每次接收到消息就会更新times为当前时间戳
type Session struct {
    Id    uint32
    Con   net.Conn
    times int64
    lock  sync.Mutex
}
//这是更新时间函数
func (this *Session)UpdateTime(){
    this.times = time.Now().Unix()
}
//---------------------------------------------------SESSION管理类------------------------------------------------------

type SessionM struct {
    sessions map[uint32]*Session
    num      uint32
    lock     sync.RWMutex
    isWebSocket bool
    ser     *Msf
}
//心跳检测   每秒遍历一次 查看所有sess 上次接收消息时间  如果超过 num 就删除该 sess
func (this *SessionM)HeartBeat(num int64){
    for {
        time.Sleep(time.Second)
        for i,v:= range this.sessions{
            if time.Now().Unix() - v.times > num {
                this.DelSessionById(i)
            }
        }
    }
}

当框架启动的时候就开一个协程

go this.SessionMaster.HeartBeat(2)

每次接收到新的消息的时候就更新接收时间

//更新接收时间
sess.UpdateTime()

以上是核心代码。细节可以看 microSocket

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容