# chongming-stream-manager

基于 ffmpeg + MediaMTX 的 RTSP 流代理管理器，用于将本地视频文件或 RTSP 网络流转发为 RTSP 流。

---

## 安装

```
uv pip install -e utils/stream_manager
```

前置依赖：系统需安装 ffmpeg。

---

## 使用场景

- 将本地视频文件循环推流为 RTSP 供外部消费
- 将 RTSP 网络流代理转发到 MediaMTX 统一出口
- AI Box 设备告警视频的实时流化

---

## API

### StreamManager 类

```
from chongming_stream_manager import StreamManager

manager = StreamManager(
    mediamtx_addr="rtsp://localhost:8554/",
    max_streams=16,
    video_path=""
)
```

| 参数 | 类型 | 默认值 | 说明 |
| --- | --- | --- | --- |
| mediamtx_addr | str | — | MediaMTX 服务器地址 |
| max_streams | int | 16 | 最大并发流数 |
| video_path | str | "" | 视频文件路径前缀 |

### start_stream(stream_id, source_url, duration=0, loop=-1) -> bool

启动一个转码推流任务。

```
success = await manager.start_stream(
    stream_id="camera_01",
    source_url="/path/to/video.mp4",
    duration=0,
    loop=-1
)
```

| 参数 | 类型 | 说明 |
| --- | --- | --- |
| stream_id | str | 流标识，输出路径 {mediamtx_addr}{stream_id} |
| source_url | str | 源：本地文件路径或 RTSP URL |
| duration | int | 推流时长（秒），0 = 不限 |
| loop | int | 循环次数，-1 = 无限（仅本地文件模式） |

### 内部机制

- 使用 asyncio.Semaphore 限制并发流数
- 每个流对应一个 ffmpeg 子进程，通过 asyncio.create_subprocess_exec 管理
- 本地文件模式：-re -stream_loop N 按原始帧率循环读取
- RTSP 模式：-rtsp_transport tcp -reconnect 1 自动重连
- 输出统一为 -c copy -f rtsp 流复制到 MediaMTX
- 后台 asyncio.Task 监控进程生命周期，超时自动终止

---

## 依赖

- ffmpeg — 系统级依赖，需单独安装
- Python 标准库（无额外 PyPI 依赖）