Add redis/redis-py@57cf7ff

NOMKSTREAM support for XADD (redis/redis-py#1507)

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

Author: Andrew-Chen-Wang

aioredis/client.py

@@ -2772,6 +2772,7 @@ def xadd(
         id: StreamIdT = "*",
         maxlen: Optional[int] = None,
         approximate: bool = True,
+        nomkstream: bool = False,
     ) -> Awaitable:
         """
         Add to a stream.
@@ -2780,7 +2781,7 @@ def xadd(
         id: Location to insert this record. By default it is appended.
         maxlen: truncate old stream members beyond this size
         approximate: actual stream length may be slightly more than maxlen
-
+        nomkstream: When set to true, do not make a stream
         """
         pieces: List[EncodableT] = []
         if maxlen is not None:
@@ -2790,6 +2791,8 @@ def xadd(
             if approximate:
                 pieces.append(b"~")
             pieces.append(str(maxlen))
+        if nomkstream:
+            pieces.append(b"NOMKSTREAM")
         pieces.append(id)
         if not isinstance(fields, dict) or len(fields) == 0:
             raise DataError("XADD fields must be a non-empty dict")

tests/test_commands.py

@@ -963,7 +963,7 @@ async def test_pttl_no_key(self, r: aioredis.Redis):
         assert await r.pttl("a") == -2
 
     @skip_if_server_version_lt("6.2.0")
-    def test_hrandfield(self, r):
+    async def test_hrandfield(self, r: aioredis.Redis):
         assert await r.hrandfield("key") is None
         await r.hset("key", mapping={"a": 1, "b": 2, "c": 3, "d": 4, "e": 5})
         assert await r.hrandfield("key") is not None
@@ -2457,6 +2457,16 @@ async def test_xadd(self, r: aioredis.Redis):
         await r.xadd(stream, {"foo": "bar"}, maxlen=2, approximate=False)
         assert await r.xlen(stream) == 2
 
+    @skip_if_server_version_lt("6.2.0")
+    async def test_xadd_nomkstream(self, r: aioredis.Redis):
+        # nomkstream option
+        stream = "stream"
+        await r.xadd(stream, {"foo": "bar"})
+        await r.xadd(stream, {"some": "other"}, nomkstream=False)
+        assert await r.xlen(stream) == 2
+        await r.xadd(stream, {"some": "other"}, nomkstream=True)
+        assert await r.xlen(stream) == 3
+
     @skip_if_server_version_lt("5.0.0")
     async def test_xclaim(self, r: aioredis.Redis):
         stream = "stream"