（aiohttp-asyncio-FFmpeg-Docker-SRS）实现异步摄像头转码服务器

本文介绍: 定义启动和停止 RTSP 转码任务的方法# 开始转码方法# 检查相同RTSP是否已有子进程在处理# 使用Semaphore限制并发进程数量# 实际的转码操作，这里需要调用ffmpeg或其他工具’ffmpeg’,’-i’, ip,# 创建异步子进程# 将任务添加到字典中# 等待子进程完成# 从字典中移除已完成的任务# 停止转码方法# 停止转码任务process.terminate() # 发送终止信号await process.wait() # 等待进程结束。

1. 背景介绍

在先前的博客文章中，我们已经搭建了一个基于SRS的流媒体服务器。现在，我们希望通过Web接口来控制这个服务器的行为，特别是对于正在进行的 RTSP 转码任务的管理。这将使我们能够在不停止整个服务器的情况下，动态地启动或停止摄像头的转码过程。

Docker部署 SRS rtmp/flv流媒体服务器-CSDN博客文章浏览阅读360次，点赞7次，收藏5次。SRS（Simple Realtime Server）是一款开源的流媒体服务器，具有高性能、高可靠性、高灵活性的特点，能够支持直播、点播、转码等多种流媒体应用场景。SRS 不仅提供了流媒体服务器，还提供了适用于多种平台的客户端 SDK 和在线转码等辅助服务，是一款十分强大的流媒体解决方案。https://blog.csdn.net/m0_56659620/article/details/135400510?spm=1001.2014.3001.5501

2. 技术选择

在选择技术方案时，考虑到构建视频流转码服务的需求，我们将采用Python编程语言，并结合asyncio和aiohttp库。这一选择基于异步框架的优势，以下是对异步框架和同步框架在视频流转码场景中的优缺点的明确总结：

异步框架的优势：

高并发处理： 异步框架通过非阻塞方式处理请求，能够高效处理大量并发请求，确保系统在高负载下保持稳定性。
异步I/O： 支持异步I/O操作，允许在等待I/O操作完成的同时继续处理其他请求，提高整体效率。
资源利用率高： 能够更有效地利用系统资源，同时处理多个请求，提高视频转码效率。
事件驱动： 采用事件驱动模型，适应实时性要求高的视频流处理，能够立即响应新的转码请求。

同步框架的缺点：

阻塞： 阻塞调用可能导致整个程序停滞，尤其在处理大文件或网络请求时可能引发性能问题，特别是在高并发场景下。
低并发： 每个请求需要独立的线程或进程，可能导致系统资源耗尽，降低并发处理能力，对于需要同时处理多个视频流的情况可能不够高效。

考虑到处理大量并发请求、提高系统性能和响应性的需求，采用异步框架是更为合适的选择。异步框架的高并发处理能力、异步I/O支持、高资源利用率以及事件驱动的特性使其更适用于实时性要求较高的视频流转码服务。

3. 代码实现（必须在linux系统运行，4步骤为部署攻略）

3.1 导入必要的库

首先，我们导入所需的库，包括asyncio、aiohttp、aiohttp_cors和logging。

import asyncio
from aiohttp import web
import aiohttp_cors
import logging

3.2 设置日志

logging.basicConfig(level=logging.INFO)

3.3 配置并发控制和任务跟踪

设置最大同时运行的ffmpeg子进程数量，并使用Semaphore限制并发进程数量。同时，使用字典跟踪正在进行的转码任务。

MAX_CONCURRENT_PROCESSES = 5
semaphore = asyncio.Semaphore(MAX_CONCURRENT_PROCESSES)
transcoding_tasks = {}

3.4 定义启动和停止转码任务的方法

定义启动和停止 RTSP 转码任务的方法

# 开始转码方法
async def perform_transcoding(ip, camera_id, rtmp_server):
    # 检查相同RTSP是否已有子进程在处理
    if camera_id in transcoding_tasks:
        return transcoding_tasks[camera_id]

    # 使用Semaphore限制并发进程数量
    async with semaphore:
        # 实际的转码操作，这里需要调用ffmpeg或其他工具
        ffmpeg_command = [
            'ffmpeg',
            '-rtsp_transport', 'tcp',
            '-i', ip,
            '-c:v', 'libx264',
            '-c:a', 'aac',
            '-f', 'flv',
            f'{rtmp_server}/live/livestream{camera_id}'
        ]

        # 创建异步子进程
        process = await asyncio.create_subprocess_exec(*ffmpeg_command)

        # 将任务添加到字典中
        transcoding_tasks[camera_id] = process

        # 等待子进程完成
        await process.communicate()

        # 从字典中移除已完成的任务
        transcoding_tasks.pop(camera_id, None)

# 停止转码方法
async def stop_transcoding(camera_id):
    # 停止转码任务
    if camera_id in transcoding_tasks:
        process = transcoding_tasks[camera_id]
        process.terminate()  # 发送终止信号
        await process.wait()  # 等待进程结束

        # 从字典中移除已停止的任务
        transcoding_tasks.pop(camera_id, None)

3.5 定义Web接口路由

定义Web接口路由，包括启动摄像头转码、停止摄像头转码和停止所有摄像头转码的路由。

# 开始转码任务
async def play_camera(request):
    data = await request.post()

    # 从表单数据中获取摄像头的ID和rtsp流
    camera_id = data.get('id')
    rtsp = data.get('ip')

    # 这里设置你的 RTMP 服务器地址
    rtmp_server = 'rtmp://192.168.14.93:1935'

    # 执行实际的转码操作
    task = await perform_transcoding(rtsp, camera_id, rtmp_server)

    # 返回包含转码后的RTMP URL的JSON响应
    rtmp_url = f'http://192.168.14.93:8080/live/livestream{camera_id}.flv'
    return web.json_response({'message': '转码启动成功', 'flv_data': rtmp_url})


# 停止转码任务
async def stop_camera(request):
    data = await request.post()
    camera_id = data.get('id')

    # 停止指定摄像头的转码任务
    await stop_transcoding(camera_id)

    return web.json_response({'code':200,'message': '转码停止成功'})


# 如果页面进行刷新或者关闭停止全部转码任务
async def stop_all_camera(request):

    # 获取所有正在运行的任务的列表
    tasks = [stop_transcoding(camera_id) for camera_id in transcoding_tasks.keys()]

    # 并发停止所有任务
    await asyncio.gather(*tasks)

    # 清空字典，表示所有任务都已停止
    transcoding_tasks.clear()

    return web.json_response({'code':200,'message': '转码停止成功'})

3.6 创建Web应用和配置CORS

创建Web应用，配置CORS（跨域资源共享）中间件，以确保接口可以被跨域访问。

app = web.Application()

# CORS配置
cors = aiohttp_cors.setup(app, defaults={
    "*": aiohttp_cors.ResourceOptions(
        allow_credentials=True,
        expose_headers="*",
        allow_headers="*",
    )
})

3.7 添加Web接口路由

添加Web接口路由，包括启动摄像头转码、停止摄像头转码和停止所有摄像头转码的路由。

app.router.add_route('POST', '/play_camera', play_camera)  # 开始转码任务路由
app.router.add_route('POST', '/stop_camera', stop_camera)  # 停止转码任务路由
app.router.add_route('POST', '/stop_all_camera', stop_all_camera)  # 停止全部转码任务路由

3.8 添加CORS中间件

添加CORS中间件，确保接口可以被跨域访问。

# 添加 CORS 中间件
for route in list(app.router.routes()):
    cors.add(route)

3.9 运行Web应用

运行Web应用，监听指定的主机和端口。

if __name__ == '__main__':
    web.run_app(app, host='0.0.0.0', port=7000,access_log=logging.getLogger())

3.10 完整代码

import asyncio
from aiohttp import web
import aiohttp_cors
import logging

# 设置日志级别
logging.basicConfig(level=logging.INFO)

# 最大同时运行的ffmpeg子进程数量
MAX_CONCURRENT_PROCESSES = 5

# 使用Semaphore限制并发进程数量
semaphore = asyncio.Semaphore(MAX_CONCURRENT_PROCESSES)

# 字典用于跟踪正在进行的转码任务
transcoding_tasks = {}

# 开始转码方法
async def perform_transcoding(ip, camera_id, rtmp_server):
    # 检查相同RTSP是否已有子进程在处理
    if camera_id in transcoding_tasks:
        return transcoding_tasks[camera_id]

    # 使用Semaphore限制并发进程数量
    async with semaphore:
        # 实际的转码操作，这里需要调用ffmpeg或其他工具
        ffmpeg_command = [
            'ffmpeg',
            '-rtsp_transport', 'tcp',
            '-i', ip,
            '-c:v', 'libx264',
            '-c:a', 'aac',
            '-f', 'flv',
            f'{rtmp_server}/live/livestream{camera_id}'
        ]

        # 创建异步子进程
        process = await asyncio.create_subprocess_exec(*ffmpeg_command)

        # 将任务添加到字典中
        transcoding_tasks[camera_id] = process

        # 等待子进程完成
        await process.communicate()

        # 从字典中移除已完成的任务
        transcoding_tasks.pop(camera_id, None)

# 停止转码方法
async def stop_transcoding(camera_id):
    # 停止转码任务
    if camera_id in transcoding_tasks:
        process = transcoding_tasks[camera_id]
        process.terminate()  # 发送终止信号
        await process.wait()  # 等待进程结束

        # 从字典中移除已停止的任务
        transcoding_tasks.pop(camera_id, None)


# 开始转码任务
async def play_camera(request):
    data = await request.post()

    # 从表单数据中获取摄像头的ID和rtsp流
    camera_id = data.get('id')
    rtsp = data.get('ip')

    # 这里设置你的 RTMP 服务器地址
    rtmp_server = 'rtmp://192.168.14.93:1935'

    # 执行实际的转码操作
    task = await perform_transcoding(rtsp, camera_id, rtmp_server)

    # 返回包含转码后的RTMP URL的JSON响应
    rtmp_url = f'http://192.168.14.93:8080/live/livestream{camera_id}.flv'
    return web.json_response({'message': '转码启动成功', 'flv_data': rtmp_url})


# 停止转码任务
async def stop_camera(request):
    data = await request.post()
    camera_id = data.get('id')

    # 停止指定摄像头的转码任务
    await stop_transcoding(camera_id)

    return web.json_response({'code':200,'message': '转码停止成功'})


# 如果页面进行刷新或者关闭停止全部转码任务
async def stop_all_camera(request):

    # 获取所有正在运行的任务的列表
    tasks = [stop_transcoding(camera_id) for camera_id in transcoding_tasks.keys()]

    # 并发停止所有任务
    await asyncio.gather(*tasks)

    # 清空字典，表示所有任务都已停止
    transcoding_tasks.clear()

    return web.json_response({'code':200,'message': '转码停止成功'})


app = web.Application()

# CORS配置
cors = aiohttp_cors.setup(app, defaults={
    "*": aiohttp_cors.ResourceOptions(
        allow_credentials=True,
        expose_headers="*",
        allow_headers="*",
    )
})

app.router.add_route('POST', '/play_camera', play_camera)  # 开始转码任务路由
app.router.add_route('POST', '/stop_camera', stop_camera)  # 停止转码任务路由
app.router.add_route('POST', '/stop_all_camera', stop_all_camera)  # 停止全部转码任务路由

# 添加 CORS 中间件
for route in list(app.router.routes()):
    cors.add(route)


if __name__ == '__main__':
    web.run_app(app, host='0.0.0.0', port=7000,access_log=logging.getLogger())

4. 部署（Docker环境）

部署所需Dockerfile文件代码如下

FROM python:3.7-slim

WORKDIR /app

COPY requirements.txt .

RUN apt-get update 
    && apt-get install -y ffmpeg 
    && rm -rf /var/lib/apt/lists/* 
    && pip install --no-cache-dir -r requirements.txt

COPY . .

CMD ["python", "async_io_io.py"]

部署所需requirements.txt如下