Add max_tries to transaction()

Author: haukex

valkey/asyncio/client.py

@@ -437,6 +437,7 @@ async def transaction(
         shard_hint: Optional[str] = None,
         value_from_callable: bool = False,
         watch_delay: Optional[float] = None,
+        max_tries: Optional[int] = None,
     ):
         """
         Convenience method for executing the callable `func` as a transaction
@@ -445,7 +446,11 @@ async def transaction(
         """
         pipe: Pipeline
         async with self.pipeline(True, shard_hint) as pipe:
+            tries = 0
             while True:
+                tries += 1
+                if max_tries and max_tries>0 and tries>max_tries:
+                    raise ValkeyError(f"Bailing out of transaction after {tries-1} tries")
                 try:
                     if watches:
                         await pipe.watch(*watches)

valkey/client.py

@@ -408,8 +408,13 @@ def transaction(
         shard_hint = kwargs.pop("shard_hint", None)
         value_from_callable = kwargs.pop("value_from_callable", False)
         watch_delay = kwargs.pop("watch_delay", None)
+        max_tries = kwargs.pop("max_tries", None)
         with self.pipeline(True, shard_hint) as pipe:
+            tries = 0
             while True:
+                tries += 1
+                if max_tries and max_tries>0 and tries>max_tries:
+                    raise ValkeyError(f"Bailing out of transaction after {tries-1} tries")
                 try:
                     if watches:
                         pipe.watch(*watches)