eric/gitlab-instance-0a899031_douyinyoutube
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

srs-mpv

Browse Source
This commit is contained in:
Your Name
2026-01-02 04:37:59 +00:00
parent 041a402a73
commit 2375331973
2 changed files with 100 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
config/URL_config.ini
View File

@@ -1 +1,30 @@
https://live.douyin.com/9638535297,主播: 异度世界
https://live.douyin.com/372551154040,主播: 向嘉_手打柠檬茶
https://live.douyin.com/203476627795,主播: 蜜桃姐姐(手打柠檬茶)
https://live.douyin.com/825627287092,主播: 杏绘茶档_(00后创业版_手打柠檬茶)
https://live.douyin.com/147650297461,主播: 娱乐摆摊记
https://live.douyin.com/806666144080,主播: 卢荟胶
https://live.douyin.com/589958309143,主播: 大萝卜
https://live.douyin.com/35932729226,主播: 内向小学生
https://live.douyin.com/456576000931,主播: 介休〖烩饕〗大块牛肉饭「摆摊」
https://live.douyin.com/782322393954,主播: 小葵(摆摊版)
https://live.douyin.com/208318182053,主播: 小碗(创业版)
https://live.douyin.com/646935372639,主播: 小雪摆摊记
https://live.douyin.com/956877163856,主播: 饭饱饱
https://live.douyin.com/881451071780,主播: YINING一柠手打柠檬茶
https://live.douyin.com/270513395244,主播: 小喻的摆摊日记
https://live.douyin.com/742474761291,主播: 阿诺椰子•
https://live.douyin.com/820845121901,主播: 地摊闺蜜_创意寿司_(马儿努力版)
https://live.douyin.com/980855021360,主播: 图爸摆摊卖枣糕(日照首家)教学员
https://live.douyin.com/46238285118,主播: 我叫小彤呀
https://live.douyin.com/497155982953,主播: 想吃糖葫芦
https://live.douyin.com/349973241990,主播: 元气满满饭团(摆摊不摆烂版)
https://live.douyin.com/814839174401,主播: 飞飞摆摊不摆烂
https://live.douyin.com/597618993066,主播: 婷婷的茶生活
https://live.douyin.com/16531926455,主播: 木黎黎
https://live.douyin.com/739670432305,主播: 干饭007
https://live.douyin.com/486789870173,主播: 凉皮西施—豆豆
https://live.douyin.com/15652099787,主播: 莹莹在摆摊
https://live.douyin.com/572442812068,主播: 小小
https://live.douyin.com/477961427964,主播: 小鱼
https://live.douyin.com/686507122231,主播: 上海少妇鲜果茶

177
main.py
View File

@@ -1164,11 +1164,9 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
push_at = datetime.datetime.today().strftime('%Y-%m-%d %H:%M:%S')
if port_info['is_live'] is False:
print(f"\r{record_name} 等待直播... ")
# 直接退出线程eric
# return
# 继续循环
continue
# continue # 继续循环
return # 线程彻底退出
if start_pushed:
if over_show_push:
@@ -1622,17 +1620,15 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
error_window.append(1)
else:
# ====================== 抖音 → HDMI 终极无人值守版（365×24h 永不卡死） ======================
# ====================== 抖音 → SRS → ffplay HDMI 终极低CPU清晰版 ======================
import queue
import unicodedata
import traceback
# import os
# import datetime
# import time
# import datetime
# import subprocess
# import threading
# import re
# import signal
class StreamEnded(Exception):
pass
@@ -1671,27 +1667,15 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
proc.kill()
print(f"[INFO] 已强制终止 {name}")
def clear_fifo(fifo_path):
"""最彻底的FIFO清理方式"""
if os.path.exists(fifo_path):
try:
os.remove(fifo_path)
except:
pass
try:
os.mkfifo(fifo_path)
print(f"[INFO] FIFO 已彻底重建: {fifo_path}")
except Exception as e:
print(f"[FATAL] 创建 FIFO 失败: {e}")
raise
# ====================== 配置参数 ======================
FIFO_PATH = "/tmp/douyin_hdmi.fifo"
SRS_RTMP_URL = "rtmp://127.0.0.1/live/douyin"
SRS_PLAY_URL = "http://127.0.0.1:8080/live/douyin.flv"
MAX_RETRY_DELAY = 120
INITIAL_RETRY_DELAY = 3
NO_FRAME_TIMEOUT = 180 # 长时间无帧才判定下播，防止误判
STALE_OUTPUT_SECONDS = 180 # FFmpeg僵死检测时间加长
MPV_RESTART_INTERVAL = 3600 * 6 # 每6小时强制重启一次mpv（防内存泄漏）
NO_FRAME_TIMEOUT = 180
STALE_OUTPUT_SECONDS = 180
FFPLAY_RESTART_INTERVAL = 3600 * 6
END_KEYWORDS = [
"connection reset by peer", "failed to write trailer", "invalid data found",
@@ -1703,90 +1687,83 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
timestamp_str = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
_clean_title = sanitize_title(title_in_name)
stream_title = sanitize_title(f"{anchor_name}{('_' + _clean_title) if _clean_title else ''}_{timestamp_str}")
print(f"[INFO] {rec_info} 启动终极无人值守HDMI转播: {stream_title}")
print(f"[INFO] {rec_info} 启动 SRS → ffplay HDMI 转播（低CPU原画版）: {stream_title}")
# ====================== 最稳 mpv 参数（全年实测无黑屏） ======================
mpv_cmd = [
"mpv", FIFO_PATH,
"--vo=drm",
"--drm-device=/dev/dri/card0",
"--drm-connector=0.HDMI-A-1", # 大多数板子有效；若黑屏改成 "HDMI-A-1"
"--hwdec=v4l2m2m-copy",
"--profile=low-latency",
"--cache=no", # 坚决关闭缓存
"--demuxer-lavf-format=flv",
"--demuxer-lavf-o=fflags=+nobuffer+genpts",
"--framedrop=no",
"--video-sync=display-resample",
"--video-rotate=270",
"--vd-lavc-fast",
"--video-unscaled=no",
"--fullscreen",
"--quiet",
"--loop-file=no",
"--no-input-default-bindings",
"--idle=no", # 防止mpv退出
]
douyin_headers = (
"User-Agent: Mozilla/5.0 (iPhone; CPU iPhone OS 18_1 like Mac OS X) AppleWebKit/605.1.15\r\n"
"Referer: https://live.douyin.com/\r\n"
"Origin: https://live.douyin.com\r\n"
)
# ====================== FFmpeg 推流到 SRS（直抄原画，低CPU） ======================
ffmpeg_command = [
"ffmpeg", "-y", "-loglevel", "info", "-nostdin",
"-headers", douyin_headers,
"-rw_timeout", "30000000",
"-reconnect", "1", "-reconnect_at_eof", "1", "-reconnect_streamed", "1",
"-reconnect_delay_max", "7",
"-fflags", "+genpts+discardcorrupt+igndts",
"-err_detect", "ignore_err",
"-i", real_url,
"-fflags", "+genpts+discardcorrupt",
"-flags", "low_delay",
"-c:v", "libx264", "-preset", "ultrafast",
"-profile:v", "high", "-b:v", "2600k", "-maxrate", "3000k",
"-bufsize", "6000k", "-g", "60", "-r", "30",
"-vsync", "cfr", "-pix_fmt", "yuv420p",
"-c:a", "aac", "-b:a", "128k", "-ar", "48000", "-ac", "2",
"-af", "aresample=async=1",
"-f", "flv", "-flvflags", "+no_duration_filesize",
FIFO_PATH
"-c:v", "copy", # 直抄视频，CPU极低，画质取决于源（抖音通常给原画）
"-c:a", "copy", # 直抄音频
"-f", "flv", SRS_RTMP_URL
]
# ====================== 主循环（终极自愈） ======================
proc = None
mpv_proc = None
# ====================== ffplay 最佳参数（低CPU + 零漂移 + 全屏） ======================
ffplay_cmd = [
"ffplay",
"-fflags", "nobuffer+genpts",
"-flags", "low_delay",
"-framedrop",
"-sync", "ext", # 零漂移
"-vf", "transpose=2", # 旋转
"-autoexit",
"-fs", # 全屏
"-loglevel", "quiet",
"-i", SRS_PLAY_URL
]
ffmpeg_proc = None
ffplay_proc = None
stderr_q = queue.Queue()
reader_t = None
retry_delay = INITIAL_RETRY_DELAY
last_mpv_restart = time.time() # 用于定时强制重启
last_ffplay_restart = time.time()
try:
while True:
current_time = time.time()
# ========== 定时强制重启 mpv（防内存泄漏/驱动僵死） ==========
if mpv_proc and (current_time - last_mpv_restart > MPV_RESTART_INTERVAL):
print("[SCHEDULE] 到达6小时，定时重启 mpv（预防性维护）")
cleanup_proc(mpv_proc, "mpv")
mpv_proc = None
if ffplay_proc and (current_time - last_ffplay_restart > FFPLAY_RESTART_INTERVAL):
print("[SCHEDULE] 6小时预防性重启 ffplay")
cleanup_proc(ffplay_proc, "ffplay")
ffplay_proc = None
# ========== 重启/启动 mpv ==========
if mpv_proc is None or mpv_proc.poll() is not None:
if mpv_proc is not None:
print("[WARN] mpv 意外退出，正在重启...")
if ffplay_proc is None or ffplay_proc.poll() is not None:
if ffplay_proc is not None:
print("[WARN] ffplay 退出，重启中...")
clear_fifo(FIFO_PATH) # 每次重启mpv必清FIFO
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 mpv")
mpv_proc = subprocess.Popen(
mpv_cmd,
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 ffplay 播放 SRS 原画流")
ffplay_proc = subprocess.Popen(
ffplay_cmd,
stdin=subprocess.DEVNULL,
stdout=subprocess.DEVNULL,
stderr=subprocess.PIPE,
text=True
)
time.sleep(2) # 给足启动时间
last_mpv_restart = time.time()
time.sleep(2)
last_ffplay_restart = time.time()
# ========== 重连等待 ==========
if retry_delay > INITIAL_RETRY_DELAY:
print(f"[WARN] 等待 {retry_delay}s 后重连...")
print(f"[WARN] 等待 {retry_delay}s 后重连抖音源...")
time.sleep(retry_delay)
retry_delay = min(retry_delay * 2, MAX_RETRY_DELAY)
cleanup_proc(proc, "FFmpeg")
cleanup_proc(ffmpeg_proc, "FFmpeg")
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 拉流")
proc = subprocess.Popen(
print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 直抄推流到 SRS")
ffmpeg_proc = subprocess.Popen(
ffmpeg_command,
stdin=subprocess.DEVNULL,
stdout=subprocess.DEVNULL,
@@ -1795,10 +1772,10 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
bufsize=1,
universal_newlines=True
)
proc.last_out_ts = time.time()
ffmpeg_proc.last_out_ts = time.time()
stderr_q = queue.Queue()
reader_t = threading.Thread(target=_stderr_reader_thread, args=(proc, stderr_q))
reader_t = threading.Thread(target=_stderr_reader_thread, args=(ffmpeg_proc, stderr_q))
reader_t.daemon = True
reader_t.start()
@@ -1808,8 +1785,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
try:
line = stderr_q.get(timeout=1.5)
except queue.Empty:
# FFmpeg僵死检测
if time.time() - proc.last_out_ts > STALE_OUTPUT_SECONDS:
if time.time() - ffmpeg_proc.last_out_ts > STALE_OUTPUT_SECONDS:
print("[WARN] FFmpeg 僵死，重启")
break
continue
@@ -1821,44 +1797,33 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
if not line:
continue
proc.last_out_ts = time.time()
ffmpeg_proc.last_out_ts = time.time()
print(f"[FFmpeg] {line}")
line_lower = line.lower()
# mpv缓冲堆积 → 立即重启mpv+清FIFO
if "1000 buffers queued" in line_lower:
print("[CRITICAL] mpv 缓冲堆积，强制重启 mpv + 清空 FIFO")
cleanup_proc(mpv_proc, "mpv")
mpv_proc = None
clear_fifo(FIFO_PATH)
break
# 下播关键字
if any(kw in line_lower for kw in END_KEYWORDS):
print("[INFO] 检测到下播关键字，停止转播")
raise StreamEnded()
# 新帧重置
if re.search(r'frame=\s*\d+', line_lower):
last_frame_time = time.time()
retry_delay = INITIAL_RETRY_DELAY
# 长时间无帧 → 下播
if time.time() - last_frame_time > NO_FRAME_TIMEOUT:
print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播")
raise StreamEnded()
except StreamEnded:
print(f"[INFO] 主播已下播，停止HDMI转播 → {record_name}")
print(f"[INFO] 主播已下播，停止 SRS → ffplay HDMI 转播 → {record_name}")
except KeyboardInterrupt:
print("[INFO] 用户手动中断")
except Exception as e:
print(f"[FATAL] 未知异常: {e}")
traceback.print_exc()
time.sleep(10) # 异常后稍等再重试
time.sleep(10)
finally:
cleanup_proc(proc, "FFmpeg")
cleanup_proc(mpv_proc, "mpv")
# 清理全局状态
cleanup_proc(ffmpeg_proc, "FFmpeg")
cleanup_proc(ffplay_proc, "ffplay")
try:
recording.discard(record_name)
recording_time_list.pop(record_name, None)
@@ -1869,7 +1834,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None:
clear_record_info(record_name, record_url)
except:
pass
color_obj.print_colored(f"[{record_name}] HDMI转播已安全停止并清理\n", color_obj.GREEN)
color_obj.print_colored(f"[{record_name}] SRS → ffplay HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN)
count_time = time.time()