Handle socket, IO, and EOF errors when opening a new connection.

pycassa · Mar 9, 2011 · c9e3b93 · c9e3b93
1 parent 2ff5a41
commit c9e3b93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycassa/pool.py b/pycassa/pool.py
@@ -121,7 +121,7 @@ def _create_connection(self):
                 server = self._get_next_server()
                 wrapper = self._get_new_wrapper(server)
                 return wrapper
-            except Thrift.TException, exc:
+            except (Thrift.TException, socket.error, IOError, EOFError), exc:
                 self._notify_on_failure(exc, server)
                 failure_count += 1
         raise AllServersUnavailable('An attempt was made to connect to each of the servers '