self/ruz-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix
All checks were successful
Deploy Dev / Build (pull_request) Successful in 5s
Details
Deploy Dev / Push (pull_request) Successful in 7s
Details
Deploy Dev / Deploy dev (pull_request) Successful in 16s
Details
Deploy Prod / Build (pull_request) Successful in 4s
Details
Deploy Prod / Push (pull_request) Successful in 8s
Details
Deploy Prod / Deploy prod (pull_request) Successful in 17s
Details

Browse Source
This commit is contained in:
emmatveev 2024-11-27 02:44:10 +03:00
parent e76cfc7994
commit 292bacc963
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
utils/queues.py
View File

@ -17,22 +17,22 @@ class QueuesException(Exception):
class TasksHandlerMixin:
def poll(self):
while True:
response = requests.get(f'{QUEUES_URL}/api/v1/take', headers={'queue': self.queue_name})
if response.status_code == 404:
response = requests.get(f'{QUEUES_URL}/api/v1/take', headers={'queue': self.queue_name}).json()
task = response.get('task')
if not task:
time.sleep(0.2)
continue
data = response.json()
try:
self.process(data['payload'])
self.process(task['payload'])
except Exception as exc:
print(f'Error processing message id={data["id"]}, payload={data["payload"]}, exc={exc}')
print(f'Error processing message id={task["id"]}, payload={task["payload"]}, exc={exc}')
continue
try:
resp = requests.post(f'{QUEUES_URL}/api/v1/finish', json={'id': data['id']})
resp = requests.post(f'{QUEUES_URL}/api/v1/finish', json={'id': task['id']})
if resp.status_code != 202:
raise QueuesException
except:
print(f'Failed to finish task id={data["id"]}')
print(f'Failed to finish task id={task["id"]}')
@property
def queue_name(self):