FIxed bug with client disconnection.

3b25b83c · Antoine Millet · b00b7ee3 · 3b25b83c
Commit 3b25b83c authored Jan 31, 2011 by Antoine Millet
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

sjrpc/server/simple.py sjrpc/server/simple.py +6 -2

No files found.
--- a/sjrpc/server/simple.py
+++ b/sjrpc/server/simple.py
@@ -45,8 +45,12 @@ class SimpleRpcServer(ConnectionManager):
        conn = self._clients.get(fd)
        self._poll.unregister(fd)
        if fd is not None:
-            del self._clients[fd]
-            conn.shutdown(callback=self._on_disconnect)
+            try:
+                del self._clients[fd]
+            except KeyError:
+                pass
+            if conn is not None:
+                conn.shutdown(callback=self._on_disconnect)

    def all_connections(self):
        return set(self._clients.values())