mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-01 00:57:28 +08:00
96 lines
3.7 KiB
C++
96 lines
3.7 KiB
C++
//
|
|
// Created by xzl on 2017/12/1.
|
|
//
|
|
|
|
#ifndef SRC_SHELL_SHELLCMD_H_
|
|
#define SRC_SHELL_SHELLCMD_H_
|
|
|
|
#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 */
|
|
|
|
#endif //SRC_SHELL_SHELLCMD_H_
|