taskiq-redis 1.0.6__tar.gz → 1.0.8__tar.gz

{taskiq_redis-1.0.6 → taskiq_redis-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: taskiq-redis
-Version: 1.0.6
+Version: 1.0.8
 Summary: Redis integration for taskiq
 Keywords: taskiq,tasks,distributed,async,redis,result_backend
 Author: taskiq-team

{taskiq_redis-1.0.6 → taskiq_redis-1.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "taskiq-redis"
-version = "1.0.6"
+version = "1.0.8"
 description = "Redis integration for taskiq"
 authors = ["taskiq-team <taskiq@norely.com>"]
 readme = "README.md"

{taskiq_redis-1.0.6 → taskiq_redis-1.0.8}/taskiq_redis/list_schedule_source.py

@@ -25,6 +25,17 @@ class ListRedisScheduleSource(ScheduleSource):
         skip_past_schedules: bool = False,
         **connection_kwargs: Any,
     ) -> None:
+        """
+        Create a new schedule source.
+
+        :param url: Redis URL
+        :param prefix: Prefix for all the keys
+        :param max_connection_pool_size: Maximum size of the connection pool
+        :param serializer: Serializer to use for the schedules
+        :param buffer_size: Buffer size for getting schedules
+        :param skip_past_schedules: Skip schedules that are in the past.
+        :param connection_kwargs: Additional connection kwargs
+        """
         super().__init__()
         self._prefix = prefix
         self._buffer_size = buffer_size
@@ -179,7 +190,7 @@ class ListRedisScheduleSource(ScheduleSource):
         current_time = datetime.datetime.now(datetime.timezone.utc)
         timed: list[bytes] = []
         # Only during first run, we need to get previous time schedules
-        if self._is_first_run and not self._skip_past_schedules:
+        if not self._skip_past_schedules:
             timed = await self._get_previous_time_schedules()
             self._is_first_run = False
         async with Redis(connection_pool=self._connection_pool) as redis:

{taskiq_redis-1.0.6 → taskiq_redis-1.0.8}/taskiq_redis/redis_broker.py

@@ -168,6 +168,7 @@ class RedisStreamBroker(BaseRedisBroker):
         maxlen: Optional[int] = None,
         idle_timeout: int = 600000,  # 10 minutes
         unacknowledged_batch_size: int = 100,
+        xread_count: Optional[int] = 100,
         additional_streams: Optional[Dict[str, str]] = None,
         **connection_kwargs: Any,
     ) -> None:
@@ -189,6 +190,7 @@ class RedisStreamBroker(BaseRedisBroker):
             Better to set it to a bigger value, to avoid unnecessary calls.
         :param maxlen: sets the maximum length of the stream
             trims (the old values of) the stream each time a new element is added
+        :param xread_count: number of messages to fetch from the stream at once.
         :param additional_streams: additional streams to read from.
             Each key is a stream name, value is a consumer id.
         :param redeliver_timeout: time in ms to wait before redelivering a message.
@@ -211,6 +213,7 @@ class RedisStreamBroker(BaseRedisBroker):
         self.additional_streams = additional_streams or {}
         self.idle_timeout = idle_timeout
         self.unacknowledged_batch_size = unacknowledged_batch_size
+        self.count = xread_count
 
     async def _declare_consumer_group(self) -> None:
         """
@@ -276,6 +279,7 @@ class RedisStreamBroker(BaseRedisBroker):
                     },
                     block=self.block,
                     noack=False,
+                    count=self.count,
                 )
                 for _, msg_list in fetched:
                     for msg_id, msg in msg_list:

{taskiq_redis-1.0.6 → taskiq_redis-1.0.8}/taskiq_redis/schedule_source.py

@@ -1,4 +1,5 @@
 import sys
+import warnings
 from contextlib import asynccontextmanager
 from typing import TYPE_CHECKING, Any, AsyncIterator, List, Optional, Tuple
 
@@ -53,6 +54,12 @@ class RedisScheduleSource(ScheduleSource):
         serializer: Optional[TaskiqSerializer] = None,
         **connection_kwargs: Any,
     ) -> None:
+        warnings.warn(
+            "RedisScheduleSource is deprecated. "
+            "Please switch to ListRedisScheduleSource",
+            DeprecationWarning,
+            stacklevel=2,
+        )
         self.prefix = prefix
         self.connection_pool: _BlockingConnectionPool = BlockingConnectionPool.from_url(
             url=url,