From e603756fc30192b0b4ab67f530f02f10fe0bcee0 Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期三, 10 四月 2019 19:17:14 +0800
Subject: [PATCH] 解决nsq消费者的topic名称bug
---
QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp | 429 +++++++++++++++++++++++++++++++++++------------------
1 files changed, 281 insertions(+), 148 deletions(-)
diff --git a/QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp b/QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp
index b96f6bd..6432932 100644
--- a/QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp
+++ b/QiaoJiaSystem/VideoAnalysFromHC/RtspAnalysElement.cpp
@@ -26,8 +26,28 @@
}
RtspAnalysElement::RtspAnalysElement(LDBTool *_dbTool) : m_lDBTool(_dbTool), maxCount(50), currentCount(0) {
- init();
resetFdfs();
+ init();
+ {
+ IMPORT_APP_ARGS;
+ //nsq set callback func
+ DBG("NsqMsgConsumer Init");
+ DBG("argv[0]:" << argv[0]);
+ if(argv[0][0] == '.' && argv[0][1] == '/'){
+ string pName(argv[0]+2, argv[0]+strlen(argv[0]));
+ DBG("pName:" << pName);
+ nsqMsgConsumer = new BasicMsg::Nsq::NsqMsgConsumer("127.0.0.1", "4150", pName,
+ to_string(appPref.getIntData("RpcVTIMPort")));
+ }else{
+ string pName(argv[0]);
+ DBG("pName:" << pName);
+ nsqMsgConsumer = new BasicMsg::Nsq::NsqMsgConsumer("127.0.0.1", "4150", pName,
+ to_string(appPref.getIntData("RpcVTIMPort")));
+ }
+ nsqMsgConsumer->setMessageCallback(std::bind(&OnMsgFunc, std::placeholders::_1, this));
+ nsqMsgConsumer->init();
+ DBG("NsqMsgConsumer Init END!!");
+ }
{
// std::thread httpServer([&]() {
// //#todo port
@@ -67,112 +87,189 @@
return rtsp_url;
}
-void RtspAnalysElement::init() {
- auto lst = m_lDBTool->searchCamDevTableAll();
- auto lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();
- appPref.setIntData("n_cut_max_duration", lst_dev.n_cut_max_duration);
+void RtspAnalysElement::OnMsgFunc(void *msgPtr, void *pThisPtr) {
- if (lst.size() > 0) {
- Json::Value json;
- for (auto item : lst) {
- string t_camIdex = item.str_cam_dev_id.toStdString();
-// std::string rtsp_url;
-// if (item.str_brand.toStdString() == "haikang") {
-// //璋冪敤rtsp娴乺tsp://admin:a1234567@192.168.1.215:554/h264/ch1/main/av_stream
-// int port = 554;//TODO
-// string username =;
-// DBG("username:" << username << "!!!");
-// string passwd = item.str_password.toStdString();
-// string ip = item.str_ip.toStdString();
-//
-// rtsp_url = "rtsp://" + username + ":" + passwd + "@" + ip + ":" + std::to_string(port) +
-// "/h264/ch1/main/av_stream";
-//
-// }//else if(){}
-
- std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554, item.str_username.toStdString(),
- item.str_password.toStdString(), item.str_brand.toStdString());
-
-
-// lRec.str_det_thr = rec.value("det_thr").toString();
-// lRec.str_cmp_thr = rec.value("cmp_thr").toString();
- auto en_sdk = m_lDBTool->searchCamSdkTableByCamId(item.str_cam_dev_id);
- vector<int> sdkVec = chnString2Vec(en_sdk.str_sdks.toStdString());
-
- //璁剧疆妫�娴嬪緱鍒�
-// std::string s_d = str_det_thr;
-// std::string s_c = str_cmp_thr;
- vector<int> sdkDetCoVec = chnString2Vec(en_sdk.str_det_thr.toStdString());
- vector<int> sdkComCoVec = chnString2Vec(en_sdk.str_cmp_thr.toStdString());
-
- int camId=item.str_cam_dev_id.toInt();
- int t_size = sdkVec.size();
- for (int i = 0; i < t_size; i++) {
- switch (sdkVec[i]) {
- case 1: {
-// INFO("face.enable");
- json["face.enable"] = "1";
- SETSCORE(sdkDetCoVec, i, t_camIdex + "face.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "face.cmp");
- DBG(" TESTCODE " << sdkDetCoVec[i] << " " << sdkComCoVec[i]);
-
- break;
+ evnsq::Message *msg = (evnsq::Message *) msgPtr;
+ DBG("Received a message, id=" << msg->id << " message=[" << msg->body.ToString() << "]");
+ RtspAnalysElement *pthis = (RtspAnalysElement *) pThisPtr;
+ Json::Reader reader;
+ Json::Value value;
+ if (reader.parse(msg->body.ToString(), value)) {
+ auto itor_json_end = value.end();
+ Json::Value cfg_val;
+ for (auto itor_json = value.begin(); itor_json != itor_json_end; ++itor_json) {
+ //#todo 浣跨敤绛栫暐妯″紡?鍙戦�佺澶氫釜鎸囦护绱Н鍚庡彂閫�?
+ if (itor_json->isString() && reader.parse(itor_json->asString(), cfg_val)) {
+ if (itor_json.name() == "cam_edit" || itor_json.name() == "editSdkRule") {
+ string cam_idx = cfg_val["str_cam_dev_id"].asString();
+ if (pthis->controllers.find(cam_idx) != pthis->controllers.end()) {
+ if (pthis->m_GB28181_Enable) {
+ auto lst = pthis->m_lDBTool->searchCamDevTableByType(1);
+ for (auto &item : lst) {
+ if (item.str_cam_dev_id.toStdString() != cam_idx)continue;
+ string rtsp_url = "GB28181";
+ pthis->addCamWithSearchRuleMap(item, rtsp_url);
+ break;
+ }
+ } else {
+ auto lst = pthis->m_lDBTool->searchCamDevTableByType(0);
+ for (auto &item : lst) {
+ if (item.str_cam_dev_id.toStdString() != cam_idx)continue;
+ std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554,
+ item.str_username.toStdString(),
+ item.str_password.toStdString(),
+ item.str_brand.toStdString());
+ pthis->addCamWithSearchRuleMap(item, rtsp_url);
+ break;
+ }
+ }
}
- case 2: {
- DBG("XX.enable " << "1");
- SETSCORE(sdkDetCoVec, i, t_camIdex + "XX.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "XX.cmp");
- break;
- }
- case 3: {
-// INFO("face.enable");
- json["yolo.enable"] = "1";
- SETSCORE(sdkDetCoVec, i, t_camIdex + "yolo.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "yolo.cmp");
- break;
- }
- case 4: {
-
- json["perimeter.enable"] = "1";
- SETSCORE(sdkDetCoVec, i, t_camIdex + "perimeter.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "perimeter.cmp");
- setDataByType(4,json,camId);
- break;
- }
- case 5: {
-
- json["crowd.enable"] = "1";
- SETSCORE(sdkDetCoVec, i, t_camIdex + "crowd.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "crowd.cmp");
- setDataByType(5,json,camId);
- break;
- }
- case 6: {
-
- json["keepRight.enable"] = "1";
- SETSCORE(sdkDetCoVec, i, t_camIdex + "keepRight.det");
- SETSCORE(sdkComCoVec, i, t_camIdex + "keepRight.cmp");
- setDataByType(6,json,camId);
- break;
- }
+ } else if (itor_json.name() == "cam_del") {
+ string cam_idx = cfg_val["str_cam_dev_id"].asString();
+ pthis->removeCamera(cam_idx);
}
+ } else {
+ DBG("itor value is not string ");
+ continue;
}
- json["rtsp"] = rtsp_url;
- json["addr"] = item.str_addr.toStdString();
-
- INFO("cam add is " << item.str_addr.toStdString());
-
-// DBG(t_camIdex);
-
- appPref.getLongData("gpu.index");
- addCamera(t_camIdex, json.toStyledString());
}
} else {
- ERR("searchCamDevTableAll size is 0");
+ ERR("Error Message!!");
}
-
}
+
+void RtspAnalysElement::init() {
+
+ m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
+
+ Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();
+
+ //#todo GB28181
+ if (m_GB28181_Enable) {
+ auto lst = m_lDBTool->searchCamDevTableByType(1);
+
+ appPref.setIntData("n_cut_max_duration", lst_dev.n_cut_max_duration);
+ appPref.setIntData("n_cut_min_duration", lst_dev.n_cut_min_duration);
+ if (lst.size() > 0) {
+ int startCamNO = appPref.getIntData("CamStartNO");
+ int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
+ auto itor = lst.begin();
+
+ if (startCamNO >= lst.size()) {
+ ERR("startCamNO > lst.size()");
+ return;
+ }
+ for (int i = 0; i < startCamNO; i++) {
+ itor++;
+ }
+
+ for (int i = 0; i < CamCount; i++) {
+ if (itor == lst.end()) {
+ ERR("itor == lst.end()");
+ return;
+ }
+ auto &item = *itor;
+
+ string rtsp_url = "GB28181";
+ addCamWithSearchRuleMap(item, rtsp_url);
+
+ itor++;
+ }
+
+// for (auto &item : lst) {
+// std::string t_camIdex = item.str_cam_dev_id.toStdString();
+// std::string rtsp_url = "GB28181";
+//
+// SdkRuleMap ruleMap = m_lDBTool->searchSdkRuleByCamId(item.str_cam_dev_id);
+// for (auto &rule:ruleMap) {
+//
+// rule.second.strAddr = item.str_addr;
+// rule.second.weekRuleVec = m_lDBTool->searchCameraWeekRuleByCamId(item.str_cam_dev_id, rule.first);
+//
+// if (rule.second.nSdkType == PerStaticSdk) {
+// float temp = 1 - (float) (rule.second.nThreshold) / 100;
+// rule.second.fSensitivity = (75 + 25 * temp) / 100;
+//
+// } else if (rule.second.nSdkType != FaceSdk) {
+// float temp = 1 - (float) (rule.second.nThreshold) / 100;
+// rule.second.fSensitivity = (5 + 90 * temp) / 100;
+// }
+// DBG("fSensitivity" << rule.second.fSensitivity);
+// }
+// appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
+// INFO("cam add is " << item.str_addr.toStdString());
+// addCamera(t_camIdex, ruleMap);
+// }
+ } else {
+ ERR("searchCamDevTableAll size is 0");
+ }
+
+ } else {
+
+ auto lst = m_lDBTool->searchCamDevTableByType(0);
+
+ appPref.setIntData("n_cut_max_duration", lst_dev.n_cut_max_duration);
+ appPref.setIntData("n_cut_min_duration", lst_dev.n_cut_min_duration);
+ if (lst.size() > 0) {
+ int startCamNO = appPref.getIntData("CamStartNO");
+ int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
+ auto itor = lst.begin();
+
+ if (startCamNO >= lst.size()) {
+ ERR("startCamNO > lst.size()");
+ return;
+ }
+
+ for (int i = 0; i < startCamNO; i++) {
+ itor++;
+ }
+
+ for (int i = 0; i < CamCount; i++) {
+ if (itor == lst.end()) {
+ ERR("itor == lst.end()");
+ return;
+ }
+ auto &item = *itor;
+ std::string rtsp_url = rtspAddrBuild(item.str_ip.toStdString(), 554, item.str_username.toStdString(),
+ item.str_password.toStdString(), item.str_brand.toStdString());
+
+ addCamWithSearchRuleMap(item, rtsp_url);
+
+ itor++;
+ }
+ } else {
+ ERR("searchCamDevTableAll size is 0");
+ }
+
+ }
+}
+
+void RtspAnalysElement::addCamWithSearchRuleMap(const Record_Cam_Dev &item, string rtsp_url) {
+ string t_camIdex = item.str_cam_dev_id.toStdString();
+
+ SdkRuleMap ruleMap = m_lDBTool->searchSdkRuleByCamId(item.str_cam_dev_id);
+ for (auto &rule:ruleMap) {
+
+ rule.second.strAddr = item.str_addr;
+ rule.second.weekRuleVec = m_lDBTool->searchCameraWeekRuleByCamId(item.str_cam_dev_id, rule.first);
+
+ if (rule.second.nSdkType == PerStaticSdk) {
+ float temp = 1 - (float) (rule.second.nThreshold) / 100;
+ rule.second.fSensitivity = (75 + 25 * temp) / 100;
+
+ } else if (rule.second.nSdkType != FaceSdk) {
+ float temp = 1 - (float) (rule.second.nThreshold) / 100;
+ rule.second.fSensitivity = (5 + 90 * temp) / 100;
+ }
+ DBG("fSensitivity" << rule.second.fSensitivity);
+ }
+ appPref.setStringData(t_camIdex + "rtsp", rtsp_url);
+
+ INFO("cam add is " << item.str_addr.toStdString());
+ addCamera(t_camIdex, ruleMap);
+}
+
RtspAnalysElement::~RtspAnalysElement() {
for (auto controller: controllers) {
@@ -181,30 +278,23 @@
controllers.clear();
}
-int RtspAnalysElement::addCamera(const std::string &index, const std::string &json) {
+int RtspAnalysElement::addCamera(const std::string &index, const std::map<int, SdkRule> &sdkRuleMap) {
if (controllers.find(index) == controllers.end()) {
if (currentCount >= maxCount) {
ERR("addCamera faild, camera's num is full!")
return -1;
}
- Json::Reader reader;
- Json::Value value;
- if (reader.parse(json, value)) {
+ controllers[index] = new AppPipeController(index, sdkRuleMap, true);
+ controllers[index]->setfdfsClient(&fdfsClient);
+ controllers[index]->start();
+ currentCount++;
+ return 0;
- controllers[index] = new AppPipeController(index, std::move(value), true);
-// controllers[index] = new AppPipeController(index, std::move(value), false);
-
- controllers[index]->setfdfsClient(&fdfsClient);
- controllers[index]->start();
- currentCount++;
- return 0;
- }
- return -1;
} else {
removeCamera(index);
DBG("removeCamera " << index)
- return addCamera(index, json);
+ return addCamera(index, sdkRuleMap);
}
}
@@ -266,39 +356,82 @@
}
return result;
}
-void RtspAnalysElement::setDataByType(int type,Json::Value& json,int camId)
-{
- switch (type) {
- case 4:
- {
- auto rule=m_lDBTool->searchPerimeterRuleByCamId(camId);
- json["perimeter.area"]=rule.strAreas.toStdString();
- json["perimeter.num"]=rule.nAlarmPeopleNum;
- json["perimeter.delay"]=rule.nTriggerDelay;
- json["perimeter.tolerance"]=rule.nTriggertolerance;
- break;
- }
- case 5:
- {
- auto rule=m_lDBTool->searchCrowdRuleByCamId(camId);
- json["crowd.area"]=rule.strAreas.toStdString();
- json["crowd.num"]=rule.nAlarmPeopleNum;
- json["crowd.delay"]=rule.nTriggerDelay;
- json["crowd.tolerance"]=rule.nTriggertolerance;
- break;
- }
- case 6:
- {
- auto rule=m_lDBTool->searchActRuleByCamId(camId);
- json["keepRight.leftArea"]=rule.strAreas.toStdString();
- json["keepRight.leftLine"]=rule.strLine.toStdString();
- json["keepRight.rightArea"]=rule.strExAreas.toStdString();
- json["keepRight.rightLine"]=rule.strExLine.toStdString();
- json["keepRight.delay"]=rule.nTriggerDelay;
- json["keepRight.tolerance"]=rule.nTriggertolerance;
- break;
- }
- default:
- break;
- }
-}
+
+//void RtspAnalysElement::setDataByType(Json::Value& json,const SdkRule& rule)
+//{
+// std::string camId=rule.strCamId.toStdString();
+// sdkRuleMap[rule.nSdkType].weekRuleVec=m_lDBTool->searchCameraWeekRuleByCamId(rule.strCamId,rule.nSdkType);
+// switch (rule.nSdkType)
+// {
+// case FaceSdk:
+// {
+// json["face.enable"] = "1";
+
+// appPref.setFloatData(camId+"face.det",rule.nQuality);
+// appPref.setFloatData(camId+"face.cmp",rule.nThreshold);
+
+
+
+// break;
+// }
+// case CarSdk:
+// {
+// // DBG("XX.enable " << "1");
+
+// appPref.setFloatData(camId+"xx.det",rule.nQuality);
+// appPref.setFloatData(camId+"xx.cmp",rule.nThreshold);
+
+// break;
+// }
+// case YoloSdk:
+// {
+// json["yolo.enable"] = "1";
+// appPref.setFloatData(camId+"yolo.det",rule.nQuality);
+// appPref.setFloatData(camId+"yolo.cmp",rule.nThreshold);
+
+// break;
+// }
+// case PerimeterSdk:
+// {
+// // auto rule=m_lDBTool->searchPerimeterRuleByCamId(camId);
+// json["perimeter.enable"] = 1;
+// json["perimeter.area"]=rule.strAreas.toStdString();
+// json["perimeter.num"]=rule.nAlarmNumLowerLimit;
+
+// json["perimeter.delay"]=rule.nTriggerDelay;
+// json["perimeter.tolerance"]=rule.nTriggerTolerance;
+// appPref.setFloatData(camId+"perimeter.det",rule.nQuality);
+// appPref.setFloatData(camId+"perimeter.cmp",rule.nThreshold);
+
+
+// break;
+// }
+// case CrowdSdk:
+// {
+// json["crowd.enable"] = 1;
+// json["crowd.area"]=rule.strAreas.toStdString();
+// json["crowd.num"]=rule.nAlarmNumLowerLimit;
+// json["crowd.delay"]=rule.nTriggerDelay;
+// json["crowd.tolerance"]=rule.nTriggerTolerance;
+// appPref.setFloatData(camId+"crowd.det",rule.nQuality);
+// appPref.setFloatData(camId+"crowd.cmp",rule.nThreshold);
+
+// break;
+// }
+// case KeepRightSdk:
+// {
+// json["keepRight.enable"] = 1;
+// json["keepRight.leftArea"]=rule.strAreas.toStdString();
+// json["keepRight.leftLine"]=rule.strLine.toStdString();
+// json["keepRight.rightArea"]=rule.strExAreas.toStdString();
+// json["keepRight.rightLine"]=rule.strExLine.toStdString();
+// json["keepRight.delay"]=rule.nTriggerDelay;
+// json["keepRight.tolerance"]=rule.nTriggerTolerance;
+// appPref.setFloatData(camId+"keepRight.det",rule.nQuality);
+// appPref.setFloatData(camId+"keepRight.cmp",rule.nThreshold);
+// break;
+// }
+// default:
+// break;
+// }
+//}
--
Gitblit v1.8.0