add active/idle only if it exists

lqs · Apr 4, 2024 · 8ec7984 · 8ec7984
1 parent 5f77cae
commit 8ec7984
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server.go b/server.go
@@ -118,10 +118,12 @@ func (s *server) updateConnState(conn net.Conn, state http.ConnState) {
 	switch state {
 	case http.StateNew:
 		s.connStateMap[conn] = state
+	case http.StateActive, http.StateIdle:
+		if _, ok := s.connStateMap[conn]; ok {
+			s.connStateMap[conn] = state
+		}
 	case http.StateHijacked, http.StateClosed:
 		delete(s.connStateMap, conn)
-	default:
-		// ignore other states
 	}
 }