NODEJS基于FFMPEG视频推流测试

以ffmpeg为核心,包装一款局域网内接收转码并推送互联网的客户端软件。本文仅使用ffmpeg基础功能,拉流、转码、推流及简单播放设置。

工作流程

1. 拉取远端视频流,视频流格式为 rtsp

2. 转换为常用播放格式 rtmp

3. 推送至播放端口 rtmp://您的推送端地址,用户使用播放软件连接该地址后可以直接进行内容播放

所需工具及软件

1. ffmpeg 命令行工具官网链接,选择它的优势在于：

   • 免费

   • 无需安装,很大的减少用户操作复杂度

   • 命令行启动调用

2. nodejs 版本号为 v6.11.3。(实际项目中使用electron,但若没有打包成客户端的需求,nodejs即可正常运行)

3. tsc 版本号为 v2.6.1。项目使用TypeScript为主要编写语言,您使用JavaScript也没有问题。

   • 若使用tsc,请使用2.0以上版本,自带的@type工具会极大提升编码效率

4. fluent-ffmpeg 版本号为 v2.1.2。该nodejs包封装了ffmpeg的命令行调用部分,加强了代码的可读性,若熟悉ffmpeg 命令行使用手册,亦可不使用该包。

   npm install --save fluent-ffmpeg
       //使用js编码的用户,可以忽略下条命令
       npm install --save @types/fluent-ffmpeg

5. VLC播放软件。用于监测推流、转码、播放是否正常。官网链接

实现代码

const ffmpegPath = "./dist/ffmpegProgram/bin/ffmpeg.exe";
    const ffprobePath = "./dist/ffmpegProgram/bin/ffprobe.exe";
    const flvtoolPath = "./dist/ffmpegProgram/bin/ffplay.exe";

    export function startPushVideo():void{
        getCommands().then((commands:ffmpegPaths[])=>{
            for(let key in commands){
                let command = commands[key];
                //设置输入流地址
                let ffCommand = ffmpeg(command.inputPath)
                //设置输出流地址
                .output(command.outputPath)
                //因需要打包客户端软件,故而将ffmpeg打包进软件中
                //需设置各应用程序的对应路径
                //若仅在本机使用,可以跳过该步骤
                //设置环境变量,添加 PATH 即可
                .setFfmpegPath(ffmpegPath)
                .setFfprobePath(ffprobePath)
                .setFlvtoolPath(flvtoolPath)
                //为保证灵活性,非必须参数采用配置文件读取模式
                .size(command.size);
                for(let key in command.args){
                    ffCommand.outputOption(command.args[key]);
                }
                ffCommand.on("start",(commandLine)=>{
                    //commandLine 为实际上调用的命令行命令,拼接逻辑为
                    //您的ffmpeg所在路径 -i inputOptions 您的拉流协议和路径 outputOptions 推送流协议和地址
                    //ffmpeg -i "rtsp://yourPullUrl" -f flv -r 25 -s 640x480 -an "rtmp://yourPushUrl"
                    console.log('[' + showTime() + '] Vedio is Pushing !');
                    console.log('[' + showTime() + '] Spawned Ffmpeg with command !');
                    console.log('[' + showTime() + '] Command: ' + commandLine);
                })
                .on('error', function(err, stdout, stderr) {
                    console.log('error: ' + err.message);
                    console.log('stdout: ' + stdout);
                    console.log('stderr: ' + stderr);
                })
                .on('end', function() {
                    console.log('[' + showTime() + '] Vedio Pushing is Finished !');
                })
                .run();
            }
        },(error)=>{
            console.log('error: ' + error);
        })
    }

小结

1. 通过监听"start"获取的命令,亦可以通过 exec(yourCommandLine) 进行调用操作,但此时无法控制 ffmpeg 的运行结果。该程序结束运行之后,ffmpeg进程依然在运行,直至流报错或手动停止进程。暂时不清楚为何 fluent-ffmpeg 可以做到在本体进程结束后,通知关闭第三方进程。猜测是通过命令行输入切断进程,若仅通过 ChildProcess.kill() 是无法关闭第三方进程。

2. 在I5 8G 机器上运行,单流推送已占用35%左右cpu,多流推送需使用其他方案解决。