初步实现hls播放器个数计数

src/Common/MediaSource.cpp

@@ -98,6 +98,13 @@ const std::weak_ptr<MediaSourceEvent>& MediaSource::getListener() const{
     return _listener;
 }
 
+int MediaSource::totalReaderCount(){
+    auto listener = _listener.lock();
+    if(!listener){
+        return readerCount();
+    }
+    return listener->totalReaderCount(*this);
+}
 bool MediaSource::seekTo(uint32_t ui32Stamp) {
     auto listener = _listener.lock();
     if(!listener){

src/Common/MediaSource.h

@@ -67,6 +67,9 @@ public:
 
     // 通知无人观看
     virtual void onNoneReader(MediaSource &sender);
+
+    // 观看总人数
+    virtual int totalReaderCount(MediaSource &sender) = 0;
 };
 
 class MediaInfo{
@@ -124,8 +127,10 @@ public:
     void setTrackSource(const std::weak_ptr<TrackSource> &track_src);
     // 设置监听者
     virtual void setListener(const std::weak_ptr<MediaSourceEvent> &listener);
-    // 获取观看者个数
+    // 本协议获取观看者个数，可能返回本协议的观看人数，也可能返回总人数
     virtual int readerCount() = 0;
+    // 观看者个数，包括(hls/rtsp/rtmp)
+    virtual int totalReaderCount();
     // 获取流当前时间戳
     virtual uint32_t getTimeStamp(TrackType trackType) = 0;
 

src/Common/MultiMediaSourceMuxer.h

@@ -30,6 +30,7 @@
 #include "Rtsp/RtspMediaSourceMuxer.h"
 #include "Rtmp/RtmpMediaSourceMuxer.h"
 #include "Record/Recorder.h"
+#include "Record/HlsManager.h"
 
 class MultiMediaSourceMuxer : public MediaSink , public std::enable_shared_from_this<MultiMediaSourceMuxer>{
 public:
@@ -64,6 +65,9 @@ public:
             Recorder::startRecord(Recorder::type_mp4,vhost, strApp, strId, true, false);
         }
 
+        _get_hls_player = [vhost,strApp,strId](){
+            return HlsManager::Instance().hlsPlayerCount(vhost,strApp,strId);
+        };
     }
     virtual ~MultiMediaSourceMuxer(){}
 
@@ -96,8 +100,8 @@ public:
      * 返回总的消费者个数
      * @return
      */
-    int readerCount() const{
-        return (_rtsp ? _rtsp->readerCount() : 0) + (_rtmp ? _rtmp->readerCount() : 0);
+    int totalReaderCount() const{
+        return (_rtsp ? _rtsp->readerCount() : 0) + (_rtmp ? _rtmp->readerCount() : 0) + _get_hls_player();
     }
 
     void setTimeStamp(uint32_t stamp){
@@ -157,6 +161,7 @@ private:
     RtmpMediaSourceMuxer::Ptr _rtmp;
     RtspMediaSourceMuxer::Ptr _rtsp;
     Listener *_listener = nullptr;
+    function<int()> _get_hls_player;
 };