Check for closed connections before select()'ing.

James William Pye · James William Pye · commit 4ed896803534 · 2011-09-10T10:52:45.000-05:00
diff --git a/postgresql/notifyman.py b/postgresql/notifyman.py
@@ -71,6 +71,11 @@ def _wait_on_wires(self, time = time, select = select):
 
 		# Connections already marked as "bad" should not be checked.
 		check = self.connections - self.garbage
+		for db in check:
+			if db.closed:
+				self.connections.remove(db)
+				self.garbage.add(db)
+		check = self.connections - self.garbage
 
 		r, w, x = select(check, (), check, max_duration)
 		# Make sure the connection's _notifies get filled.
