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

你可能感兴趣的:(goSocket)