Цвылев Александр Вадимович
313af3a070
POST /tracks/{id}/optimize enqueues a transcode; GET /stream/{id}?quality=
serves a cached Opus rendition (miss → master + background warm); GET
/stream/{id}/hls/{playlist.m3u8,segment} serves the worker-generated HLS
rendition (AAC-in-TS), with ?token= propagated onto segment URLs for players
that can't set headers. Hexagonal: Transcoder port, FfmpegTranscoder adapter,
TranscodeService + cache-path helpers, transcode_track worker (idempotent),
schemas + deps wiring. ffmpeg already in the image.
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
"""arq task: transcode a track to a cached Opus rendition and/or HLS (§6.6).
|
|
|
|
Triggered by ``POST /tracks/{id}/optimize`` and by a streaming cache miss. Reads
|
|
the master once (via ``storage.as_local_path``) and writes into
|
|
``transcode_cache_path``. Idempotent — existing outputs are skipped, so repeated
|
|
enqueues (e.g. several plays before the first finishes) are cheap. The DB session
|
|
is released before ffmpeg runs; the track entity is a detached value object.
|
|
"""
|
|
|
|
import uuid
|
|
from typing import Any
|
|
|
|
from app.application.transcode_service import (
|
|
HLS_BITRATE,
|
|
bitrate_for_quality,
|
|
hls_dir,
|
|
hls_playlist_path,
|
|
opus_path,
|
|
)
|
|
from app.core.config import get_settings
|
|
from app.core.logging import get_logger
|
|
from app.infrastructure.db import session_scope
|
|
from app.infrastructure.db.repositories import SqlAlchemyTrackRepository
|
|
from app.infrastructure.storage.provider import get_file_storage
|
|
from app.infrastructure.transcode.ffmpeg import FfmpegTranscoder
|
|
|
|
log = get_logger("worker.transcode")
|
|
|
|
|
|
async def transcode_track(
|
|
_ctx: dict[str, Any],
|
|
*,
|
|
track_id: str,
|
|
quality: str = "high",
|
|
hls: bool = True,
|
|
) -> dict[str, Any]:
|
|
settings = get_settings()
|
|
tid = uuid.UUID(track_id)
|
|
|
|
async with session_scope() as session:
|
|
track = await SqlAlchemyTrackRepository(session).get_by_id(tid)
|
|
if track is None or track.storage_uri is None:
|
|
log.warning("transcode_skip_no_file", track_id=track_id)
|
|
return {"track_id": track_id, "opus": False, "hls": False}
|
|
|
|
storage = get_file_storage()
|
|
transcoder = FfmpegTranscoder(settings.ffmpeg_path)
|
|
root = settings.transcode_cache_path
|
|
did_opus = False
|
|
did_hls = False
|
|
|
|
async with storage.as_local_path(track.storage_uri) as src:
|
|
bitrate = bitrate_for_quality(quality)
|
|
if bitrate is not None:
|
|
dest = opus_path(root, tid, bitrate)
|
|
if not dest.exists():
|
|
await transcoder.to_opus(src, dest, bitrate_kbps=bitrate)
|
|
did_opus = True
|
|
if hls and not hls_playlist_path(root, tid).exists():
|
|
await transcoder.to_hls(src, hls_dir(root, tid), bitrate_kbps=HLS_BITRATE)
|
|
did_hls = True
|
|
|
|
log.info("transcode_done", track_id=track_id, opus=did_opus, hls=did_hls)
|
|
return {"track_id": track_id, "opus": did_opus, "hls": did_hls}
|