79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
|
|
import asyncio
|
|
import time
|
|
from app.config import ffmpeg_path, rtmp_url, flv_url, rtc_url
|
|
from app.services.stream_manager import stream_manager
|
|
from app.utils.logger import get_logger
|
|
from app.utils.md5 import generate_md5
|
|
import subprocess
|
|
|
|
logger = get_logger(__name__)
|
|
test_url = "rtsp://admin:jitu0818@192.168.4.102:554/Streaming/Channels/101"
|
|
|
|
async def log_output(process, output_url):
|
|
while True:
|
|
output = await asyncio.to_thread(process.stderr.readline)
|
|
if output == "" and process.poll() is not None:
|
|
break
|
|
if output:
|
|
logger.debug(f"ffmpeg output for {output_url}: {output.strip()}")
|
|
if process.returncode != 0:
|
|
logger.error(f"ffmpeg process for {output_url} failed with return code {process.returncode}")
|
|
|
|
|
|
async def start_stream(stream_id: str, output_url: str):
|
|
logger.info(f"Starting stream {stream_id}")
|
|
input_url = test_url # RTSP 流地址
|
|
# 构建 ffmpeg 命令
|
|
command = [
|
|
ffmpeg_path,
|
|
'-i', input_url, # 输入 RTSP 流
|
|
'-c:v', 'libx264', # 视频编码为 libx264
|
|
'-preset', 'ultrafast',
|
|
'-tune', 'zerolatency',
|
|
'-g', '5',
|
|
'-b:v', '6000k', # 码率 6Mbps
|
|
'-r', '30',
|
|
'-b:a', '128k',
|
|
'-c:a', 'aac', # 音频编码为 aac
|
|
'-f', 'flv',
|
|
output_url # 推送到 MediaMTX 流服务
|
|
]
|
|
|
|
# 使用 asyncio.to_thread 异步启动 ffmpeg 进程
|
|
process = await asyncio.to_thread(subprocess.Popen, command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
logger.info(f"ffmpeg command for {stream_id}: {' '.join(command)}")
|
|
|
|
# 异步记录输出
|
|
asyncio.create_task(log_output(process, stream_id))
|
|
|
|
# 保存进程信息
|
|
stream_manager[stream_id] = {
|
|
"process": process,
|
|
"create_time": int(time.time() * 1000)
|
|
}
|
|
return process
|
|
|
|
|
|
async def stream_rtsp_to_url(stream_id: str, type: str="flv"):
|
|
md5_route = generate_md5(stream_id)
|
|
output_url = f"{rtmp_url}/{md5_route}"
|
|
# 如果流已经在运行，检查进程是否仍在运行
|
|
if stream_id in stream_manager:
|
|
logger.info(f"Stream {stream_id} is already running.")
|
|
process = stream_manager[stream_id]["process"]
|
|
if process.poll() is None:
|
|
logger.info(f"Stream {stream_id} is still running.")
|
|
if type == "rtc":
|
|
return f"{rtc_url}{md5_route}.flv"
|
|
return f"{flv_url}/{md5_route}.flv"
|
|
# 如果流未在运行，启动新流
|
|
logger.info(f"Stream {stream_id} is not running. Starting new stream.")
|
|
process = await start_stream(stream_id, output_url)
|
|
if process.poll() is None:
|
|
logger.info(f"Stream {stream_id} started successfully.")
|
|
if type == "rtc":
|
|
return f"{rtc_url}{md5_route}.flv"
|
|
return f"{flv_url}/{md5_route}.flv"
|
|
return None
|