[fix] BasicSessionStat对连接做非空检查，防止panic

q191201771 · Apr 18, 2024 · b8cbe76 · b8cbe76
1 parent d4c38c1
commit b8cbe76
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/base/basic_session_stat.go b/pkg/base/basic_session_stat.go
@@ -130,6 +130,10 @@ func (s *BasicSessionStat) GetStat() StatSession {
 }
 
 func (s *BasicSessionStat) GetStatWithConn(conn IStatable) StatSession {
+	if conn == nil {
+		return s.stat
+	}
+
 	connStat := conn.GetStat()
 	s.stat.ReadBytesSum = connStat.ReadBytesSum
 	s.stat.WroteBytesSum = connStat.WroteBytesSum
@@ -141,6 +145,10 @@ func (s *BasicSessionStat) IsAlive() (readAlive, writeAlive bool) {
 }
 
 func (s *BasicSessionStat) IsAliveWitchConn(conn IStatable) (readAlive, writeAlive bool) {
+	if conn == nil {
+		return s.isAlive(0, 0)
+	}
+
 	currStat := conn.GetStat()
 	return s.isAlive(currStat.ReadBytesSum, currStat.WroteBytesSum)
 }