Pārlūkot izejas kodu

Merge pull request #22 from zlk1983/patch-1

Update connection.py
Kaifeng Xu 5 gadi atpakaļ
vecāks
revīzija
dacc9a8355
1 mainītis faili ar 4 papildinājumiem un 2 dzēšanām
  1. 4 2
      fdfs_client/connection.py

+ 4 - 2
fdfs_client/connection.py

@@ -107,11 +107,11 @@ class ConnectionPool(object):
         """Create a new connection."""
         """Create a new connection."""
         if self._conns_created >= self.max_conn:
         if self._conns_created >= self.max_conn:
             raise ConnectionError('[-] Error: Too many connections.')
             raise ConnectionError('[-] Error: Too many connections.')
-        num_try = 10
+        num_try = 3
         while True:
         while True:
             try:
             try:
                 if num_try <= 0:
                 if num_try <= 0:
-                    sys.exit()
+                    break
                 conn_instance = self.conn_class(**self.conn_kwargs)
                 conn_instance = self.conn_class(**self.conn_kwargs)
                 conn_instance.connect()
                 conn_instance.connect()
                 self._conns_created += 1
                 self._conns_created += 1
@@ -120,6 +120,8 @@ class ConnectionPool(object):
                 print(e)
                 print(e)
                 num_try -= 1
                 num_try -= 1
                 conn_instance = None
                 conn_instance = None
+        if num_try <= 0:
+            raise ConnectionError("Fail to connect with Fdfs-server after trying 3 times")
         return conn_instance
         return conn_instance
 
 
     def get_connection(self):
     def get_connection(self):