Skip to content

Z
ZLMediaKit
Commit b5c1726c by xiongziliang
Options

rtmp未找到流后延迟重试

parent 36f688f1
正在显示 包含 48 行增加 和 10 行删除
src/Rtmp/RtmpSession.cpp
...@@ -75,6 +75,12 @@ void RtmpSession::onManager() { ...@@ -75,6 +75,12 @@ void RtmpSession::onManager() {
shutdown(); shutdown();
} }
} }
if(m_delayTask){
if(time(NULL) > m_iTaskTimeLine){
m_delayTask();
m_delayTask = nullptr;
}
}
} }
void RtmpSession::onRecv(const Buffer::Ptr &pBuf) { void RtmpSession::onRecv(const Buffer::Ptr &pBuf) {
...@@ -196,18 +202,35 @@ void RtmpSession::onCmd_deleteStream(AMFDecoder &dec) { ...@@ -196,18 +202,35 @@ void RtmpSession::onCmd_deleteStream(AMFDecoder &dec) {
throw std::runtime_error(StrPrinter << "Stop publishing." << endl); throw std::runtime_error(StrPrinter << "Stop publishing." << endl);
} }
void RtmpSession::doPlay(AMFDecoder &dec){ void RtmpSession::doPlayResponse(const string &err,bool tryDelay) {
auto onRes = [this](const string &err) { //获取流对象
auto src = dynamic_pointer_cast<RtmpMediaSource>(MediaSource::find(RTMP_SCHEMA, auto src = dynamic_pointer_cast<RtmpMediaSource>(MediaSource::find(RTMP_SCHEMA,
m_mediaInfo.m_vhost, m_mediaInfo.m_vhost,
m_mediaInfo.m_app, m_mediaInfo.m_app,
m_mediaInfo.m_streamid, m_mediaInfo.m_streamid,
true)); true));
if(src ){
if(!src->ready()){
//流未准备好那么相当于没有
src = nullptr;
}
}
//是否鉴权成功
bool authSuccess = err.empty(); bool authSuccess = err.empty();
bool ok = (src.operator bool() && authSuccess); if(authSuccess && !src && tryDelay ){
if(ok){ //校验成功，但是流不存在而导致的不能播放，我们看看该流延时几秒后是否确实不能播放
ok = ok && src->ready(); doDelay(3,[this](){
//延时后就不再延时重试了
doPlayResponse("",false);
});
return;
} }
///////回复流程///////
//是否播放成功
bool ok = (src.operator bool() && authSuccess);
if (ok) { if (ok) {
//stream begin //stream begin
sendUserControl(CONTROL_STREAM_BEGIN, STREAM_MEDIA); sendUserControl(CONTROL_STREAM_BEGIN, STREAM_MEDIA);
...@@ -298,26 +321,27 @@ void RtmpSession::doPlay(AMFDecoder &dec){ ...@@ -298,26 +321,27 @@ void RtmpSession::doPlay(AMFDecoder &dec){
if (src->getRing()->readerCount() == 1) { if (src->getRing()->readerCount() == 1) {
src->seekTo(0); src->seekTo(0);
} }
}; }
void RtmpSession::doPlay(AMFDecoder &dec){
weak_ptr<RtmpSession> weakSelf = dynamic_pointer_cast<RtmpSession>(shared_from_this()); weak_ptr<RtmpSession> weakSelf = dynamic_pointer_cast<RtmpSession>(shared_from_this());
Broadcast::AuthInvoker invoker = [weakSelf,onRes](const string &err){ Broadcast::AuthInvoker invoker = [weakSelf](const string &err){
auto strongSelf = weakSelf.lock(); auto strongSelf = weakSelf.lock();
if(!strongSelf){ if(!strongSelf){
return; return;
} }
strongSelf->async([weakSelf,onRes,err](){ strongSelf->async([weakSelf,err](){
auto strongSelf = weakSelf.lock(); auto strongSelf = weakSelf.lock();
if(!strongSelf){ if(!strongSelf){
return; return;
} }
onRes(err); strongSelf->doPlayResponse(err,true);
}); });
}; };
auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,m_mediaInfo,invoker,*this); auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,m_mediaInfo,invoker,*this);
if(!flag){ if(!flag){
//该事件无人监听,默认不鉴权 //该事件无人监听,默认不鉴权
onRes(""); doPlayResponse("",true);
} }
} }
void RtmpSession::onCmd_play2(AMFDecoder &dec) { void RtmpSession::onCmd_play2(AMFDecoder &dec) {
...@@ -411,6 +435,7 @@ void RtmpSession::onRtmpChunk(RtmpPacket &chunkData) { ...@@ -411,6 +435,7 @@ void RtmpSession::onRtmpChunk(RtmpPacket &chunkData) {
if (!m_pPublisherSrc) { if (!m_pPublisherSrc) {
throw std::runtime_error("Not a rtmp publisher!"); throw std::runtime_error("Not a rtmp publisher!");
} }
chunkData.timeStamp = m_stampTicker[chunkData.typeId % 2].elapsedTime();
m_pPublisherSrc->onGetMedia(std::make_shared<RtmpPacket>(chunkData)); m_pPublisherSrc->onGetMedia(std::make_shared<RtmpPacket>(chunkData));
if(!m_bPublisherSrcRegisted && m_pPublisherSrc->ready()){ if(!m_bPublisherSrcRegisted && m_pPublisherSrc->ready()){
m_bPublisherSrcRegisted = true; m_bPublisherSrcRegisted = true;
...@@ -457,5 +482,11 @@ void RtmpSession::onSendMedia(const RtmpPacket::Ptr &pkt) { ...@@ -457,5 +482,11 @@ void RtmpSession::onSendMedia(const RtmpPacket::Ptr &pkt) {
sendRtmp(pkt->typeId, pkt->streamId, pkt->strBuf, modifiedStamp, pkt->chunkId , true); sendRtmp(pkt->typeId, pkt->streamId, pkt->strBuf, modifiedStamp, pkt->chunkId , true);
} }
void RtmpSession::doDelay(int delaySec, const std::function<void()> &fun) {
m_delayTask = fun;
m_iTaskTimeLine = time(NULL) + delaySec;
}
} /* namespace Rtmp */ } /* namespace Rtmp */
} /* namespace ZL */ } /* namespace ZL */
src/Rtmp/RtmpSession.h
...@@ -57,6 +57,7 @@ private: ...@@ -57,6 +57,7 @@ private:
MediaInfo m_mediaInfo; MediaInfo m_mediaInfo;
double m_dNowReqID = 0; double m_dNowReqID = 0;
Ticker m_ticker;//数据接收时间 Ticker m_ticker;//数据接收时间
SmoothTicker m_stampTicker[2];//时间戳生产器
typedef void (RtmpSession::*rtmpCMDHandle)(AMFDecoder &dec); typedef void (RtmpSession::*rtmpCMDHandle)(AMFDecoder &dec);
static unordered_map<string, rtmpCMDHandle> g_mapCmd; static unordered_map<string, rtmpCMDHandle> g_mapCmd;
...@@ -77,6 +78,7 @@ private: ...@@ -77,6 +78,7 @@ private:
void onCmd_play(AMFDecoder &dec); void onCmd_play(AMFDecoder &dec);
void onCmd_play2(AMFDecoder &dec); void onCmd_play2(AMFDecoder &dec);
void doPlay(AMFDecoder &dec); void doPlay(AMFDecoder &dec);
void doPlayResponse(const string &err,bool tryDelay);
void onCmd_seek(AMFDecoder &dec); void onCmd_seek(AMFDecoder &dec);
void onCmd_pause(AMFDecoder &dec); void onCmd_pause(AMFDecoder &dec);
void setMetaData(AMFDecoder &dec); void setMetaData(AMFDecoder &dec);
...@@ -104,6 +106,11 @@ private: ...@@ -104,6 +106,11 @@ private:
safeShutdown(); safeShutdown();
return true; return true;
} }
void doDelay(int delaySec,const std::function<void()> &fun);
std::function<void()> m_delayTask;
uint32_t m_iTaskTimeLine = 0;
}; };
} /* namespace Rtmp */ } /* namespace Rtmp */
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论