mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2026-06-13 03:31:45 +08:00
添加FFmpeg拉流功能
This commit is contained in:
xiongziliang
211
server/FFmpegSource.cpp
Normal file
211
server/FFmpegSource.cpp
Normal file
@@ -0,0 +1,211 @@
|
||||
//
|
||||
// Created by xzl on 2018/5/24.
|
||||
//
|
||||
|
||||
#include "FFmpegSource.h"
|
||||
#include "Common/config.h"
|
||||
#include "Common/MediaSource.h"
|
||||
#include "Util/File.h"
|
||||
#include "System.h"
|
||||
|
||||
namespace FFmpeg {
|
||||
#define FFmpeg_FIELD "ffmpeg."
|
||||
const char kBin[] = FFmpeg_FIELD"bin";
|
||||
const char kCmd[] = FFmpeg_FIELD"cmd";
|
||||
const char kLog[] = FFmpeg_FIELD"log";
|
||||
|
||||
onceToken token([]() {
|
||||
mINI::Instance()[kBin] = trim(System::execute("which ffmpeg"));
|
||||
mINI::Instance()[kCmd] = "%s -i %s -c:a aac -strict -2 -ar 44100 -ab 48k -c:v libx264 -f flv %s";
|
||||
mINI::Instance()[kLog] = exeDir() + "ffmpeg/ffmpeg.log";
|
||||
});
|
||||
}
|
||||
|
||||
FFmpegSource::FFmpegSource() {
|
||||
_poller = EventPollerPool::Instance().getPoller();
|
||||
}
|
||||
|
||||
FFmpegSource::~FFmpegSource() {
|
||||
NoticeCenter::Instance().delListener(this, Broadcast::kBroadcastStreamNoneReader);
|
||||
DebugL;
|
||||
}
|
||||
|
||||
|
||||
void FFmpegSource::play(const string &src_url,const string &dst_url,int timeout_ms,const onPlay &cb) {
|
||||
GET_CONFIG_AND_REGISTER(string,ffmpeg_bin,FFmpeg::kBin);
|
||||
GET_CONFIG_AND_REGISTER(string,ffmpeg_cmd,FFmpeg::kCmd);
|
||||
GET_CONFIG_AND_REGISTER(string,ffmpeg_log,FFmpeg::kLog);
|
||||
|
||||
_src_url = src_url;
|
||||
_dst_url = dst_url;
|
||||
_media_info.parse(dst_url);
|
||||
|
||||
char cmd[1024] = {0};
|
||||
snprintf(cmd, sizeof(cmd),ffmpeg_cmd.data(),ffmpeg_bin.data(),src_url.data(),dst_url.data());
|
||||
_process.run(cmd,ffmpeg_log);
|
||||
InfoL << cmd;
|
||||
|
||||
if(_media_info._host == "127.0.0.1"){
|
||||
//推流给自己的，通过判断流是否注册上来判断是否正常
|
||||
if(_media_info._schema != RTSP_SCHEMA && _media_info._schema != RTMP_SCHEMA){
|
||||
cb(SockException(Err_other,"本服务只支持rtmp/rtsp推流"));
|
||||
return;
|
||||
}
|
||||
weak_ptr<FFmpegSource> weakSelf = shared_from_this();
|
||||
findAsync(timeout_ms,[cb,weakSelf,timeout_ms](const MediaSource::Ptr &src){
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if(!strongSelf){
|
||||
//自己已经销毁
|
||||
return;
|
||||
}
|
||||
if(src){
|
||||
//推流给自己成功
|
||||
cb(SockException());
|
||||
strongSelf->startTimer(timeout_ms);
|
||||
return;
|
||||
}
|
||||
//推流失败
|
||||
if(!strongSelf->_process.wait(false)){
|
||||
//ffmpeg进程已经退出
|
||||
cb(SockException(Err_other,StrPrinter << "ffmpeg已经退出,exit code = " << strongSelf->_process.exit_code()));
|
||||
return;
|
||||
}
|
||||
//ffmpeg进程还在线，但是等待推流超时
|
||||
cb(SockException(Err_other,"等待超时"));
|
||||
});
|
||||
} else{
|
||||
//推流给其他服务器的，通过判断FFmpeg进程是否在线判断是否成功
|
||||
weak_ptr<FFmpegSource> weakSelf = shared_from_this();
|
||||
_timer = std::make_shared<Timer>(timeout_ms / 1000,[weakSelf,cb,timeout_ms](){
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if(!strongSelf){
|
||||
//自身已经销毁
|
||||
return false;
|
||||
}
|
||||
//FFmpeg还在线，那么我们认为推流成功
|
||||
if(strongSelf->_process.wait(false)){
|
||||
cb(SockException());
|
||||
strongSelf->startTimer(timeout_ms);
|
||||
return false;
|
||||
}
|
||||
//ffmpeg进程已经退出
|
||||
cb(SockException(Err_other,StrPrinter << "ffmpeg已经退出,exit code = " << strongSelf->_process.exit_code()));
|
||||
return false;
|
||||
},_poller);
|
||||
}
|
||||
}
|
||||
|
||||
void FFmpegSource::findAsync(int maxWaitMS, const function<void(const MediaSource::Ptr &src)> &cb) {
|
||||
auto src = MediaSource::find(_media_info._schema,
|
||||
_media_info._vhost,
|
||||
_media_info._app,
|
||||
_media_info._streamid,
|
||||
false);
|
||||
if(src || !maxWaitMS){
|
||||
cb(src);
|
||||
return;
|
||||
}
|
||||
|
||||
void *listener_tag = this;
|
||||
//若干秒后执行等待媒体注册超时回调
|
||||
auto onRegistTimeout = _poller->doDelayTask(maxWaitMS,[cb,listener_tag](){
|
||||
//取消监听该事件
|
||||
NoticeCenter::Instance().delListener(listener_tag,Broadcast::kBroadcastMediaChanged);
|
||||
cb(nullptr);
|
||||
return 0;
|
||||
});
|
||||
|
||||
weak_ptr<FFmpegSource> weakSelf = shared_from_this();
|
||||
auto onRegist = [listener_tag,weakSelf,cb,onRegistTimeout](BroadcastMediaChangedArgs) {
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if(!strongSelf) {
|
||||
//本身已经销毁，取消延时任务
|
||||
onRegistTimeout->cancel();
|
||||
NoticeCenter::Instance().delListener(listener_tag,Broadcast::kBroadcastMediaChanged);
|
||||
return;
|
||||
}
|
||||
|
||||
if(!bRegist ||
|
||||
schema != strongSelf->_media_info._schema ||
|
||||
vhost != strongSelf->_media_info._vhost ||
|
||||
app != strongSelf->_media_info._app ||
|
||||
stream != strongSelf->_media_info._streamid){
|
||||
//不是自己感兴趣的事件，忽略之
|
||||
return;
|
||||
}
|
||||
|
||||
//查找的流终于注册上了；取消延时任务，防止多次回调
|
||||
onRegistTimeout->cancel();
|
||||
//取消事件监听
|
||||
NoticeCenter::Instance().delListener(listener_tag,Broadcast::kBroadcastMediaChanged);
|
||||
|
||||
//切换到自己的线程再回复
|
||||
strongSelf->_poller->async([listener_tag,weakSelf,cb](){
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if(!strongSelf) {
|
||||
return;
|
||||
}
|
||||
//再找一遍媒体源，一般能找到
|
||||
strongSelf->findAsync(0,cb);
|
||||
}, false);
|
||||
};
|
||||
//监听媒体注册事件
|
||||
NoticeCenter::Instance().addListener(listener_tag, Broadcast::kBroadcastMediaChanged, onRegist);
|
||||
}
|
||||
|
||||
/**
|
||||
* 定时检查媒体是否在线
|
||||
*/
|
||||
void FFmpegSource::startTimer(int timeout_ms) {
|
||||
weak_ptr<FFmpegSource> weakSelf = shared_from_this();
|
||||
_timer = std::make_shared<Timer>(1, [weakSelf, timeout_ms]() {
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if (!strongSelf) {
|
||||
//自身已经销毁
|
||||
return false;
|
||||
}
|
||||
if (strongSelf->_media_info._host == "127.0.0.1") {
|
||||
//推流给自己的，我们通过检查是否已经注册来判断FFmpeg是否工作正常
|
||||
strongSelf->findAsync(0, [&](const MediaSource::Ptr &src) {
|
||||
//同步查找流
|
||||
if (!src) {
|
||||
//流不在线，重新拉流
|
||||
strongSelf->play(strongSelf->_src_url, strongSelf->_dst_url, timeout_ms,
|
||||
[](const SockException &) {});
|
||||
}
|
||||
});
|
||||
} else {
|
||||
//推流给其他服务器的，我们通过判断FFmpeg进程是否在线，如果FFmpeg推流中断，那么它应该会自动退出
|
||||
if (!strongSelf->_process.wait(false)) {
|
||||
//ffmpeg不在线，重新拉流
|
||||
strongSelf->play(strongSelf->_src_url, strongSelf->_dst_url, timeout_ms, [](const SockException &) {});
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}, _poller);
|
||||
|
||||
NoticeCenter::Instance().delListener(this, Broadcast::kBroadcastStreamNoneReader);
|
||||
NoticeCenter::Instance().addListener(this, Broadcast::kBroadcastStreamNoneReader,[weakSelf](BroadcastStreamNoneReaderArgs) {
|
||||
auto strongSelf = weakSelf.lock();
|
||||
if (!strongSelf) {
|
||||
//自身已经销毁
|
||||
return;
|
||||
}
|
||||
|
||||
if(sender.getVhost() != strongSelf->_media_info._vhost ||
|
||||
sender.getApp() != strongSelf->_media_info._app ||
|
||||
sender.getId() != strongSelf->_media_info._streamid){
|
||||
//不是自己感兴趣的事件，忽略之
|
||||
return;
|
||||
}
|
||||
|
||||
//该流无人观看，我们停止吧
|
||||
if(strongSelf->_onClose){
|
||||
strongSelf->_onClose();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
void FFmpegSource::setOnClose(const function<void()> &cb){
|
||||
_onClose = cb;
|
||||
}
|
||||
Reference in New Issue
Block a user