83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
# app/db/repositories/home_featured.py
|
||
from typing import List, Sequence
|
||
|
||
from asyncpg import Connection
|
||
|
||
from app.db.repositories.base import BaseRepository
|
||
|
||
|
||
class HomeFeaturedRepository(BaseRepository):
|
||
"""
|
||
维护首页推送文章的排序列表（最多 10 条）。
|
||
仅存 slug + sort_order，真正返回文章数据时再通过 ArticlesRepository 拉取。
|
||
"""
|
||
|
||
def __init__(self, conn: Connection) -> None:
|
||
super().__init__(conn)
|
||
|
||
async def _ensure_table(self) -> None:
|
||
await self.connection.execute(
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS home_featured_articles (
|
||
slug TEXT PRIMARY KEY,
|
||
sort_order INT NOT NULL DEFAULT 0,
|
||
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
||
);
|
||
""",
|
||
)
|
||
|
||
async def list_slugs(self, *, limit: int = 10) -> List[str]:
|
||
await self._ensure_table()
|
||
rows = await self.connection.fetch(
|
||
"""
|
||
SELECT slug
|
||
FROM home_featured_articles
|
||
ORDER BY sort_order ASC, updated_at DESC, created_at DESC
|
||
LIMIT $1;
|
||
""",
|
||
limit,
|
||
)
|
||
return [row["slug"] for row in rows]
|
||
|
||
async def save_slugs(self, *, slugs: Sequence[str], limit: int = 10) -> List[str]:
|
||
"""
|
||
保存首页推送顺序，自动去重并截断到 limit。
|
||
返回最终生效的 slug 顺序。
|
||
"""
|
||
await self._ensure_table()
|
||
clean_slugs: List[str] = []
|
||
for slug in slugs:
|
||
normalized = str(slug).strip()
|
||
if not normalized:
|
||
continue
|
||
if normalized not in clean_slugs:
|
||
clean_slugs.append(normalized)
|
||
clean_slugs = clean_slugs[:limit]
|
||
|
||
async with self.connection.transaction():
|
||
if clean_slugs:
|
||
await self.connection.execute(
|
||
"""
|
||
DELETE FROM home_featured_articles
|
||
WHERE slug <> ALL($1::text[]);
|
||
""",
|
||
clean_slugs,
|
||
)
|
||
for idx, slug in enumerate(clean_slugs):
|
||
await self.connection.execute(
|
||
"""
|
||
INSERT INTO home_featured_articles (slug, sort_order)
|
||
VALUES ($1, $2)
|
||
ON CONFLICT (slug) DO UPDATE
|
||
SET sort_order = EXCLUDED.sort_order,
|
||
updated_at = NOW();
|
||
""",
|
||
slug,
|
||
idx,
|
||
)
|
||
else:
|
||
await self.connection.execute("DELETE FROM home_featured_articles;")
|
||
|
||
return clean_slugs
|