fix HeartbeatCodec may be make so many goroutine

snail007 · Feb 3, 2025 · b425e4f · b425e4f
1 parent 0bf1bbb
commit b425e4f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/util/net/heartbeat_codec.go b/util/net/heartbeat_codec.go
@@ -260,9 +260,9 @@ func (s *HeartbeatCodec) Write(b []byte) (n int, err error) {
 	}
 	p.Submit(func() {
 		defer func() {
-			close(done)
 			//put msg buffer back to pool
 			msg.PutBackBytesBuf()
+			close(done)
 		}()
 		n, err = s.Conn.Write(msg.Bytes())
 		if err == nil {
@@ -285,6 +285,7 @@ func (s *HeartbeatCodec) Close() (err error) {
 		err = s.Conn.Close()
 		s.readPool.Stop()
 		s.writePool.Stop()
+		s.writeHbPool.Stop()
 	})
 	return
 }