Add redis/redis-py@51516cb

Support for CLIENT TRACKINFO (redis/redis-py#1560)

Signed-off-by: Andrew-Chen-Wang <[email protected]>

Author: Andrew-Chen-Wang

aioredis/client.py

@@ -718,6 +718,7 @@ class Redis(Commands):
         "CLIENT SETNAME": bool_ok,
         "CLIENT UNBLOCK": lambda r: r and int(r) == 1 or False,
         "CLIENT PAUSE": bool_ok,
+        "CLIENT TRACKINGINFO": lambda r: list(map(str_if_bytes, r)),
         "CLUSTER ADDSLOTS": bool_ok,
         "CLUSTER COUNT-FAILURE-REPORTS": lambda x: int(x),
         "CLUSTER COUNTKEYSINSLOT": lambda x: int(x),

aioredis/commands.py

@@ -436,6 +436,13 @@ def client_id(self: _SELF_ANNOTATION) -> Awaitable:
         """Returns the current connection id"""
         return self.execute_command("CLIENT ID")
 
+    def client_trackinginfo(self: _SELF_ANNOTATION) -> Awaitable:
+        """Returns the information about the current client connection's
+        use of the server assisted client side cache.
+        See https://redis.io/commands/client-trackinginfo
+        """
+        return self.execute_command("CLIENT TRACKINGINFO")
+
     def client_setname(self: _SELF_ANNOTATION, name: str) -> Awaitable:
         """Sets the current connection name"""
         return self.execute_command("CLIENT SETNAME", name)

tests/test_commands.py

@@ -380,6 +380,12 @@ async def test_client_list_client_id(self, r: aioredis.Redis):
     async def test_client_id(self, r: aioredis.Redis):
         assert await r.client_id() > 0
 
+    @skip_if_server_version_lt('6.2.0')
+    async def test_client_trackinginfo(self, r: aioredis.Redis):
+        res = await r.client_trackinginfo()
+        assert len(res) > 2
+        assert 'prefixes' in res
+
     @skip_if_server_version_lt("5.0.0")
     async def test_client_unblock(self, r: aioredis.Redis):
         myid = await r.client_id()