添加拉流代理性能测试工具

This commit is contained in:
xiongziliang 2021-01-17 10:32:33 +08:00
parent b2fe8c926c
commit 03d0b6a9ed

44
tests/test_proxy.cpp Normal file
View File

@ -0,0 +1,44 @@
#include <map>
#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;
}