添加媒体生成源信息

src/Common/Device.cpp

@@ -166,5 +166,9 @@ void DevChannel::initAudio(const AudioInfo &info) {
     }
 }
 
+MediaOriginType DevChannel::getOriginType(MediaSource &sender) const {
+    return MediaOriginType::device_chn;
+}
+
 } /* namespace mediakit */
 

src/Common/Device.h

@@ -128,6 +128,9 @@ public:
     void inputPCM(char *pcData, int iDataLen, uint32_t uiStamp);
 #endif //ENABLE_FAAC
 
+private:
+    MediaOriginType getOriginType(MediaSource &sender) const override;
+
 private:
 
 #ifdef ENABLE_X264

src/Common/MediaSource.cpp

@@ -19,6 +19,20 @@ namespace mediakit {
 recursive_mutex s_media_source_mtx;
 MediaSource::SchemaVhostAppStreamMap s_media_source_map;
 
+string getOriginTypeString(MediaOriginType type){
+#define SWITCH_CASE(type) case type : return #type
+    switch (type) {
+        SWITCH_CASE(MediaOriginType::unknown);
+        SWITCH_CASE(MediaOriginType::rtmp_push);
+        SWITCH_CASE(MediaOriginType::rtsp_push);
+        SWITCH_CASE(MediaOriginType::rtp_push);
+        SWITCH_CASE(MediaOriginType::pull);
+        SWITCH_CASE(MediaOriginType::ffmpeg_pull);
+        SWITCH_CASE(MediaOriginType::mp4_vod);
+        SWITCH_CASE(MediaOriginType::device_chn);
+    }
+}
+
 MediaSource::MediaSource(const string &schema, const string &vhost, const string &app, const string &stream_id){
     GET_CONFIG(bool, enableVhost, General::kEnableVhost);
     if (!enableVhost) {
@@ -76,6 +90,30 @@ int MediaSource::totalReaderCount(){
     return listener->totalReaderCount(*this);
 }
 
+MediaOriginType MediaSource::getOriginType() const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return MediaOriginType::unknown;
+    }
+    return listener->getOriginType(const_cast<MediaSource &>(*this));
+}
+
+string MediaSource::getOriginUrl() const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return "";
+    }
+    return listener->getOriginUrl(const_cast<MediaSource &>(*this));
+}
+
+std::shared_ptr<SockInfo> MediaSource::getOriginSock() const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return nullptr;
+    }
+    return listener->getOriginSock(const_cast<MediaSource &>(*this));
+}
+
 bool MediaSource::seekTo(uint32_t stamp) {
     auto listener = _listener.lock();
     if(!listener){
@@ -365,6 +403,7 @@ bool MediaSource::unregist() {
 /////////////////////////////////////MediaInfo//////////////////////////////////////
 
 void MediaInfo::parse(const string &url_in){
+    _full_url = url_in;
     string url = url_in;
     auto pos = url.find("?");
     if (pos != string::npos) {
@@ -486,6 +525,30 @@ void MediaSourceEvent::onReaderChanged(MediaSource &sender, int size){
     }, nullptr);
 }
 
+MediaOriginType MediaSourceEventInterceptor::getOriginType(MediaSource &sender) const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return MediaOriginType::unknown;
+    }
+    return listener->getOriginType(sender);
+}
+
+string MediaSourceEventInterceptor::getOriginUrl(MediaSource &sender) const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return "";
+    }
+    return listener->getOriginUrl(sender);
+}
+
+std::shared_ptr<SockInfo> MediaSourceEventInterceptor::getOriginSock(MediaSource &sender) const {
+    auto listener = _listener.lock();
+    if (!listener) {
+        return nullptr;
+    }
+    return listener->getOriginSock(sender);
+}
+
 bool MediaSourceEventInterceptor::seekTo(MediaSource &sender, uint32_t stamp) {
     auto listener = _listener.lock();
     if (!listener) {

src/Common/MediaSource.h

@@ -22,6 +22,7 @@
 #include "Util/TimeTicker.h"
 #include "Util/NoticeCenter.h"
 #include "Util/List.h"
+#include "Network/Socket.h"
 #include "Rtsp/Rtsp.h"
 #include "Rtmp/Rtmp.h"
 #include "Extension/Track.h"
@@ -36,6 +37,19 @@ namespace toolkit{
 
 namespace mediakit {
 
+enum class MediaOriginType : uint8_t {
+    unknown = 0,
+    rtmp_push ,
+    rtsp_push,
+    rtp_push,
+    pull,
+    ffmpeg_pull,
+    mp4_vod,
+    device_chn
+};
+
+string getOriginTypeString(MediaOriginType type);
+
 class MediaSource;
 class MediaSourceEvent{
 public:
@@ -43,6 +57,13 @@ public:
     MediaSourceEvent(){};
     virtual ~MediaSourceEvent(){};
 
+    // 获取媒体源类型
+    virtual MediaOriginType getOriginType(MediaSource &sender) const { return MediaOriginType::unknown; }
+    // 获取媒体源url或者文件路径
+    virtual string getOriginUrl(MediaSource &sender) const { return ""; }
+    // 获取媒体源客户端相关信息
+    virtual std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const { return nullptr; }
+
     // 通知拖动进度条
     virtual bool seekTo(MediaSource &sender, uint32_t stamp) { return false; }
     // 通知其停止产生流
@@ -76,6 +97,10 @@ public:
     MediaSourceEventInterceptor(){}
     ~MediaSourceEventInterceptor() override {}
 
+    MediaOriginType getOriginType(MediaSource &sender) const override;
+    string getOriginUrl(MediaSource &sender) const override;
+    std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const override;
+
     bool seekTo(MediaSource &sender, uint32_t stamp) override;
     bool close(MediaSource &sender, bool force) override;
     int totalReaderCount(MediaSource &sender) override;
@@ -102,6 +127,7 @@ public:
     void parse(const string &url);
 
 public:
+    string _full_url;
     string _schema;
     string _host;
     string _port;
@@ -156,6 +182,13 @@ public:
     // 观看者个数，包括(hls/rtsp/rtmp)
     virtual int totalReaderCount();
 
+    // 获取媒体源类型
+    MediaOriginType getOriginType() const;
+    // 获取媒体源url或者文件路径
+    string getOriginUrl() const;
+    // 获取媒体源客户端相关信息
+    std::shared_ptr<SockInfo> getOriginSock() const;
+
     // 拖动进度条
     bool seekTo(uint32_t stamp);
     // 关闭该流