53d9e003af
- 回忆录：事实边界补充允许清单；传记文体示例与 JSON 叙事要求对齐 - default 职业提示 occupation_context；cadre/military 退休语境 - GET 章节读路径零写入，prepare_chapter_read_view + markdown_for_response - 文本归一抽到 core/text_normalize；移除弃用 reply 策略与 recompose_chapters_for_story - ConversationService：WS 连接/用户段落/结束对话；对外错误固定文案 - 测试：HTTP 脱敏契约、章节读视图、occupation 与 background_voice
84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
"""Celery：story 变更后重组关联章节的 canonical_markdown（物化视图）。"""
|
||
|
||
from datetime import datetime, timezone
|
||
|
||
from celery import shared_task
|
||
|
||
from app.core.chapter_pipeline_lock import (
|
||
acquire_chapter_pipeline_lock,
|
||
release_chapter_pipeline_lock,
|
||
)
|
||
from app.core.config import settings
|
||
from app.core.db import get_sync_db
|
||
from app.core.logging import get_logger
|
||
from app.core.memory_compaction_schedule import schedule_memory_compaction_run
|
||
from app.features.memoir import repo as memoir_repo
|
||
from app.features.memoir.models import Chapter
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
|
||
@shared_task(bind=True, max_retries=3, default_retry_delay=30)
|
||
def recompose_chapter(self, chapter_id: str) -> dict:
|
||
"""
|
||
按章节物化 canonical_markdown：仅当 markdown_compose_dirty 为 True 时执行；
|
||
与 pipeline 共用章节级 Redis 锁，拿不到锁则跳过（依赖后续触发重试）。
|
||
"""
|
||
lock_ttl = int(settings.chapter_pipeline_lock_ttl_seconds)
|
||
user_id: str | None = None
|
||
composed = False
|
||
with get_sync_db() as session:
|
||
chapter = session.get(Chapter, chapter_id)
|
||
if not chapter:
|
||
logger.info("recompose_chapter: chapter_id={} status=not_found", chapter_id)
|
||
return {"status": "not_found"}
|
||
if chapter.markdown_compose_dirty is not True:
|
||
logger.info(
|
||
"recompose_chapter: chapter_id={} status=skip_not_dirty",
|
||
chapter_id,
|
||
)
|
||
return {"status": "skip_not_dirty"}
|
||
uid = str(chapter.user_id)
|
||
stage = str(chapter.category)
|
||
lock_handle = acquire_chapter_pipeline_lock(uid, stage, ttl_seconds=lock_ttl)
|
||
if lock_handle is None:
|
||
logger.info(
|
||
"event=recompose_chapter status=skip_lock_contention "
|
||
"chapter_id={} user_id={} stage={}",
|
||
chapter_id,
|
||
uid,
|
||
stage,
|
||
)
|
||
return {"status": "skip_lock_contention"}
|
||
try:
|
||
composed = memoir_repo.compose_chapter_from_story_links_sync(
|
||
session, chapter_id
|
||
)
|
||
session.commit()
|
||
user_id = uid
|
||
except Exception as exc:
|
||
session.rollback()
|
||
logger.warning(
|
||
"recompose_chapter failed chapter_id={} err={}", chapter_id, exc
|
||
)
|
||
raise self.retry(exc=exc) from exc
|
||
finally:
|
||
release_chapter_pipeline_lock(lock_handle)
|
||
|
||
if user_id:
|
||
schedule_memory_compaction_run(
|
||
user_id,
|
||
{
|
||
"trigger_source": "chapter_recompose",
|
||
"trigger_time": datetime.now(timezone.utc).isoformat(),
|
||
"pipeline_run_id": str(self.request.id),
|
||
"recomposed_chapter_ids": [chapter_id],
|
||
},
|
||
)
|
||
logger.info(
|
||
"recompose_chapter: chapter_id={} status={}",
|
||
chapter_id,
|
||
"composed" if composed else "empty",
|
||
)
|
||
return {"status": "composed" if composed else "empty", "chapter_id": chapter_id}
|