fix: 修复普通heartbeat以及ws-heartbeat的return问题

middleware/heartbeat.go

@@ -28,20 +28,7 @@ func WithHeartbeat(interval time.Duration, hbdFn func(conn *conn.Connection) []b
 		panic("dataFn must not be nil")
 	}
 
-	return func(ngin *nnet.Engine) {
-		ngin.EventManager().RegisterEvent(event.EvtOnConnected, m.start)
-
-		ngin.Use(func(next rt.HandlerFunc) rt.HandlerFunc {
-			return func(conn *conn.Connection, pkg packet.IPacket) {
-				m.handle(conn, pkg)
-
-				next(conn, pkg)
-			}
-		})
-	}
-}
-
-func (m *HeartbeatMiddleware) start(conn *conn.Connection) {
+	var startFn event.OnConnectedFn = func(nc *conn.Connection) {
 		ticker := time.NewTicker(m.interval)
 
 		defer func() {
@@ -54,9 +41,9 @@ func (m *HeartbeatMiddleware) start(conn *conn.Connection) {
 				deadline := time.Now().Add(-2 * m.interval).Unix()
 				if atomic.LoadInt64(&m.lastAt) < deadline {
 					nlog.Errorf("Heartbeat timeout, LastTime=%d, Deadline=%d", atomic.LoadInt64(&m.lastAt), deadline)
-				return
+					break
 				}
-			err := conn.SendBytes(m.hbdFn(conn))
+				err := nc.SendBytes(m.hbdFn(nc))
 				if err != nil {
 					nlog.Errorf("Heartbeat err: %v", err)
 					return
@@ -65,6 +52,19 @@ func (m *HeartbeatMiddleware) start(conn *conn.Connection) {
 		}
 	}
 
+	return func(ngin *nnet.Engine) {
+		ngin.EventManager().RegisterEvent(event.EvtOnConnected, startFn)
+
+		ngin.Use(func(next rt.HandlerFunc) rt.HandlerFunc {
+			return func(conn *conn.Connection, pkg packet.IPacket) {
+				m.handle(conn, pkg)
+
+				next(conn, pkg)
+			}
+		})
+	}
+}
+
 func (m *HeartbeatMiddleware) handle(_ *conn.Connection, _ packet.IPacket) {
 	atomic.StoreInt64(&m.lastAt, time.Now().Unix())
 }

middleware/heartbeat_ws.go

@@ -48,7 +48,7 @@ func WithHeartbeatWS(interval time.Duration, hbdFn WsHeartbeatFn) nnet.RunOption
 				deadline := time.Now().Add(-2 * m.interval).Unix()
 				if atomic.LoadInt64(&m.lastAt) < deadline {
 					nlog.Errorf("Heartbeat timeout, LastTime=%d, Deadline=%d", atomic.LoadInt64(&m.lastAt), deadline)
-					return
+					break
 				}
 
 				err := m.hbdFn(nc.WsConn())