mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2026-06-13 11:41:44 +08:00
添加完整的服务器用例
This commit is contained in:
xiongziliang
324
server/WebApi.cpp
Normal file
324
server/WebApi.cpp
Normal file
@@ -0,0 +1,324 @@
|
||||
#include <signal.h>
|
||||
#include <functional>
|
||||
#include <sstream>
|
||||
#include <unordered_map>
|
||||
#include "jsoncpp/json.h"
|
||||
#include "Util/util.h"
|
||||
#include "Util/logger.h"
|
||||
#include "Util/onceToken.h"
|
||||
#include "Util/NoticeCenter.h"
|
||||
#include "Util/SqlPool.h"
|
||||
#include "Common/config.h"
|
||||
#include "Common/MediaSource.h"
|
||||
#include "Http/HttpRequester.h"
|
||||
#include "Http/HttpSession.h"
|
||||
#include "Network/TcpServer.h"
|
||||
|
||||
using namespace Json;
|
||||
using namespace toolkit;
|
||||
using namespace mediakit;
|
||||
|
||||
#define API_ARGS HttpSession::KeyValue &headerIn, \
|
||||
HttpSession::KeyValue &headerOut, \
|
||||
HttpSession::KeyValue &allArgs, \
|
||||
Json::Value &val
|
||||
|
||||
#define API_REGIST(field, name, ...) \
|
||||
s_map_api.emplace("/index/"#field"/"#name,[](API_ARGS,const HttpSession::HttpResponseInvoker &invoker){ \
|
||||
static auto lam = [&](API_ARGS) __VA_ARGS__ ; \
|
||||
lam(headerIn, headerOut, allArgs, val); \
|
||||
invoker("200 OK", headerOut, val.toStyledString()); \
|
||||
});
|
||||
|
||||
#define API_REGIST_INVOKER(field, name, ...) \
|
||||
s_map_api.emplace("/index/"#field"/"#name,[](API_ARGS,const HttpSession::HttpResponseInvoker &invoker) __VA_ARGS__);
|
||||
|
||||
//异步http api lambad定义
|
||||
typedef std::function<void(API_ARGS,const HttpSession::HttpResponseInvoker &invoker)> AsyncHttpApi;
|
||||
//api列表
|
||||
static map<string, AsyncHttpApi> s_map_api;
|
||||
|
||||
namespace API {
|
||||
typedef enum {
|
||||
SqlFailed = -200,
|
||||
AuthFailed = -100,
|
||||
OtherFailed = -1,
|
||||
Success = 0
|
||||
} ApiErr;
|
||||
|
||||
#define API_FIELD "api."
|
||||
const char kApiDebug[] = API_FIELD"apiDebug";
|
||||
static onceToken token([]() {
|
||||
mINI::Instance()[kApiDebug] = "0";
|
||||
});
|
||||
}//namespace API
|
||||
|
||||
class ApiRetException: public std::runtime_error {
|
||||
public:
|
||||
ApiRetException(const char *str = "success" ,int code = API::Success):runtime_error(str){
|
||||
_code = code;
|
||||
}
|
||||
~ApiRetException() = default;
|
||||
int code(){ return _code; }
|
||||
private:
|
||||
int _code;
|
||||
};
|
||||
|
||||
class AuthException : public ApiRetException {
|
||||
public:
|
||||
AuthException(const char *str):ApiRetException(str,API::AuthFailed){}
|
||||
~AuthException() = default;
|
||||
};
|
||||
|
||||
|
||||
//获取HTTP请求中url参数、content参数
|
||||
static HttpSession::KeyValue getAllArgs(const Parser &parser) {
|
||||
HttpSession::KeyValue allArgs;
|
||||
{
|
||||
//TraceL << parser.FullUrl() << "\r\n" << parser.Content();
|
||||
auto &urlArgs = parser.getUrlArgs();
|
||||
auto contentArgs = parser.parseArgs(parser.Content());
|
||||
for (auto &pr : contentArgs) {
|
||||
allArgs.emplace(pr.first, HttpSession::urlDecode(pr.second));
|
||||
}
|
||||
for (auto &pr : urlArgs) {
|
||||
allArgs.emplace(pr.first, HttpSession::urlDecode(pr.second));
|
||||
}
|
||||
}
|
||||
return allArgs;
|
||||
}
|
||||
|
||||
static inline void addHttpListener(){
|
||||
GET_CONFIG_AND_REGISTER(bool, api_debug, API::kApiDebug);
|
||||
//注册监听kBroadcastHttpRequest事件
|
||||
NoticeCenter::Instance().addListener(nullptr, Broadcast::kBroadcastHttpRequest, [](BroadcastHttpRequestArgs) {
|
||||
auto it = s_map_api.find(parser.Url());
|
||||
if (it == s_map_api.end()) {
|
||||
consumed = false;
|
||||
return;
|
||||
}
|
||||
//该api已被消费
|
||||
consumed = true;
|
||||
AsyncHttpApi api = it->second;
|
||||
//异步执行该api，防止阻塞NoticeCenter
|
||||
EventPollerPool::Instance().getExecutor()->async([api, parser, invoker]() {
|
||||
//执行API
|
||||
Json::Value val;
|
||||
val["code"] = API::Success;
|
||||
HttpSession::KeyValue &headerIn = parser.getValues();
|
||||
HttpSession::KeyValue headerOut;
|
||||
HttpSession::KeyValue allArgs = getAllArgs(parser);
|
||||
headerOut["Content-Type"] = "application/json; charset=utf-8";
|
||||
if(api_debug){
|
||||
auto newInvoker = [invoker,parser,allArgs](const string &codeOut,
|
||||
const HttpSession::KeyValue &headerOut,
|
||||
const string &contentOut){
|
||||
stringstream ss;
|
||||
for(auto &pr : allArgs ){
|
||||
ss << pr.first << " : " << pr.second << "\r\n";
|
||||
}
|
||||
|
||||
DebugL << "request:\r\n" << parser.Method() << " " << parser.FullUrl() << "\r\n"
|
||||
<< "content:\r\n" << parser.Content() << "\r\n"
|
||||
<< "args:\r\n" << ss.str()
|
||||
<< "response:\r\n"
|
||||
<< contentOut << "\r\n";
|
||||
|
||||
invoker(codeOut,headerOut,contentOut);
|
||||
};
|
||||
((HttpSession::HttpResponseInvoker &)invoker) = newInvoker;
|
||||
}
|
||||
|
||||
try {
|
||||
api(headerIn, headerOut, allArgs, val, invoker);
|
||||
} catch(AuthException &ex){
|
||||
val["code"] = API::AuthFailed;
|
||||
val["msg"] = ex.what();
|
||||
invoker("200 OK", headerOut, val.toStyledString());
|
||||
} catch(ApiRetException &ex){
|
||||
val["code"] = ex.code();
|
||||
val["msg"] = ex.what();
|
||||
invoker("200 OK", headerOut, val.toStyledString());
|
||||
} catch(SqlException &ex){
|
||||
val["code"] = API::SqlFailed;
|
||||
val["msg"] = StrPrinter << "操作数据库失败:" << ex.what() << ":" << ex.getSql();
|
||||
WarnL << ex.what() << ":" << ex.getSql();
|
||||
invoker("200 OK", headerOut, val.toStyledString());
|
||||
} catch (std::exception &ex) {
|
||||
val["code"] = API::OtherFailed;
|
||||
val["msg"] = ex.what();
|
||||
invoker("200 OK", headerOut, val.toStyledString());
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
//安装api接口
|
||||
void installWebApi() {
|
||||
addHttpListener();
|
||||
|
||||
/**
|
||||
* 获取线程负载
|
||||
*/
|
||||
API_REGIST_INVOKER(api, getThreadsLoad, {
|
||||
EventPollerPool::Instance().getExecutorDelay([invoker, headerOut](const vector<int> &vecDelay) {
|
||||
Value val;
|
||||
auto vec = EventPollerPool::Instance().getExecutorLoad();
|
||||
int i = 0;
|
||||
for (auto load : vec) {
|
||||
Value obj(objectValue);
|
||||
obj["load"] = load;
|
||||
obj["delay"] = vecDelay[i++];
|
||||
val["data"].append(obj);
|
||||
}
|
||||
invoker("200 OK", headerOut, val.toStyledString());
|
||||
});
|
||||
});
|
||||
|
||||
/**
|
||||
* 获取服务器配置
|
||||
*/
|
||||
API_REGIST(api, getServerConfig, {
|
||||
Value obj;
|
||||
for (auto &pr : mINI::Instance()) {
|
||||
obj[pr.first] = (string &) pr.second;
|
||||
}
|
||||
val["data"].append(obj);
|
||||
});
|
||||
|
||||
/**
|
||||
* 设置服务器配置
|
||||
*/
|
||||
API_REGIST(api, setServerConfig, {
|
||||
auto &ini = mINI::Instance();
|
||||
int changed = 0;
|
||||
for (auto &pr : allArgs) {
|
||||
if (ini.find(pr.first) == ini.end()) {
|
||||
//没有这个key
|
||||
continue;
|
||||
}
|
||||
if (ini[pr.first] == pr.second) {
|
||||
continue;
|
||||
}
|
||||
ini[pr.first] = pr.second;
|
||||
//替换成功
|
||||
++changed;
|
||||
}
|
||||
if (changed > 0) {
|
||||
NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastReloadConfig);
|
||||
ini.dumpFile();
|
||||
}
|
||||
val["changed"] = changed;
|
||||
});
|
||||
|
||||
/**
|
||||
* 获取服务器api列表
|
||||
*/
|
||||
API_REGIST(api,getApiList,{
|
||||
for(auto &pr : s_map_api){
|
||||
val["data"].append(pr.first);
|
||||
}
|
||||
});
|
||||
|
||||
/**
|
||||
* 重启服务器
|
||||
*/
|
||||
API_REGIST(api,restartServer,{
|
||||
EventPollerPool::Instance().getPoller()->doDelayTask(1000,[](){
|
||||
//尝试正常退出
|
||||
::kill(getpid(), SIGINT);
|
||||
|
||||
//3秒后强制退出
|
||||
EventPollerPool::Instance().getPoller()->doDelayTask(3000,[](){
|
||||
exit(0);
|
||||
return 0;
|
||||
});
|
||||
|
||||
return 0;
|
||||
});
|
||||
val["msg"] = "服务器将在一秒后自动重启";
|
||||
});
|
||||
|
||||
|
||||
API_REGIST(api,getMediaList,{
|
||||
//获取所有MediaSource列表
|
||||
val["code"] = 0;
|
||||
val["msg"] = "success";
|
||||
MediaSource::for_each_media([&](const string &schema,
|
||||
const string &vhost,
|
||||
const string &app,
|
||||
const string &stream,
|
||||
const MediaSource::Ptr &media){
|
||||
if(!allArgs["schema"].empty() && allArgs["schema"] != schema){
|
||||
return;
|
||||
}
|
||||
if(!allArgs["vhost"].empty() && allArgs["vhost"] != vhost){
|
||||
return;
|
||||
}
|
||||
if(!allArgs["app"].empty() && allArgs["app"] != app){
|
||||
return;
|
||||
}
|
||||
Value item;
|
||||
item["schema"] = schema;
|
||||
item["vhost"] = vhost;
|
||||
item["app"] = app;
|
||||
item["stream"] = stream;
|
||||
val["data"]["array"].append(item);
|
||||
});
|
||||
});
|
||||
|
||||
API_REGIST(api,kick_pusher,{
|
||||
//踢掉推流器
|
||||
auto src = MediaSource::find(allArgs["schema"],
|
||||
allArgs["vhost"],
|
||||
allArgs["app"],
|
||||
allArgs["stream"]);
|
||||
if(src){
|
||||
bool flag = src->close();
|
||||
val["code"] = flag ? 0 : -1;
|
||||
val["msg"] = flag ? "success" : "kick failed";
|
||||
}else{
|
||||
val["code"] = -2;
|
||||
val["msg"] = "can not find the pusher";
|
||||
}
|
||||
});
|
||||
|
||||
API_REGIST(api,kick_session,{
|
||||
//踢掉tcp会话
|
||||
auto id = allArgs["id"];
|
||||
if(id.empty()){
|
||||
val["code"] = -1;
|
||||
val["msg"] = "illegal parameter:id";
|
||||
return;
|
||||
}
|
||||
auto session = SessionMap::Instance().get(id);
|
||||
if(!session){
|
||||
val["code"] = -2;
|
||||
val["msg"] = "can not find the target";
|
||||
return;
|
||||
}
|
||||
session->safeShutdown();
|
||||
val["code"] = 0;
|
||||
val["msg"] = "success";
|
||||
});
|
||||
|
||||
|
||||
////////////以下是注册的Hook API////////////
|
||||
API_REGIST(hook,on_publish,{
|
||||
//开始推流事件
|
||||
val["code"] = 0;
|
||||
val["msg"] = "success";
|
||||
});
|
||||
|
||||
API_REGIST(hook,on_play,{
|
||||
//开始播放事件
|
||||
val["code"] = 0;
|
||||
val["msg"] = "success";
|
||||
});
|
||||
|
||||
API_REGIST(hook,on_flow_report,{
|
||||
//流量统计hook api
|
||||
val["code"] = 0;
|
||||
val["msg"] = "success";
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user