feat:close session when send heart beat message failed

apache · Dec 13, 2024 · ce334f7 · ce334f7
1 parent eb16366
commit ce334f7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/remoting/getty/listener.go b/pkg/remoting/getty/listener.go
@@ -115,18 +115,22 @@ func (g *gettyClientHandler) OnMessage(session getty.Session, pkg interface{}) {
 
 func (g *gettyClientHandler) OnCron(session getty.Session) {
 	log.Debug("session{%s} Oncron executing", session.Stat())
-	g.transferBeatHeart(session, message.HeartBeatMessagePing)
+	err := g.transferHeartBeat(session, message.HeartBeatMessagePing)
+	if err != nil {
+		log.Errorf("failed to send heart beat: {%#v}", err.Error())
+		g.sessionManager.releaseSession(session)
+	}
 }
 
-func (g *gettyClientHandler) transferBeatHeart(session getty.Session, msg message.HeartBeatMessage) {
+func (g *gettyClientHandler) transferHeartBeat(session getty.Session, msg message.HeartBeatMessage) error {
 	rpcMessage := message.RpcMessage{
 		ID:         int32(g.idGenerator.Inc()),
 		Type:       message.GettyRequestTypeHeartbeatRequest,
 		Codec:      byte(codec.CodecTypeSeata),
 		Compressor: 0,
 		Body:       msg,
 	}
-	GetGettyRemotingInstance().SendASync(rpcMessage, session, nil)
+	return GetGettyRemotingInstance().SendASync(rpcMessage, session, nil)
 }
 
 func (g *gettyClientHandler) RegisterProcessor(msgType message.MessageType, processor processor.RemotingProcessor) {