JSON.DEBUG

Author: AvitalFineRedis

rejson/client.py

@@ -67,6 +67,7 @@ def __init__(self, encoder=None, decoder=None, *args, **kwargs):
                 'JSON.ARRPOP': self._decode,
                 'JSON.ARRTRIM': long,
                 'JSON.OBJLEN': long,
+                'JSON.DEBUG': long,
         }
         for k, v in six.iteritems(MODULE_CALLBACKS):
             self.set_response_callback(k, v)
@@ -260,6 +261,12 @@ def jsonobjlen(self, name, path=Path.rootPath()):
         """
         return self.execute_command('JSON.OBJLEN', name, str_path(path))
 
+    def jsondebugmemory(self, name, path=Path.rootPath()):
+        """
+        Returns the memory usage in bytes of a value under ``path`` from key ``name``.
+        """
+        return self.execute_command("JSON.DEBUG", "MEMORY", name, str_path(path))
+
     def pipeline(self, transaction=True, shard_hint=None):
         """
         Return a new pipeline object that can queue multiple commands for

tests/test_rejson.py

@@ -174,6 +174,12 @@ def testObjLenShouldSucceed(self):
         rj.jsonset('obj', Path.rootPath(), obj)
         self.assertEqual(len(obj), rj.jsonobjlen('obj', Path.rootPath()))
 
+    def testDebugMemoryShouldSucceed(self):
+        "Test JSONDebug"
+
+        rj.jsonset('str', Path.rootPath(), 'foo')
+        self.assertEqual(24, rj.jsondebugmemory('str', Path.rootPath()))
+
     def testPipelineShouldSucceed(self):
         "Test pipeline"