Vladless
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
import asyncio
|
|
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
from apscheduler.triggers.interval import IntervalTrigger
|
|
|
|
from database import async_session_maker, cancel_expired_pending_payments
|
|
from logger import logger
|
|
|
|
|
|
async def scheduled_audit_drain() -> None:
|
|
from audit import drain_audit_redis_to_db
|
|
|
|
try:
|
|
drained = await drain_audit_redis_to_db(async_session_maker)
|
|
logger.info("[AuditDrain] Ночной drain завершён, записано событий: {}", drained)
|
|
except Exception as error:
|
|
logger.error("[AuditDrain] Ошибка ночного drain: {}", error)
|
|
|
|
|
|
async def scheduled_stats_report() -> None:
|
|
from handlers.admin.stats.stats_handler import send_daily_stats_report
|
|
|
|
async with async_session_maker() as session:
|
|
await send_daily_stats_report(session)
|
|
|
|
|
|
async def sweep_stale_payments_job() -> None:
|
|
async with async_session_maker() as session:
|
|
await cancel_expired_pending_payments(session)
|
|
await session.commit()
|
|
|
|
|
|
def scheduled_audit_drain_process_runner() -> None:
|
|
asyncio.run(scheduled_audit_drain())
|
|
|
|
|
|
def scheduled_stats_report_process_runner() -> None:
|
|
asyncio.run(scheduled_stats_report())
|
|
|
|
|
|
def sweep_stale_payments_process_runner() -> None:
|
|
asyncio.run(sweep_stale_payments_job())
|
|
|
|
|
|
async def cleanup_expired_gifts_job() -> None:
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import update as sa_update
|
|
|
|
from database.models import Gift
|
|
|
|
async with async_session_maker() as session:
|
|
try:
|
|
result = await session.execute(
|
|
sa_update(Gift).where(Gift.expiry_time < datetime.utcnow(), Gift.is_used is False).values(is_used=True)
|
|
)
|
|
count = result.rowcount
|
|
await session.commit()
|
|
if count:
|
|
logger.info("[GiftCleanup] Просроченных подарков помечено использованными: {}", count)
|
|
except Exception as error:
|
|
logger.error("[GiftCleanup] Ошибка очистки подарков: {}", error)
|
|
|
|
|
|
def cleanup_expired_gifts_process_runner() -> None:
|
|
asyncio.run(cleanup_expired_gifts_job())
|
|
|
|
|
|
async def log_db_pool_status() -> None:
|
|
"""Раз в минуту логирует состояние пула соединений: даёт видимость «упираемся ли в лимит»."""
|
|
try:
|
|
from database.db import engine
|
|
|
|
pool = engine.pool
|
|
size = pool.size()
|
|
checked_out = pool.checkedout()
|
|
checked_in = getattr(pool, "checkedin", lambda: size - checked_out)()
|
|
overflow = getattr(pool, "overflow", lambda: -1)()
|
|
logger.info(
|
|
"[DBPool] size={} in_use={} idle={} overflow={}",
|
|
size,
|
|
checked_out,
|
|
checked_in,
|
|
overflow,
|
|
)
|
|
except Exception as error:
|
|
logger.debug("[DBPool] не удалось получить статус пула: {}", error)
|
|
|
|
|
|
AUDIT_DRAIN_TRIGGER = CronTrigger(hour=0, minute=0, timezone="Europe/Moscow")
|
|
DAILY_STATS_REPORT_TRIGGER = CronTrigger(hour=0, minute=1, timezone="Europe/Moscow")
|
|
STALE_PAYMENTS_SWEEP_TRIGGER = CronTrigger(minute=0, timezone="Europe/Moscow")
|
|
EXPIRED_GIFTS_CLEANUP_TRIGGER = CronTrigger(hour=3, minute=0, timezone="Europe/Moscow")
|
|
DB_POOL_STATUS_TRIGGER = IntervalTrigger(minutes=1)
|