Merge pull request #108 from takewofly/fix_on_message

fix: when session is closed, no need do OnMessage
apache · Mar 16, 2024 · 71b8185 · 71b8185
2 parents 93d91b4 + 582f4ca
commit 71b8185
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/session.go b/session.go
@@ -562,6 +562,11 @@ func (s *session) run() {
 
 func (s *session) addTask(pkg interface{}) {
 	f := func() {
+		// If the session is closed, there is no need to perform CPU-intensive operations.
+		if s.IsClosed() {
+			log.Errorf("[Id:%d, name=%s, endpoint=%s] Session is closed", s.ID(), s.name, s.EndPoint())
+			return
+		}
 		s.listener.OnMessage(s, pkg)
 		s.IncReadPkgNum()
 	}