generated from zhangwei/Train_Identify
97 lines
2.5 KiB
C++
97 lines
2.5 KiB
C++
#include "ApiEngine.h"
|
||
|
||
using namespace ai_matrix;
|
||
|
||
ApiEngine::ApiEngine() {}
|
||
|
||
ApiEngine::~ApiEngine() {}
|
||
|
||
APP_ERROR ApiEngine::Init()
|
||
{
|
||
strPort0_ = engineName_ + "_" + std::to_string(engineId_) + "_0";
|
||
strPort1_ = engineName_ + "_" + std::to_string(engineId_) + "_1";
|
||
|
||
LogInfo << "engineId_:" << engineId_ << " ApiEngine Init ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
APP_ERROR ApiEngine::DeInit()
|
||
{
|
||
LogInfo << "ApiEngine engineId_:" << engineId_ << " DeInit ok";
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
APP_ERROR ApiEngine::Process()
|
||
{
|
||
int iRet = APP_ERR_OK;
|
||
|
||
httplib::Server svr;
|
||
|
||
svr.Get("/hello", [](const httplib::Request& req, httplib::Response& res){
|
||
res.set_content("Hello, World!", "text/plain");
|
||
});
|
||
|
||
|
||
|
||
svr.Post("/queryDataSource", [=](const httplib::Request& req, httplib::Response& res){
|
||
std::string add_result = queryDataSource(req.body);
|
||
Json::Value response;
|
||
if (add_result.empty())
|
||
{
|
||
response["success"] = true;
|
||
response["msg"] = add_result;
|
||
}
|
||
else
|
||
{
|
||
response["success"] = false;
|
||
response["msg"] = add_result;
|
||
}
|
||
res.set_content(response.toStyledString(), "application/json");
|
||
});
|
||
|
||
LogInfo << "开启Api服务，端口:" << 8080;
|
||
svr.listen("0.0.0.0", 8080);
|
||
return APP_ERR_OK;
|
||
}
|
||
|
||
/**
|
||
* 查看当前使用的数据源
|
||
* @param req
|
||
* @return
|
||
*/
|
||
std::string ApiEngine::queryDataSource(const std::string &req)
|
||
{
|
||
try {
|
||
Json::CharReaderBuilder readerBuilder;
|
||
std::istringstream iss(req);
|
||
Json::Value root;
|
||
std::string errs;
|
||
bool parsingSuccessful = Json::parseFromStream(readerBuilder, iss, &root, &errs);
|
||
if (parsingSuccessful)
|
||
{
|
||
// std::shared_ptr<VTrainInfo> pTrainInfo = std::make_shared<VTrainInfo>();
|
||
// pTrainInfo->strTrackNo = root["strTrackNo"].asString();
|
||
// if (root.isMember("numCoordinate"))
|
||
MyYaml::GetIns()->GetUseDataSourceConfig();
|
||
|
||
|
||
|
||
// 转发到上传接口引擎中
|
||
// outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(pTrainInfo));
|
||
// 存储到本地数据库
|
||
// outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pTrainInfo));
|
||
|
||
}
|
||
else
|
||
{
|
||
return "接口调用参数异常，所传数据非json：" + req;
|
||
}
|
||
|
||
return "";
|
||
}
|
||
catch (std::exception &e)
|
||
{
|
||
return "接口调用失败！请求内容:" + req + "。 \n异常内容:" + std::string(e.what());
|
||
}
|
||
|
||
} |