33 lines
915 B
Python
33 lines
915 B
Python
|
import asyncpg
|
||
|
|
|
||
|
|
from onguard24.config import Settings
|
||
|
|
|
||
|
|
|
||
|
|
def normalize_dsn(url: str) -> str:
|
||
|
|
if url.startswith("postgres://"):
|
||
|
|
return url.replace("postgres://", "postgresql://", 1)
|
||
|
|
return url
|
||
|
|
|
||
|
|
|
||
|
|
async def create_pool(settings: Settings) -> asyncpg.Pool | None:
|
||
|
|
if not settings.database_url.strip():
|
||
|
|
return None
|
||
|
|
dsn = normalize_dsn(settings.database_url.strip())
|
||
|
|
return await asyncpg.create_pool(dsn=dsn, min_size=1, max_size=10)
|
||
|
|
|
||
|
|
|
||
|
|
MIGRATION_001 = """
|
||
|
|
CREATE TABLE IF NOT EXISTS ingress_events (
|
||
|
|
id uuid PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
|
|
source text NOT NULL,
|
||
|
|
received_at timestamptz NOT NULL DEFAULT now(),
|
||
|
|
body jsonb NOT NULL
|
||
|
|
);
|
||
|
|
CREATE INDEX IF NOT EXISTS ingress_events_received_at_idx ON ingress_events (received_at DESC);
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
async def migrate(pool: asyncpg.Pool) -> None:
|
||
|
|
async with pool.acquire() as conn:
|
||
|
|
await conn.execute(MIGRATION_001)
|