完善获取丢包率pull request; 新增MediaSource::getOwnerPoller接口

src/Common/MediaSource.cpp

@@ -1,4 +1,4 @@
-/*
+/*
  * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
  *
  * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
@@ -221,6 +221,14 @@ int MediaSource::getLossRate(mediakit::TrackType type) {
     return listener->getLossRate(*this, type);
 }
 
+toolkit::EventPoller::Ptr MediaSource::getOwnerPoller() {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return nullptr;
+    }
+    return listener->getOwnerPoller(*this);
+}
+
 void MediaSource::onReaderChanged(int size) {
     auto listener = _listener.lock();
     if (listener) {
@@ -706,6 +714,15 @@ int MediaSourceEventInterceptor::getLossRate(MediaSource &sender, TrackType type
     return -1; //异常返回-1
 }
 
+toolkit::EventPoller::Ptr MediaSourceEventInterceptor::getOwnerPoller(MediaSource &sender) {
+    auto listener = _listener.lock();
+    if (listener) {
+        return listener->getOwnerPoller(sender);
+    }
+    return nullptr;
+}
+
+
 bool MediaSourceEventInterceptor::setupRecord(MediaSource &sender, Recorder::type type, bool start, const string &custom_path, size_t max_second) {
     auto listener = _listener.lock();
     if (!listener) {

src/Common/MediaSource.h

@@ -1,4 +1,4 @@
-/*
+/*
  * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
  *
  * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
@@ -77,8 +77,10 @@ public:
     virtual void onReaderChanged(MediaSource &sender, int size);
     //流注册或注销事件
     virtual void onRegist(MediaSource &sender, bool regist) {};
-
-    virtual int getLossRate(MediaSource &sender, TrackType type) {return -1;};
+    // 获取丢包率
+    virtual int getLossRate(MediaSource &sender, TrackType type) { return -1; }
+    // 获取所在线程
+    virtual toolkit::EventPoller::Ptr getOwnerPoller(MediaSource &sender) { return nullptr; }
 
     ////////////////////////仅供MultiMediaSourceMuxer对象继承////////////////////////
     // 开启或关闭录制
@@ -145,6 +147,8 @@ public:
     void startSendRtp(MediaSource &sender, const SendRtpArgs &args, const std::function<void(uint16_t, const toolkit::SockException &)> cb) override;
     bool stopSendRtp(MediaSource &sender, const std::string &ssrc) override;
     int getLossRate(MediaSource &sender, TrackType type) override;
+    toolkit::EventPoller::Ptr getOwnerPoller(MediaSource &sender) override;
+
 private:
     std::weak_ptr<MediaSourceEvent> _listener;
 };
@@ -297,6 +301,10 @@ public:
     void startSendRtp(const MediaSourceEvent::SendRtpArgs &args, const std::function<void(uint16_t, const toolkit::SockException &)> cb);
     // 停止发送ps-rtp
     bool stopSendRtp(const std::string &ssrc);
+    // 获取丢包率
+    int getLossRate(mediakit::TrackType type);
+    // 获取所在线程
+    toolkit::EventPoller::Ptr getOwnerPoller();
 
     ////////////////static方法，查找或生成MediaSource////////////////