Address review comments, remove redis support

README.md

@@ -4,6 +4,7 @@
 [![PyPI](https://img.shields.io/pypi/v/asyncio-connection-pool?style=flat)][package]
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/asyncio-connection-pool?style=flat)][package]
 [![codecov](https://codecov.io/gh/fellowapp/asyncio-connection-pool/graph/badge.svg?token=F3D4D9EG6M)](https://codecov.io/gh/fellowapp/asyncio-connection-pool)
+[![License](https://img.shields.io/pypi/l/prosemirror.svg?style=flat)](https://github.com/fellowapp/asyncio-connection-pool/blob/master/LICENSE.md)
 [![Fellow Careers](https://img.shields.io/badge/fellow.app-hiring-576cf7.svg?style=flat)](https://fellow.app/careers/)
 
 [main CI]: https://github.com/fellowapp/asyncio-connection-pool/actions?query=workflow%3ACI+branch%3Amain

asyncio_connection_pool/contrib/datadog.py

@@ -1,5 +1,5 @@
 from contextlib import AsyncExitStack, asynccontextmanager
-from typing import AsyncIterator, Awaitable, TypeVar
+from typing import Any, AsyncIterator, Awaitable, Coroutine, TypeVar
 
 from datadog.dogstatsd.base import statsd
 from ddtrace import tracer
@@ -85,29 +85,35 @@ def _record_connection_acquiring(self, value: int = 0) -> None:
             tags=self._extra_tags,
         )
 
-    async def _connection_maker(self) -> Conn:
+    def _connection_maker(self) -> Coroutine[Any, Any, Conn]:
         statsd.increment(
             f"{self._service_name}.pool.getting_connection",
             tags=[*self._extra_tags, "method:new"],
         )
 
-        with tracer.trace(
-            f"{self._service_name}.pool._create_new_connection",
-            service=self._service_name,
-        ):
-            return await super()._connection_maker()
+        async def connection_maker(self) -> Conn:
+            with tracer.trace(
+                f"{self._service_name}.pool._create_new_connection",
+                service=self._service_name,
+            ):
+                return await super()._connection_maker()
+
+        return connection_maker(self)
 
-    async def _connection_waiter(self) -> Conn:
+    def _connection_waiter(self) -> Coroutine[Any, Any, Conn]:
         statsd.increment(
             f"{self._service_name}.pool.getting_connection",
             tags=[*self._extra_tags, "method:wait"],
         )
 
-        with tracer.trace(
-            f"{self._service_name}.pool._wait_for_connection",
-            service=self._service_name,
-        ):
-            return await super()._connection_waiter()
+        async def connection_waiter(self) -> Conn:
+            with tracer.trace(
+                f"{self._service_name}.pool._wait_for_connection",
+                service=self._service_name,
+            ):
+                return await super()._connection_waiter()
+
+        return connection_waiter(self)
 
     def _get_conn(self) -> Awaitable[Conn]:
         if not self.available.empty():

pyproject.toml

@@ -14,7 +14,6 @@ dependencies = []
 
 [project.optional-dependencies]
 datadog = ["ddtrace", "datadog"]
-aioredis = ["redis>=4.2"]
 dev = [
     "codecov~=2.1",
     "mypy~=1.9",

test/test_pool.py

@@ -131,7 +131,7 @@ async def worker():
     assert pool.in_use == nworkers, "all workers should have their connections now"
     ev2.set()
     await asyncio.gather(*coros)
-    assert pool.currently_allocating == 0
+    assert pool.in_use == 0
     assert (
         pool.available.qsize() == nworkers
     ), "all workers should have returned their connections"