Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix SQS message polling to abort after wait_seconds_timeout

Browse Source
This commit is contained in:
Waldemar Hummer 2017-04-06 21:09:58 +10:00
parent bba197e29f
commit a5727bf64a
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
moto/sqs/models.py
View File

@ -289,6 +289,10 @@ class SQSBackend(BaseBackend):
# queue.messages only contains visible messages # queue.messages only contains visible messages
while True: while True:
if result or (wait_seconds_timeout and unix_time() > polling_end):
break
if len(queue.messages) == 0: if len(queue.messages) == 0:
import time import time
time.sleep(0.001) time.sleep(0.001)
@ -304,9 +308,6 @@ class SQSBackend(BaseBackend):
if len(result) >= count: if len(result) >= count:
break break
if result or unix_time() > polling_end:
break
return result return result
def delete_message(self, queue_name, receipt_handle): def delete_message(self, queue_name, receipt_handle):