adding key count to redis info

Author: samuelcolvin

HISTORY.rst

@@ -5,8 +5,9 @@ History
 
 v0.10.2 (2017-XX-XX)
 ....................
-* only save job on task in drain if reenqueing
+* only save job on task in drain if re-enqueuing
 * add semaphore timeout to drains
+* add key count to ``log_redis_info``
 
 v0.10.1 (2017-08-16)
 ....................

arq/utils.py

@@ -107,11 +107,12 @@ async def get_redis_conn(self):
     async def log_redis_info(self, log_func):
         pool = await self.get_redis_pool()
         async with pool.get() as redis:
-            info = await redis.info()
+            info, key_count = await asyncio.gather(redis.info(), redis.dbsize())
             log_func(
                 f'redis version: {info["server"]["redis_version"]}, '
                 f'mem. usage: {info["memory"]["used_memory_human"]}, '
-                f'clients connected: {info["clients"]["connected_clients"]}'
+                f'clients connected: {info["clients"]["connected_clients"]}, '
+                f'keys in db: {key_count}'
             )
 
     async def close(self):

tests/test_utils.py

@@ -75,8 +75,12 @@ async def test_redis_success_log(loop, caplog):
     await pool.wait_closed()
 
 
-async def test_redis_log(loop, caplog):
+async def test_redis_log(loop):
     r = RedisMixin(loop=loop)
+    async with await r.get_redis_conn() as redis:
+        await redis.flushall()
+        await redis.set(b'a', b'1')
+        await redis.set(b'b', b'2')
 
     log_msgs = []
 
@@ -88,6 +92,7 @@ def _log(s):
     print(log_msgs)
     assert len(log_msgs) == 1
     assert 'redis version: 3.' in log_msgs[0]
+    assert log_msgs[0].endswith(', clients connected: 1, keys in db: 2')
 
 
 @pytest.mark.parametrize('previous,expected,kwargs', [