添加媒体生成源信息

src/Rtp/RtpProcess.h

@@ -26,6 +26,7 @@ namespace mediakit{
 class RtpProcess : public HttpRequestSplitter, public RtpReceiver, public SockInfo, public MediaSinkInterface, public std::enable_shared_from_this<RtpProcess>{
 public:
     typedef std::shared_ptr<RtpProcess> Ptr;
+    friend class RtpProcessHelper;
     RtpProcess(const string &stream_id);
     ~RtpProcess();
 

src/Rtp/RtpSelector.cpp

@@ -145,6 +145,18 @@ int RtpProcessHelper::totalReaderCount(MediaSource &sender) {
     return _process ? _process->totalReaderCount() : sender.totalReaderCount();
 }
 
+MediaOriginType RtpProcessHelper::getOriginType(MediaSource &sender) const{
+    return MediaOriginType::rtp_push;
+}
+
+string RtpProcessHelper::getOriginUrl(MediaSource &sender) const {
+    return _process ? _process->_media_info._full_url : "";
+}
+
+std::shared_ptr<SockInfo> RtpProcessHelper::getOriginSock(MediaSource &sender) const{
+    return _process;
+}
+
 RtpProcess::Ptr &RtpProcessHelper::getProcess() {
     return _process;
 }

src/Rtp/RtpSelector.h

@@ -34,6 +34,12 @@ protected:
     bool close(MediaSource &sender,bool force) override;
     // 观看总人数
     int totalReaderCount(MediaSource &sender) override;
+    // 获取媒体源类型
+    MediaOriginType getOriginType(MediaSource &sender) const override;
+    // 获取媒体源url或者文件路径
+    string getOriginUrl(MediaSource &sender) const override;
+    // 获取媒体源客户端相关信息
+    std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const override;
 
 private:
     weak_ptr<RtpSelector > _parent;