mirror of
https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git
synced 2026-11-01 01:24:45 +03:00
grillazz
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
"""Background database health-check scheduling."""
|
|
|
|
from collections.abc import AsyncIterator
|
|
from contextlib import asynccontextmanager
|
|
from datetime import datetime
|
|
|
|
from apscheduler import AsyncScheduler, ConflictPolicy
|
|
from apscheduler.datastores.sqlalchemy import SQLAlchemyDataStore
|
|
from apscheduler.eventbrokers.redis import RedisEventBroker
|
|
from apscheduler.triggers.interval import IntervalTrigger
|
|
from rotoger import get_logger
|
|
from sqlalchemy import text
|
|
|
|
from app.config import settings as global_settings
|
|
from app.database import AsyncSessionFactory, engine
|
|
|
|
logger = get_logger()
|
|
|
|
|
|
async def tick() -> bool:
|
|
"""Run the scheduler's lightweight database heartbeat."""
|
|
async with AsyncSessionFactory() as session:
|
|
stmt = text("select 1;")
|
|
await logger.ainfo(f">>>> Be or not to be...{datetime.now()}")
|
|
result = await session.execute(stmt)
|
|
await logger.ainfo(f">>>> Result: {result.scalar()}")
|
|
return True
|
|
|
|
|
|
@asynccontextmanager
|
|
async def scheduler_lifespan() -> AsyncIterator[None]:
|
|
"""Start the scheduler and stop it when the app lifespan ends."""
|
|
scheduler = AsyncScheduler(
|
|
SQLAlchemyDataStore(engine, schema="scheduler"),
|
|
RedisEventBroker(client_or_url=global_settings.redis_url.unicode_string()),
|
|
)
|
|
|
|
async with scheduler:
|
|
await scheduler.add_schedule(
|
|
tick,
|
|
IntervalTrigger(seconds=25),
|
|
id="tick-sql-25",
|
|
conflict_policy=ConflictPolicy.replace,
|
|
)
|
|
await scheduler.start_in_background()
|
|
yield
|