gahusb
2ff31b2e76
- services/_shared/heartbeat.py (A1) WorkerStats/utc_now_iso/heartbeat_loop 소비 - image-render / video-render / music-render / insta-render 각 worker.py: stats = WorkerStats() 모듈 레벨 추가, poll_once에서 dispatch 전 busy=True, ack 후 jobs_done+1 / fail 후 jobs_failed+1 + last_job_at + busy=False - 각 main.py: lifespan에 aioredis(decode_responses=False) + heartbeat_loop 태스크 spawn, 종료 시 cancel + aclose - 각 tests/test_worker.py: test_poll_once_increments_jobs_done 추가 (image:flux / video:sora / music:suno / insta:_process_one mock) Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_019LV86jBozkNhSFXJA412fq
96 lines
2.8 KiB
Python
96 lines
2.8 KiB
Python
"""music-render FastAPI entry — health + lifespan + sync forward endpoints.
|
|
|
|
NAS music-lab이 sync helpers(lyrics, credits, timestamped, style-boost)를
|
|
httpx로 forward해서 이 endpoint들을 호출.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
from contextlib import asynccontextmanager
|
|
|
|
import redis.asyncio as aioredis
|
|
from fastapi import FastAPI, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
import worker
|
|
from _shared.heartbeat import heartbeat_loop
|
|
from providers.sync_ops import (
|
|
generate_lyrics, get_credits,
|
|
get_timestamped_lyrics, generate_style_boost,
|
|
)
|
|
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(name)s %(levelname)s %(message)s")
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
worker_task = asyncio.create_task(worker.worker_loop())
|
|
hb_redis = aioredis.from_url(os.getenv("REDIS_URL", "redis://192.168.45.54:6379"), decode_responses=False)
|
|
hb_task = asyncio.create_task(heartbeat_loop(hb_redis, "music-render", "render", worker.stats))
|
|
logger.info("music-render lifespan 시작")
|
|
try:
|
|
yield
|
|
finally:
|
|
for t in (worker_task, hb_task):
|
|
t.cancel()
|
|
try:
|
|
await t
|
|
except asyncio.CancelledError:
|
|
pass
|
|
await hb_redis.aclose()
|
|
logger.info("music-render lifespan 종료")
|
|
|
|
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
return {"ok": True, "service": "music-render"}
|
|
|
|
|
|
# ── Sync forward endpoints ──────────────────────────────────────────────
|
|
# NAS music-lab의 /api/music/lyrics 등 sync helpers가 이 endpoint들로 forward.
|
|
|
|
class LyricsRequest(BaseModel):
|
|
prompt: str
|
|
|
|
|
|
@app.post("/api/music-render/sync/lyrics")
|
|
def sync_lyrics(req: LyricsRequest):
|
|
result = generate_lyrics(req.prompt)
|
|
if not result:
|
|
raise HTTPException(502, "가사 생성 실패")
|
|
return result
|
|
|
|
|
|
@app.get("/api/music-render/sync/credits")
|
|
def sync_credits():
|
|
result = get_credits()
|
|
if result is None:
|
|
raise HTTPException(502, "크레딧 조회 실패")
|
|
return result
|
|
|
|
|
|
@app.get("/api/music-render/sync/timestamped-lyrics")
|
|
def sync_timestamped_lyrics(task_id: str, suno_id: str):
|
|
result = get_timestamped_lyrics(task_id, suno_id)
|
|
if not result:
|
|
raise HTTPException(502, "타임스탬프 가사 조회 실패")
|
|
return result
|
|
|
|
|
|
class StyleBoostRequest(BaseModel):
|
|
content: str
|
|
|
|
|
|
@app.post("/api/music-render/sync/style-boost")
|
|
def sync_style_boost(req: StyleBoostRequest):
|
|
result = generate_style_boost(req.content)
|
|
if not result:
|
|
raise HTTPException(502, "스타일 부스트 생성 실패")
|
|
return result
|