first commit

app/scheduler.py

@@ -0,0 +1,73 @@
+import asyncio
+import redis.asyncio as redis
+from taskiq import ScheduledTask
+
+from app.taskiq.broker import broker, scheduler, schedule_source, migrate_table_task
+from app.core.logging import migration_logger
+
+REDIS_URL = "redis://127.0.0.1:6379"
+PREFIX = "migration_schedule"
+
+async def sync_schedules_to_redis():
+    """Полная перезапись расписаний в Redis на основе данных из БД."""
+    migration_logger.info("Синхронизация расписаний с Redis...")
+
+    # 1. Получаем все активные расписания из БД через database.py
+    from app.core.database import db_connector
+    from app.models.replication import ReplicationSchedule
+    
+    # Инициализируем engine и session (обращение к свойству создает sessionmaker)
+    _ = db_connector.dst_engine
+    SessionLocal = db_connector.dst_session
+    session = SessionLocal()
+    
+    try:
+        schedules = session.query(ReplicationSchedule).filter(
+            ReplicationSchedule.enabled == True
+        ).all()
+        
+        # 2. Очищаем все старые ключи расписаний в Redis
+        r = await redis.from_url(REDIS_URL)
+        keys = await r.keys(f"{PREFIX}:*")
+        if keys:
+            await r.delete(*keys)
+        await r.close()
+
+        # 3. Добавляем каждое расписание через add_schedule
+        added = 0
+        for schedule in schedules:
+            hour = schedule.schedule_time.hour
+            minute = schedule.schedule_time.minute
+            days_list = schedule.days_list  # список чисел 0..6
+
+            # Формируем cron-выражение
+            if days_list:
+                cron_expr = f"{minute} {hour} * * {','.join(map(str, days_list))}"
+            else:
+                cron_expr = f"{minute} {hour} * * *"
+
+            # Используем timezone string для cron_offset (UTC+9 = Asia/Tokyo)
+            task = ScheduledTask(
+                task_name="migrate_table_task",
+                labels={},
+                cron=cron_expr,
+                cron_offset='Asia/Tokyo',  # Timezone name для UTC+9
+                args=[schedule.table_name],
+                kwargs={"full_reload": schedule.full_reload}
+            )
+
+            # Добавляем в Redis через источник
+            await schedule_source.add_schedule(task)
+            added += 1
+            migration_logger.info(f"Добавлено: {schedule.table_name} в {hour:02d}:{minute:02d} (cron: {cron_expr})")
+
+        migration_logger.info(f"Синхронизировано {added} активных расписаний")
+    finally:
+        session.close()
+
+
+async def on_scheduler_startup():
+    """Хук для синхронизации расписаний при запуске планировщика."""
+    migration_logger.info("Запуск планировщика taskiq...")
+    await sync_schedules_to_redis()
+    migration_logger.info("Планировщик готов к работе")