Refactor metrics handling and remove scheduler

- Removed the metrics scheduler functionality from the bot, transitioning to real-time metrics updates via middleware. - Enhanced logging for metrics operations across various handlers to improve monitoring and debugging capabilities. - Integrated metrics tracking for user activities and database errors, providing better insights into bot performance. - Cleaned up code by removing obsolete comments and unused imports, improving overall readability and maintainability.
2025-09-03 19:18:04 +03:00
parent 650acd5bce
commit ae7bd476bb
14 changed files with 248 additions and 219 deletions
--- a/helper_bot/handlers/voice/services.py
+++ b/helper_bot/handlers/voice/services.py
@@ -14,6 +14,13 @@ from helper_bot.handlers.voice.constants import (
 )
 from logs.custom_logger import logger

+# Local imports - metrics
+from helper_bot.utils.metrics import (
+    metrics, 
+    track_time, 
+    track_errors,
+    db_query_time
+)

 class VoiceMessage:
    """Модель голосового сообщения"""
@@ -30,6 +37,8 @@ class VoiceBotService:
        self.bot_db = bot_db
        self.settings = settings
    
+    @track_time("get_welcome_sticker", "voice_bot_service")
+    @track_errors("voice_bot_service", "get_welcome_sticker")
    async def get_welcome_sticker(self) -> Optional[FSInputFile]:
        """Получить случайный приветственный стикер"""
        try:
@@ -47,6 +56,8 @@ class VoiceBotService:
                await self._send_error_to_logs(f'Отправка приветственных стикеров лажает. Ошибка: {e}')
            return None
    
+    @track_time("send_welcome_messages", "voice_bot_service")
+    @track_errors("voice_bot_service", "send_welcome_messages")
    async def send_welcome_messages(self, message, user_emoji: str):
        """Отправить приветственные сообщения"""
        try:
@@ -141,6 +152,8 @@ class VoiceBotService:
            logger.error(f"Ошибка при отправке приветственных сообщений: {e}")
            raise VoiceMessageError(f"Не удалось отправить приветственные сообщения: {e}")
    
+    @track_time("get_random_audio", "voice_bot_service")
+    @track_errors("voice_bot_service", "get_random_audio")
    async def get_random_audio(self, user_id: int) -> Optional[Tuple[str, str, str]]:
        """Получить случайное аудио для прослушивания"""
        try:
@@ -165,6 +178,8 @@ class VoiceBotService:
            logger.error(f"Ошибка при получении случайного аудио: {e}")
            raise AudioProcessingError(f"Не удалось получить случайное аудио: {e}")
    
+    @track_time("mark_audio_as_listened", "voice_bot_service")
+    @track_errors("voice_bot_service", "mark_audio_as_listened")
    async def mark_audio_as_listened(self, file_name: str, user_id: int) -> None:
        """Пометить аудио как прослушанное"""
        try:
@@ -173,6 +188,8 @@ class VoiceBotService:
            logger.error(f"Ошибка при пометке аудио как прослушанного: {e}")
            raise DatabaseError(f"Не удалось пометить аудио как прослушанное: {e}")
    
+    @track_time("clear_user_listenings", "voice_bot_service")
+    @track_errors("voice_bot_service", "clear_user_listenings")
    async def clear_user_listenings(self, user_id: int) -> None:
        """Очистить прослушивания пользователя"""
        try:
@@ -181,6 +198,8 @@ class VoiceBotService:
            logger.error(f"Ошибка при очистке прослушиваний: {e}")
            raise DatabaseError(f"Не удалось очистить прослушивания: {e}")
    
+    @track_time("get_remaining_audio_count", "voice_bot_service")
+    @track_errors("voice_bot_service", "get_remaining_audio_count")
    async def get_remaining_audio_count(self, user_id: int) -> int:
        """Получить количество оставшихся непрослушанных аудио"""
        try:
@@ -190,11 +209,15 @@ class VoiceBotService:
            logger.error(f"Ошибка при получении количества аудио: {e}")
            raise DatabaseError(f"Не удалось получить количество аудио: {e}")
    
+    @track_time("get_main_keyboard", "voice_bot_service")
+    @track_errors("voice_bot_service", "get_main_keyboard")
    def _get_main_keyboard(self):
        """Получить основную клавиатуру"""
        from helper_bot.keyboards.keyboards import get_main_keyboard
        return get_main_keyboard()
    
+    @track_time("send_error_to_logs", "voice_bot_service")
+    @track_errors("voice_bot_service", "send_error_to_logs")
    async def _send_error_to_logs(self, message: str) -> None:
        """Отправить ошибку в логи"""
        try:
@@ -215,6 +238,8 @@ class AudioFileService:
    def __init__(self, bot_db):
        self.bot_db = bot_db
    
+    @track_time("generate_file_name", "audio_file_service")
+    @track_errors("audio_file_service", "generate_file_name")
    async def generate_file_name(self, user_id: int) -> str:
        """Сгенерировать имя файла для аудио"""
        try:
@@ -245,6 +270,8 @@ class AudioFileService:
            logger.error(f"Ошибка при генерации имени файла: {e}")
            raise FileOperationError(f"Не удалось сгенерировать имя файла: {e}")
    
+    @track_time("save_audio_file", "audio_file_service")
+    @track_errors("audio_file_service", "save_audio_file")
    async def save_audio_file(self, file_name: str, user_id: int, date_added: datetime, file_id: str) -> None:
        """Сохранить информацию об аудио файле в базу данных"""
        try:
@@ -253,6 +280,8 @@ class AudioFileService:
            logger.error(f"Ошибка при сохранении аудио файла в БД: {e}")
            raise DatabaseError(f"Не удалось сохранить аудио файл в БД: {e}")
    
+    @track_time("download_and_save_audio", "audio_file_service")
+    @track_errors("audio_file_service", "download_and_save_audio")
    async def download_and_save_audio(self, bot, message, file_name: str) -> None:
        """Скачать и сохранить аудио файл"""
        try:
--- a/helper_bot/handlers/voice/voice_handler.py
+++ b/helper_bot/handlers/voice/voice_handler.py
@@ -22,6 +22,13 @@ from helper_bot.keyboards import get_reply_keyboard
 from helper_bot.handlers.private.constants import FSM_STATES
 from helper_bot.handlers.private.constants import BUTTON_TEXTS

+# Local imports - metrics
+from helper_bot.utils.metrics import (
+    metrics, 
+    track_time, 
+    track_errors,
+    db_query_time
+)

 class VoiceHandlers:
    def __init__(self, db, settings):
@@ -115,6 +122,8 @@ class VoiceHandlers:
            F.text == "😊Узнать эмодзи"
        )

+    @track_time("voice_bot_button_handler", "voice_handlers")
+    @track_errors("voice_handlers", "voice_bot_button_handler")
    async def voice_bot_button_handler(self, message: types.Message, state: FSMContext, bot_db: MagicData("bot_db"), settings: MagicData("settings")):
        """Обработчик кнопки 'Голосовой бот' из основной клавиатуры"""
        try:
@@ -135,6 +144,8 @@ class VoiceHandlers:
            # В случае ошибки вызываем start
            await self.start(message, state, bot_db, settings)

+    @track_time("restart_function", "voice_handlers")
+    @track_errors("voice_handlers", "restart_function")
    async def restart_function(
        self,
        message: types.Message, 
@@ -150,6 +161,8 @@ class VoiceHandlers:
        await message.answer(text='🎤 Записывайся или слушай!', reply_markup=markup)
        await state.set_state(STATE_START)

+    @track_time("handle_emoji_message", "voice_handlers")
+    @track_errors("voice_handlers", "handle_emoji_message")
    async def handle_emoji_message(
        self,
        message: types.Message,
@@ -162,6 +175,8 @@ class VoiceHandlers:
        if user_emoji is not None:
            await message.answer(f'Твоя эмодзя - {user_emoji}', parse_mode='HTML')

+    @track_time("help_function", "voice_handlers")
+    @track_errors("voice_handlers", "help_function")
    async def help_function(
        self,
        message: types.Message, 
@@ -177,6 +192,8 @@ class VoiceHandlers:
        )
        await state.set_state(STATE_START)

+    @track_time("start", "voice_handlers")
+    @track_errors("voice_handlers", "start")
    async def start(
        self,
        message: types.Message, 
@@ -201,6 +218,8 @@ class VoiceHandlers:
        except Exception as e:
            logger.error(f"Ошибка при отметке получения приветствия: {e}")

+    @track_time("cancel_handler", "voice_handlers")
+    @track_errors("voice_handlers", "cancel_handler")
    async def cancel_handler(
        self,
        message: types.Message, 
@@ -215,6 +234,8 @@ class VoiceHandlers:
        await message.answer(text='Добро пожаловать в меню!', reply_markup=markup, parse_mode='HTML')
        await state.set_state(FSM_STATES["START"])

+    @track_time("refresh_listen_function", "voice_handlers")
+    @track_errors("voice_handlers", "refresh_listen_function")
    async def refresh_listen_function(
        self,
        message: types.Message, 
@@ -239,6 +260,8 @@ class VoiceHandlers:
        await state.set_state(STATE_START)


+    @track_time("standup_write", "voice_handlers")
+    @track_errors("voice_handlers", "standup_write")
    async def standup_write(
        self,
        message: types.Message, 
@@ -261,6 +284,8 @@ class VoiceHandlers:
        await state.set_state(STATE_STANDUP_WRITE)


+    @track_time("suggest_voice", "voice_handlers")
+    @track_errors("voice_handlers", "suggest_voice")
    async def suggest_voice(
        self,
        message: types.Message, 
@@ -299,6 +324,8 @@ class VoiceHandlers:
            await state.set_state(STATE_STANDUP_WRITE)


+    @track_time("standup_listen_audio", "voice_handlers")
+    @track_errors("voice_handlers", "standup_listen_audio")
    async def standup_listen_audio(
        self,
        message: types.Message,