修复media命令相关bug

2026-06-14 03:55:58 +08:00 · 2019-05-28 10:02:12 +08:00
parent 7e770841d1
commit 2f976214ce
2 changed files with 27 additions and 28 deletions
--- a/src/Shell/ShellCMD.h
+++ b/src/Shell/ShellCMD.h
@@ -0,0 +1,94 @@
+//
+// Created by xzl on 2017/12/1.
+//
+
+#include "Util/CMD.h"
+#include "Common/MediaSource.h"
+using namespace toolkit;
+
+namespace mediakit {
+
+
+class CMD_media: public CMD {
+public:
+    CMD_media(){
+        _parser.reset(new OptionParser([](const std::shared_ptr<ostream> &stream,mINI &ini){
+            MediaSource::for_each_media([&](const string &schema,
+                                            const string &vhost,
+                                            const string &app,
+                                            const string &streamid,
+                                            const MediaSource::Ptr &media){
+                if(!ini["schema"].empty() && ini["schema"] != schema){
+                    //筛选协议不匹配
+                    return;
+                }
+                if(!ini["vhost"].empty() && ini["vhost"] != vhost){
+                    //筛选虚拟主机不匹配
+                    return;
+                }
+                if(!ini["app"].empty() && ini["app"] != app){
+                    //筛选应用名不匹配
+                    return;
+                }
+                if(!ini["stream"].empty() && ini["stream"] != streamid){
+                    //流id不匹配
+                    return;
+                }
+                if(ini.find("list") != ini.end()){
+                    //列出源
+                    (*stream) << "\t"
+                              << schema << "/"
+                              << vhost << "/"
+                              << app << "/"
+                              << streamid
+                              << "\r\n";
+                    return;
+                }
+
+                EventPollerPool::Instance().getPoller()->async([ini,media,stream,schema,vhost,app,streamid](){
+                    if(ini.find("kick") != ini.end()){
+                        //踢出源
+                        do{
+                            if(!media) {
+                                break;
+                            }
+                            if(!media->close(true)) {
+                                break;
+                            }
+                            (*stream) << "\t踢出成功:"
+                                      << schema << "/"
+                                      << vhost << "/"
+                                      << app << "/"
+                                      << streamid
+                                      << "\r\n";
+                            return;
+                        }while(0);
+                        (*stream) << "\t踢出失败:"
+                                  << schema << "/"
+                                  << vhost << "/"
+                                  << app << "/"
+                                  << streamid
+                                  << "\r\n";
+                    }
+                },false);
+
+
+            });
+        }));
+        (*_parser) << Option('k', "kick", Option::ArgNone,nullptr,false, "踢出媒体源", nullptr);
+        (*_parser) << Option('l', "list", Option::ArgNone,nullptr,false, "列出媒体源", nullptr);
+        (*_parser) << Option('S', "schema", Option::ArgRequired,nullptr,false, "协议筛选", nullptr);
+        (*_parser) << Option('v', "vhost", Option::ArgRequired,nullptr,false, "虚拟主机筛选", nullptr);
+        (*_parser) << Option('a', "app", Option::ArgRequired,nullptr,false, "应用名筛选", nullptr);
+        (*_parser) << Option('s', "stream", Option::ArgRequired,nullptr,false, "流id筛选", nullptr);
+    }
+    virtual ~CMD_media() {}
+    const char *description() const override {
+        return "媒体源相关操作.";
+    }
+};
+
+
+
+
+} /* namespace mediakit */