self/botalka
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

master #38

Merged
emmatveev merged 14 commits from master into prod 2024-12-08 19:57:42 +03:00
Conversation 0 Commits 14 Files Changed 7 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit ec2c39ed35 - Show all commits

4
utils/queues.py
View File

@ -10,7 +10,7 @@ class QueuesException(Exception):
class TasksHandlerMixin:
def poll(self):
while True:
response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(self.queue_name))
response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(queue=self.queue_name))
task: tasks_pb2.Task = response.task
if not task:
time.sleep(0.2)
@ -21,7 +21,7 @@ class TasksHandlerMixin:
print(f'Error processing message id={task["id"]}, payload={task["payload"]}, exc={exc}')
continue
try:
self.stub.Finish(tasks_pb2.FinishRequest(task.id))
self.stub.Finish(tasks_pb2.FinishRequest(id=task.id))
except:
print(f'Failed to finish task id={task.id}')