You've already forked FrameTour-RenderWorker
feat(重构): 实现新的渲染服务架构
- 新增 RenderTask
This commit is contained in:
79
biz/task.py
79
biz/task.py
@@ -1,44 +1,55 @@
|
||||
import json
|
||||
import logging
|
||||
|
||||
from opentelemetry.trace import Status, StatusCode
|
||||
|
||||
from biz.ffmpeg import parse_ffmpeg_task, start_ffmpeg_task, clear_task_tmp_file, probe_video_info, fade_out_audio
|
||||
# 使用新的服务架构
|
||||
from services import DefaultTaskService, DefaultRenderService, DefaultTemplateService
|
||||
from telemetry import get_tracer
|
||||
from template import get_template_def
|
||||
from util import api
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 创建服务实例(单例模式)
|
||||
_render_service = None
|
||||
_template_service = None
|
||||
_task_service = None
|
||||
|
||||
def _get_services():
|
||||
"""获取服务实例(懒加载)"""
|
||||
global _render_service, _template_service, _task_service
|
||||
|
||||
if _render_service is None:
|
||||
_render_service = DefaultRenderService()
|
||||
|
||||
if _template_service is None:
|
||||
_template_service = DefaultTemplateService()
|
||||
_template_service.load_local_templates() # 加载本地模板
|
||||
|
||||
if _task_service is None:
|
||||
_task_service = DefaultTaskService(_render_service, _template_service)
|
||||
|
||||
return _task_service, _render_service, _template_service
|
||||
|
||||
def start_task(task_info):
|
||||
"""启动任务处理(保持向后兼容的接口)"""
|
||||
tracer = get_tracer(__name__)
|
||||
with tracer.start_as_current_span("start_task") as span:
|
||||
task_info = api.normalize_task(task_info)
|
||||
span.set_attribute("task", json.dumps(task_info))
|
||||
span.set_attribute("scenicId", task_info.get("scenicId", "?"))
|
||||
span.set_attribute("templateId", task_info.get("templateId"))
|
||||
template_info = get_template_def(task_info.get("templateId"))
|
||||
api.report_task_start(task_info)
|
||||
ffmpeg_task = parse_ffmpeg_task(task_info, template_info)
|
||||
result = start_ffmpeg_task(ffmpeg_task)
|
||||
if not result:
|
||||
with tracer.start_as_current_span("start_task_legacy") as span:
|
||||
try:
|
||||
task_service, _, _ = _get_services()
|
||||
|
||||
# 使用新的任务服务处理
|
||||
result = task_service.process_task(task_info)
|
||||
|
||||
if result:
|
||||
span.set_status(Status(StatusCode.OK))
|
||||
logger.info("Task completed successfully: %s", task_info.get("id"))
|
||||
else:
|
||||
span.set_status(Status(StatusCode.ERROR))
|
||||
logger.error("Task failed: %s", task_info.get("id"))
|
||||
|
||||
return None # 保持原有返回值格式
|
||||
|
||||
except Exception as e:
|
||||
span.set_status(Status(StatusCode.ERROR))
|
||||
return api.report_task_failed(task_info)
|
||||
width, height, duration = probe_video_info(ffmpeg_task)
|
||||
span.set_attribute("probe.width", width)
|
||||
span.set_attribute("probe.height", height)
|
||||
span.set_attribute("probe.duration", duration)
|
||||
# 音频淡出
|
||||
new_fn = fade_out_audio(ffmpeg_task.get_output_file(), duration)
|
||||
ffmpeg_task.set_output_file(new_fn)
|
||||
oss_result = api.upload_task_file(task_info, ffmpeg_task)
|
||||
if not oss_result:
|
||||
span.set_status(Status(StatusCode.ERROR))
|
||||
return api.report_task_failed(task_info)
|
||||
# 获取视频长度宽度和时长
|
||||
clear_task_tmp_file(ffmpeg_task)
|
||||
api.report_task_success(task_info, videoInfo={
|
||||
"width": width,
|
||||
"height": height,
|
||||
"duration": duration
|
||||
})
|
||||
span.set_status(Status(StatusCode.OK))
|
||||
return None
|
||||
logger.error("Task processing failed: %s", e, exc_info=True)
|
||||
return None
|
||||
|
Reference in New Issue
Block a user