Skip to content

Commit 38026c5

Browse files
committed
feat: handle invalid parameters on init
1 parent 2f10d71 commit 38026c5

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

taskiq_sqs/broker.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from taskiq import AsyncBroker
1010
from taskiq.abc.result_backend import AsyncResultBackend
1111
from taskiq.acks import AckableMessage
12+
from taskiq.exceptions import BrokerError
1213
from taskiq.message import BrokerMessage
1314

1415
logger = 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]

0 commit comments

Comments
 (0)