限制未就绪状态track的最大帧缓存数量

This commit is contained in:
xia-chu 2021-04-26 14:58:06 +08:00
parent 259247132a
commit a28c76a33c

View File

@ -11,10 +11,13 @@
#include "MediaSink.h"
//最多等待未初始化的Track 10秒超时之后会忽略未初始化的Track
#define MAX_WAIT_MS_READY 10000
static size_t constexpr kMaxWaitReadyMS= 10000;
//如果直播流只有单Track最多等待3秒超时后未收到其他Track的数据则认为是单Track
#define MAX_WAIT_MS_ADD_TRACK 3000
static size_t constexpr kMaxAddTrackMS = 3000;
//如果track未就绪我们先缓存帧数据但是有最大个数限制(100帧时大约4秒),防止内存溢出
static size_t constexpr kMaxUnreadyFrame = 100;
namespace mediakit{
@ -37,8 +40,14 @@ void MediaSink::addTrack(const Track::Ptr &track_in) {
if (_all_track_ready) {
onTrackFrame(frame);
} else {
auto &frame_unread = _frame_unread[frame->getCodecId()];
if (frame_unread.size() > kMaxUnreadyFrame) {
//未就绪的的track不能缓存太多的帧否则可能内存溢出
frame_unread.clear();
WarnL << "cached frame of unready track(" << frame->getCodecName() << ") is too much, now cleared";
}
//还有Track未就绪先缓存之
_frame_unread[frame->getCodecId()].emplace_back(Frame::getCacheAbleFrame(frame));
frame_unread.emplace_back(Frame::getCacheAbleFrame(frame));
}
}));
}
@ -84,7 +93,7 @@ void MediaSink::checkTrackIfReady(const Track::Ptr &track){
}
if(!_all_track_ready){
if(_ticker.elapsedTime() > MAX_WAIT_MS_READY){
if(_ticker.elapsedTime() > kMaxWaitReadyMS){
//如果超过规定时间那么不再等待并忽略未准备好的Track
emitAllTrackReady();
return;
@ -101,7 +110,7 @@ void MediaSink::checkTrackIfReady(const Track::Ptr &track){
return;
}
if(_track_map.size() == 1 && _ticker.elapsedTime() > MAX_WAIT_MS_ADD_TRACK){
if(_track_map.size() == 1 && _ticker.elapsedTime() > kMaxAddTrackMS){
//如果只有一个Track那么在该Track添加后我们最多还等待若干时间(可能后面还会添加Track)
emitAllTrackReady();
return;