11

core/services/cache.py

@@ -0,0 +1,70 @@
+import json
+from datetime import datetime, timedelta
+from typing import Any
+
+from astrbot.api import logger
+
+from ..sqlite import AsyncSQLiteDB
+
+
+class JSONCacheRepository:
+    """基于 achievement_cache 表的通用 JSON 缓存。"""
+
+    def __init__(self, database: AsyncSQLiteDB):
+        self._database = database
+
+    async def get(
+        self,
+        key: str,
+        *,
+        max_age: timedelta = timedelta(days=30),
+    ) -> tuple[Any | None, bool]:
+        try:
+            row = await self._database.select_one(
+                "achievement_cache",
+                "key=?",
+                (key,),
+            )
+        except Exception as exc:
+            logger.error(f"读取 JSON 缓存失败: key={key}, error={exc}")
+            return None, True
+
+        if not row:
+            return None, True
+
+        try:
+            payload = json.loads(row.get("content", "{}"))
+            updated_at = datetime.strptime(
+                row.get("updated_at", ""),
+                "%Y-%m-%d %H:%M:%S",
+            )
+        except (json.JSONDecodeError, TypeError, ValueError) as exc:
+            logger.error(f"解析 JSON 缓存失败: key={key}, error={exc}")
+            return None, True
+
+        return payload, datetime.now() - updated_at > max_age
+
+    async def set(self, key: str, payload: Any) -> None:
+        try:
+            content = json.dumps(payload, ensure_ascii=False)
+        except (TypeError, ValueError) as exc:
+            logger.error(f"序列化 JSON 缓存失败: key={key}, error={exc}")
+            return
+
+        try:
+            await self._database.execute(
+                """
+                INSERT INTO achievement_cache (key, content, updated_at)
+                VALUES (?, ?, ?)
+                ON CONFLICT(key) DO UPDATE SET
+                    content=excluded.content,
+                    updated_at=excluded.updated_at
+                """,
+                (
+                    key,
+                    content,
+                    datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                ),
+            )
+        except Exception as exc:
+            logger.error(f"写入 JSON 缓存失败: key={key}, error={exc}")