Support for command count

Part of redis#1546
chayim · Sep 1, 2021 · 874b582 · 874b582
1 parent efdba1a
commit 874b582
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/redis/client.py b/redis/client.py
@@ -685,6 +685,8 @@ class Redis(Commands, object):
         'CLUSTER SET-CONFIG-EPOCH': bool_ok,
         'CLUSTER SETSLOT': bool_ok,
         'CLUSTER SLAVES': parse_cluster_nodes,
+        'COMMAND': int,
+        'COMMAND COUNT': int,
         'CONFIG GET': parse_config_get,
         'CONFIG RESETSTAT': bool_ok,
         'CONFIG SET': bool_ok,

diff --git a/redis/commands.py b/redis/commands.py
@@ -2906,6 +2906,9 @@ def module_list(self):
         """
         return self.execute_command('MODULE LIST')
 
+    def command_count(self):
+        return self.execute_command('COMMAND COUNT')
+
 
 class Script:
     "An executable Lua script object returned by ``register_script``"

diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -3228,6 +3228,12 @@ def test_module_list(self, r):
         assert isinstance(r.module_list(), list)
         assert not r.module_list()
 
+    @skip_if_server_version_lt('2.8.13')
+    def test_command_count(self, r):
+        res = r.command_count()
+        assert isinstance(res, int)
+        assert res >= 100
+
 
 class TestBinarySave: