Merge pull request 'master' (#60) from master into prod

Reviewed-on: https://gitea.chocomarsh.com/self/botalka/pulls/60
This commit is contained in:
emmatveev 2025-06-15 03:51:49 +03:00
commit 9c6b14e4cf
2 changed files with 8 additions and 29 deletions

View File

@ -23,6 +23,7 @@ class Daemon(base.Daemon, queues.TasksHandlerMixin):
return 'botalka_mailbox'
def process(self, payload: dict):
print(f"Get message {payload}")
message = Message.model_validate(payload)
bot = platform.platform_client.get_config('bots')[message.project][message.name]
if not bot['mailbox_enabled']:

View File

@ -1,6 +1,5 @@
import datetime
import os
from threading import Thread
import zoneinfo
import requests
import time
@ -18,27 +17,6 @@ class QueuesException(Exception):
class TasksHandlerMixin:
def _send_metric(self, start, success, end):
try:
metric = requests.post('http://monitoring:1237/api/v1/metrics/task', json={
'timestamp': start.strftime("%Y-%m-%dT%H:%M:%S") + "Z",
'service': 'botalka',
'environment': stage,
'queue': self.queue_name,
'success': success,
'execution_time_ms': (end - start).microseconds // 1000,
})
if metric.status_code == 202:
print('metric ok')
else:
print(f'metric failed: {metric.status_code}')
except Exception as e:
print(f'metric failed: {e}')
def send_metric(self, start, success, end):
# Thread(target=self._send_metric, args=(start, success, end)).start()
# self._send_metric(start, success, end)
...
def poll(self):
while True:
@ -60,13 +38,13 @@ class TasksHandlerMixin:
print(f'Error processing message id={task["id"]}, payload={task["payload"]}, exc={exc}')
success = False
end = datetime.datetime.now().astimezone(zoneinfo.ZoneInfo("Europe/Moscow"))
try:
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={task["id"]}')
self.send_metric(start, success, end)
if success:
try:
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={task["id"]}')
@property
def queue_name(self):