2021-01-17 10:35:01 +08:00
|
|
|
|
#include <map>
|
2021-01-17 10:32:33 +08:00
|
|
|
|
#include <signal.h>
|
|
|
|
|
|
#include <iostream>
|
|
|
|
|
|
#include "Util/logger.h"
|
|
|
|
|
|
#include "Network/TcpServer.h"
|
|
|
|
|
|
#include "Common/config.h"
|
|
|
|
|
|
#include "Player/PlayerProxy.h"
|
|
|
|
|
|
#include "Http/WebSocketSession.h"
|
|
|
|
|
|
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
using namespace toolkit;
|
|
|
|
|
|
using namespace mediakit;
|
|
|
|
|
|
|
|
|
|
|
|
//此程序为zlm的拉流代理性能测试工具，用于测试拉流代理性能
|
|
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
|
|
{
|
|
|
|
|
|
Logger::Instance().add(std::make_shared<ConsoleChannel>());
|
|
|
|
|
|
Logger::Instance().setWriter(std::make_shared<AsyncLogWriter>());
|
|
|
|
|
|
|
|
|
|
|
|
EventPollerPool::Instance().preferCurrentThread(false);
|
|
|
|
|
|
mINI::Instance()[General::kRtspDemand] = atoi(argv[3]);
|
|
|
|
|
|
mINI::Instance()[General::kRtmpDemand] = atoi(argv[3]);
|
|
|
|
|
|
mINI::Instance()[General::kHlsDemand] = 1;
|
|
|
|
|
|
mINI::Instance()[General::kTSDemand] = 1;
|
|
|
|
|
|
mINI::Instance()[General::kFMP4Demand] = 1;
|
|
|
|
|
|
|
|
|
|
|
|
string url = argv[1];
|
|
|
|
|
|
int count = atoi(argv[2]);
|
|
|
|
|
|
map<string, PlayerProxy::Ptr> proxyMap;
|
|
|
|
|
|
for (auto i = 0; i < count; ++i) {
|
|
|
|
|
|
auto stream = to_string(i);
|
|
|
|
|
|
PlayerProxy::Ptr player(new PlayerProxy(DEFAULT_VHOST, "live", stream, false, false));
|
|
|
|
|
|
(*player)[kRtpType] = Rtsp::RTP_TCP;
|
|
|
|
|
|
player->play(url);
|
|
|
|
|
|
proxyMap.emplace(stream, player);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static semaphore sem;
|
|
|
|
|
|
signal(SIGINT, [](int) { sem.post(); });// 设置退出信号
|
|
|
|
|
|
sem.wait();
|
|
|
|
|
|
}
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
