File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 99from taskiq import AsyncBroker
1010from taskiq .abc .result_backend import AsyncResultBackend
1111from taskiq .acks import AckableMessage
12+ from taskiq .exceptions import BrokerError
1213from taskiq .message import BrokerMessage
1314
1415logger = logging .getLogger (__name__ )
@@ -34,8 +35,11 @@ def __init__(
3435 self ._sqs = boto3 .resource ("sqs" )
3536 self ._sqs_queue : Optional [Queue ] = None
3637
37- self .wait_time_seconds = wait_time_seconds
38- self .max_number_of_messages = max_number_of_messages
38+ if max_number_of_messages > 10 :
39+ raise BrokerError ("MaxNumberOfMessages can be no greater than 10" )
40+
41+ self .wait_time_seconds = max (wait_time_seconds , 0 )
42+ self .max_number_of_messages = max (max_number_of_messages , 1 )
3943
4044 async def _get_queue (self ) -> Queue :
4145 queue_name = self .sqs_queue_url .split ("/" )[- 1 ]
You can’t perform that action at this time.
0 commit comments