整理MediaSource派生类代码

2026-07-08 23:10:04 +08:00 · 2019-12-25 11:04:12 +08:00
parent bb4e8b73b5
commit 1bfe4937cd
10 changed files with 246 additions and 153 deletions
--- a/src/Rtmp/RtmpMediaSource.h
+++ b/src/Rtmp/RtmpMediaSource.h
@@ -48,118 +48,161 @@ using namespace toolkit;

 namespace mediakit {

-class RtmpMediaSource: public MediaSource ,public RingDelegate<RtmpPacket::Ptr> {
+/**
+ * rtmp媒体源的数据抽象
+ * rtmp有关键的三要素，分别是metadata、config帧，普通帧
+ * 其中metadata是非必须的，有些编码格式也没有config帧(比如MP3)
+ * 只要生成了这三要素，那么要实现rtmp推流、rtmp服务器就很简单了
+ * rtmp推拉流协议中，先传递metadata，然后传递config帧，然后一直传递普通帧
+ */
+class RtmpMediaSource : public MediaSource, public RingDelegate<RtmpPacket::Ptr> {
 public:
 	typedef std::shared_ptr<RtmpMediaSource> Ptr;
 	typedef RingBuffer<RtmpPacket::Ptr> RingType;

+	/**
+	 * 构造函数
+	 * @param vhost 虚拟主机名
+	 * @param app 应用名
+	 * @param stream_id 流id
+	 * @param ring_size 可以设置固定的环形缓冲大小，0则自适应
+	 */
 	RtmpMediaSource(const string &vhost,
-	                const string &strApp,
-	                const string &strId,
-	                int ringSize = 0) :
-			MediaSource(RTMP_SCHEMA,vhost,strApp,strId), _ringSize(ringSize) {
+					const string &app,
+					const string &stream_id,
+					int ring_size = 0) :
+			MediaSource(RTMP_SCHEMA, vhost, app, stream_id), _ring_size(ring_size) {
 		_metadata = TitleMeta().getMetadata();
 	}

 	virtual ~RtmpMediaSource() {}

+	/**
+	 * 	获取媒体源的环形缓冲
+	 */
 	const RingType::Ptr &getRing() const {
-		//获取媒体源的rtp环形缓冲
-		return _pRing;
+		return _ring;
 	}

+	/**
+	 * 获取播放器个数
+	 * @return
+	 */
 	int readerCount() override {
-        return _pRing ? _pRing->readerCount() : 0;
+		return _ring ? _ring->readerCount() : 0;
 	}

+	/**
+	 * 获取metadata
+	 */
 	const AMFValue &getMetaData() const {
-		lock_guard<recursive_mutex> lock(_mtxMap);
+		lock_guard<recursive_mutex> lock(_mtx);
 		return _metadata;
 	}
-	template<typename FUN>
-	void getConfigFrame(const FUN &f) {
-		lock_guard<recursive_mutex> lock(_mtxMap);
-		for (auto &pr : _mapCfgFrame) {
+
+	/**
+	 * 获取所有的config帧
+	 */
+	template<typename FUNC>
+	void getConfigFrame(const FUNC &f) {
+		lock_guard<recursive_mutex> lock(_mtx);
+		for (auto &pr : _config_frame_map) {
 			f(pr.second);
 		}
 	}

-	virtual void onGetMetaData(const AMFValue &metadata) {
-		lock_guard<recursive_mutex> lock(_mtxMap);
+	/**
+	 * 设置metadata
+	 */
+	virtual void setMetaData(const AMFValue &metadata) {
+		lock_guard<recursive_mutex> lock(_mtx);
 		_metadata = metadata;
 	}

-    void onWrite(const RtmpPacket::Ptr &pkt,bool isKey = true) override {
-		lock_guard<recursive_mutex> lock(_mtxMap);
+	/**
+	 * 输入rtmp包
+	 * @param pkt rtmp包
+	 * @param isKey 是否为关键帧
+	 */
+	void onWrite(const RtmpPacket::Ptr &pkt, bool isKey = true) override {
+		lock_guard<recursive_mutex> lock(_mtx);
 		if (pkt->isCfgFrame()) {
-			_mapCfgFrame[pkt->typeId] = pkt;
-            return;
+			_config_frame_map[pkt->typeId] = pkt;
+			return;
 		}

-        if(!_pRing){
-            weak_ptr<RtmpMediaSource> weakSelf = dynamic_pointer_cast<RtmpMediaSource>(shared_from_this());
-            _pRing = std::make_shared<RingType>(_ringSize,[weakSelf](const EventPoller::Ptr &,int size,bool){
-                auto strongSelf = weakSelf.lock();
-                if(!strongSelf){
-                    return;
-                }
-                strongSelf->onReaderChanged(size);
-            });
-            onReaderChanged(0);
-        }
+		if (!_ring) {
+			weak_ptr<RtmpMediaSource> weakSelf = dynamic_pointer_cast<RtmpMediaSource>(shared_from_this());
+			auto lam = [weakSelf](const EventPoller::Ptr &, int size, bool) {
+				auto strongSelf = weakSelf.lock();
+				if (!strongSelf) {
+					return;
+				}
+				strongSelf->onReaderChanged(size);
+			};
+			_ring = std::make_shared<RingType>(_ring_size, std::move(lam));
+			onReaderChanged(0);

-		//如果输入了非config帧，那么说明不再可能获取config帧以及metadata,所以我们强制其为已注册
-		if(!_registed){
-			_registed = true;
+			//如果输入了非config帧，
+			//那么说明不再可能获取config帧以及metadata,
+			//所以我们强制其为已注册
 			regist();
 		}
-
-		_mapStamp[pkt->typeId] = pkt->timeStamp;
-		_pRing->write(pkt,pkt->isVideoKeyFrame());
+		_track_stamps_map[pkt->typeId] = pkt->timeStamp;
+		_ring->write(pkt, pkt->isVideoKeyFrame());
 		checkNoneReader();
-    }
+	}

+	/**
+	 * 获取当前时间戳
+	 */
 	uint32_t getTimeStamp(TrackType trackType) override {
-		lock_guard<recursive_mutex> lock(_mtxMap);
-		switch (trackType){
+		lock_guard<recursive_mutex> lock(_mtx);
+		switch (trackType) {
 			case TrackVideo:
-				return _mapStamp[MSG_VIDEO];
+				return _track_stamps_map[MSG_VIDEO];
 			case TrackAudio:
-				return _mapStamp[MSG_AUDIO];
+				return _track_stamps_map[MSG_AUDIO];
 			default:
-				return MAX(_mapStamp[MSG_VIDEO],_mapStamp[MSG_AUDIO]);
+				return MAX(_track_stamps_map[MSG_VIDEO], _track_stamps_map[MSG_AUDIO]);
 		}
 	}

 private:
-    void onReaderChanged(int size){
-	    //我们记录最后一次活动时间
-        _readerTicker.resetTime();
-        if(size != 0 || readerCount() != 0){
-            //还有消费者正在观看该流
-            _asyncEmitNoneReader = false;
-            return;
-        }
-        _asyncEmitNoneReader  = true;
-    }
+	/**
+	 * 每次增减消费者都会触发该函数
+	 */
+	void onReaderChanged(int size) {
+		//我们记录最后一次活动时间
+		_reader_changed_ticker.resetTime();
+		if (size != 0 || readerCount() != 0) {
+			//还有消费者正在观看该流
+			_async_emit_none_reader = false;
+			return;
+		}
+		_async_emit_none_reader = true;
+	}

-    void checkNoneReader(){
-        GET_CONFIG(int,stream_none_reader_delay,General::kStreamNoneReaderDelayMS);
-        if(_asyncEmitNoneReader && _readerTicker.elapsedTime() > stream_none_reader_delay){
-            _asyncEmitNoneReader = false;
-            onNoneReader();
-        }
-    }
+	/**
+	 * 检查是否无人消费该流，
+	 * 如果无人消费且超过一定时间会触发onNoneReader事件
+	 */
+	void checkNoneReader() {
+		GET_CONFIG(int, stream_none_reader_delay, General::kStreamNoneReaderDelayMS);
+		if (_async_emit_none_reader && _reader_changed_ticker.elapsedTime() > stream_none_reader_delay) {
+			_async_emit_none_reader = false;
+			onNoneReader();
+		}
+	}
 protected:
+	int _ring_size;
+	bool _async_emit_none_reader = false;
+	mutable recursive_mutex _mtx;
+	Ticker _reader_changed_ticker;
 	AMFValue _metadata;
-    unordered_map<int, RtmpPacket::Ptr> _mapCfgFrame;
-	unordered_map<int,uint32_t> _mapStamp;
-	mutable recursive_mutex _mtxMap;
-	RingBuffer<RtmpPacket::Ptr>::Ptr _pRing; //rtp环形缓冲
-	int _ringSize;
-	Ticker _readerTicker;
-    bool _asyncEmitNoneReader = false;
-	bool _registed = false;
+	RingBuffer<RtmpPacket::Ptr>::Ptr _ring;
+	unordered_map<int, uint32_t> _track_stamps_map;
+	unordered_map<int, RtmpPacket::Ptr> _config_frame_map;
 };

 } /* namespace mediakit */
--- a/src/Rtmp/RtmpMediaSourceMuxer.h
+++ b/src/Rtmp/RtmpMediaSourceMuxer.h
@@ -54,7 +54,7 @@ public:
    }

    void onAllTrackReady(){
-        _mediaSouce->onGetMetaData(getMetadata());
+        _mediaSouce->setMetaData(getMetadata());
    }

    // 设置TrackSource
--- a/src/Rtmp/RtmpPlayerImp.h
+++ b/src/Rtmp/RtmpPlayerImp.h
@@ -65,7 +65,7 @@ private:
    bool onCheckMeta(const AMFValue &val) override {
        _pRtmpMediaSrc = dynamic_pointer_cast<RtmpMediaSource>(_pMediaSrc);
        if(_pRtmpMediaSrc){
-            _pRtmpMediaSrc->onGetMetaData(val);
+            _pRtmpMediaSrc->setMetaData(val);
        }
        _delegate.reset(new RtmpDemuxer(val));
        return true;
--- a/src/Rtmp/RtmpSession.cpp
+++ b/src/Rtmp/RtmpSession.cpp
@@ -438,7 +438,7 @@ void RtmpSession::setMetaData(AMFDecoder &dec) {
 	}
    auto metadata = dec.load<AMFValue>();
 //    dumpMetadata(metadata);
-	_pPublisherSrc->onGetMetaData(metadata);
+    _pPublisherSrc->setMetaData(metadata);
 }

 void RtmpSession::onProcessCmd(AMFDecoder &dec) {
--- a/src/Rtmp/RtmpToRtspMediaSource.h
+++ b/src/Rtmp/RtmpToRtspMediaSource.h
@@ -57,12 +57,12 @@ public:
 	}
 	virtual ~RtmpToRtspMediaSource(){}

-	void onGetMetaData(const AMFValue &metadata) override {
+	void setMetaData(const AMFValue &metadata) override {
 		if(!_demuxer){
 			//在未调用onWrite前，设置Metadata能触发生成RtmpDemuxer
 			_demuxer = std::make_shared<RtmpDemuxer>(metadata);
 		}
-		RtmpMediaSource::onGetMetaData(metadata);
+		RtmpMediaSource::setMetaData(metadata);
 	}

 	void onWrite(const RtmpPacket::Ptr &pkt,bool key_pos = true) override {