Handle errors properly, with pool size logic

aio-libs · Aug 5, 2016 · ffae551 · ffae551
1 parent 72da685
commit ffae551
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/aiomcache/pool.py b/aiomcache/pool.py
@@ -85,8 +85,12 @@ def release(self, conn):
     def _create_new_conn(self):
         if self._size < self._maxsize:
             self._size += 1
-            reader, writer = yield from asyncio.open_connection(
-                self._host, self._port, loop=self._loop)
+            try:
+                reader, writer = yield from asyncio.open_connection(
+                    self._host, self._port, loop=self._loop)
+            except:
+                self._size -= 1
+                raise
             return _connection(reader, writer)
         else:
             return None