ZLMediaKit/src/Shell/ShellCMD.h
xia-chu 5686027fc2 修复MediaSource::close线程安全问题
主动或无人观看关闭流可能会由于线程安全问题导致崩溃
2025-09-09 22:02:20 +08:00

67 lines
2.9 KiB
C++

/*
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
*
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
*
* Use of this source code is governed by MIT-like license that can be found in the
* LICENSE file in the root of the source tree. All contributing project authors
* may be found in the AUTHORS file in the root of the source tree.
*/
#ifndef SRC_SHELL_SHELLCMD_H_
#define SRC_SHELL_SHELLCMD_H_
#include "Util/CMD.h"
#include "Common/MediaSource.h"
namespace mediakit {
class CMD_media : public toolkit::CMD {
public:
CMD_media() {
_parser.reset(new toolkit::OptionParser([](const std::shared_ptr<std::ostream> &stream, toolkit::mINI &ini) {
MediaSource::for_each_media([&](const MediaSource::Ptr &media) {
if (ini.find("list") != ini.end()) {
// 列出源 [AUTO-TRANSLATED:cee6cf16]
// List sources
(*stream) << "\t" << media->getUrl() << "\r\n";
return;
}
toolkit::EventPollerPool::Instance().getPoller()->async([ini, media, stream]() {
if (ini.find("kick") != ini.end()) {
// 踢出源 [AUTO-TRANSLATED:67fdbfaa]
// Kick out sources
do {
if (!media) {
break;
}
media->getOwnerPoller()->async([media]() {
media->close(true);
});
(*stream) << "\t踢出成功:" << media->getUrl() << "\r\n";
return;
} while (0);
(*stream) << "\t踢出失败:" << media->getUrl() << "\r\n";
}
}, false);
}, ini["schema"], ini["vhost"], ini["app"], ini["stream"]);
}));
(*_parser) << toolkit::Option('k', "kick", toolkit::Option::ArgNone, nullptr, false, "踢出媒体源", nullptr);
(*_parser) << toolkit::Option('l', "list", toolkit::Option::ArgNone, nullptr, false, "列出媒体源", nullptr);
(*_parser) << toolkit::Option('S', "schema", toolkit::Option::ArgRequired, nullptr, false, "协议筛选", nullptr);
(*_parser) << toolkit::Option('v', "vhost", toolkit::Option::ArgRequired, nullptr, false, "虚拟主机筛选", nullptr);
(*_parser) << toolkit::Option('a', "app", toolkit::Option::ArgRequired, nullptr, false, "应用名筛选", nullptr);
(*_parser) << toolkit::Option('s', "stream", toolkit::Option::ArgRequired, nullptr, false, "流id筛选", nullptr);
}
const char *description() const override {
return "媒体源相关操作.";
}
};
} /* namespace mediakit */
#endif //SRC_SHELL_SHELLCMD_H_