From ecb47d89dcb2b1d8e4be9b6cb0a84d8b49e2e3cc Mon Sep 17 00:00:00 2001
From: pansen <pansen626@sina.com>
Date: 星期三, 10 四月 2019 12:20:00 +0800
Subject: [PATCH] 添加iceServer

---
 QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp |  241 +++++++++++++++++++++++++++++-------------------
 1 files changed, 145 insertions(+), 96 deletions(-)

diff --git a/QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp b/QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp
index 01b854a..ce34ed9 100644
--- a/QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp
+++ b/QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp
@@ -7,90 +7,7 @@
 
 
 using std::string;
-
-RtspAnalysManager::RtspAnalysManager(LDBTool *_dbTool) : m_lDBTool(nullptr), m_maxCount(50), m_currentCount(0) {
-    DBG("MYH DEBUG HERE");
-    m_lDBTool = new LDBTool;
-    init();
-
-    //nsq   set callback func
-    DBG("NsqMsgConsumer Init");
-    nsqMsgConsumer = new BasicMsg::Nsq::NsqMsgConsumer("127.0.0.1", "4150", "cut_dura_edit", to_string(appPref.getIntData("RpcVTIMPort")));
-    nsqMsgConsumer->setMessageCallback(std::bind(&OnMsgFunc, std::placeholders::_1, this));
-    nsqMsgConsumer->init();
-    DBG("NsqMsgConsumer Init END!!");
-
-}
-
-void RtspAnalysManager::OnMsgFunc(void *msgPtr, void *pThisPtr) {
-    evnsq::Message *msg = (evnsq::Message *) msgPtr;
-    DBG("Received a message, id=" << msg->id << " message=[" << msg->body.ToString() << "]");
-    RtspAnalysManager *pthis = (RtspAnalysManager *) pThisPtr;
-    Json::Reader reader;
-    Json::Value value;
-    if (reader.parse(msg->body.ToString(), value)) {
-        int cut_max_duration = value["cut_max_duration"].asInt();
-        int cut_min_duration = value["cut_min_duration"].asInt();
-        DBG("cut_max_duration: " << cut_max_duration << "   cut_min_duration: " << cut_min_duration);
-        appPref.setIntData("n_cut_max_duration", cut_max_duration);
-        appPref.setIntData("n_cut_min_duration", cut_min_duration);
-
-        DBG("cut_max_duration: " << cut_max_duration << "   cut_min_duration: " << cut_min_duration);
-//        m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
-        //#todo GB28181
-        if (pthis->m_GB28181_Enable) {
-            DBG("searchCamDevTableByType before");
-            auto lst = pthis->m_lDBTool->searchCamDevTableByType(1);
-            DBG("searchCamDevTableByType size:" << lst.size());
-
-            if (lst.size() > 0) {
-                for (auto item : lst) {
-                    std::string t_camIdex = item.str_cam_dev_id.toStdString();
-                    DBG("-----camID:" << t_camIdex << "min:" << cut_min_duration << "max:" << cut_max_duration);
-                    pthis->m_controllers_videoCapElem[t_camIdex]->SetVideoMinMaxSeconds(cut_min_duration, cut_max_duration);
-                }
-            } else {
-                ERR("searchCamDevTableByType size is 0");
-            }
-
-        } else {
-            auto lst = pthis->m_lDBTool->searchCamDevTableByType(0);
-
-            if (lst.size() > 0) {
-                int startCamNO = appPref.getIntData("CamStartNO");
-                int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
-                INFO("StartCamNO: " << startCamNO << " CamCount: " << CamCount);
-                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;
-                    }
-                    std::string t_camIdex = itor->str_cam_dev_id.toStdString();
-                    DBG("-----camID:" << t_camIdex << "min:" << cut_min_duration << "max:" << cut_max_duration);
-                    pthis->m_controllers[t_camIdex]->SetVideoMinMaxSeconds(cut_min_duration, cut_max_duration);
-                    itor++;
-                }
-
-            } else {
-                ERR("searchCamDevTableByType size is 0");
-            }
-
-        }
-    } else {
-        ERR("Error Message!!");
-    }
-}
+void *gRtspAnalysManagerCamera = nullptr;
 
 static std::string rtspAddrBuild(std::string ip, const int port, std::string username,
                                  std::string passwd, std::string brand) {
@@ -113,9 +30,121 @@
     return rtsp_url;
 }
 
+RtspAnalysManager::RtspAnalysManager(LDBTool *_dbTool) : m_lDBTool(nullptr), m_maxCount(50), m_currentCount(0) {
+//    INFO("MYH DEBUG HERE");
+    m_lDBTool = _dbTool;
+    init();
+
+    IMPORT_APP_ARGS;
+    //nsq   set callback func
+    DBG("NsqMsgConsumer Init");
+    string pName(argv[0]);
+    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!!");
+
+}
+
+void RtspAnalysManager::OnMsgFunc(void *msgPtr, void *pThisPtr) {
+    evnsq::Message *msg = (evnsq::Message *) msgPtr;
+    DBG("Received a message, id=" << msg->id << " message=[" << msg->body.ToString() << "]");
+    RtspAnalysManager *pthis = (RtspAnalysManager *) 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() == "cut_dura_edit") {
+                    int cut_max_duration = cfg_val["cut_max_duration"].asInt();
+                    int cut_min_duration = cfg_val["cut_min_duration"].asInt();
+                    DBG("cut_max_duration: " << cut_max_duration << "   cut_min_duration: " << cut_min_duration);
+                    appPref.setIntData("n_cut_max_duration", cut_max_duration);
+                    appPref.setIntData("n_cut_min_duration", cut_min_duration);
+
+                    DBG("cut_max_duration: " << cut_max_duration << "   cut_min_duration: " << cut_min_duration);
+//        m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
+                    //#todo GB28181
+                    if (pthis->m_GB28181_Enable) {
+                        for (auto &item : pthis->m_controllers_videoCapElem) {
+                            item.second->SetVideoMinMaxSeconds(cut_min_duration, cut_max_duration);
+                        }
+                    } else {
+                        for (auto &item : pthis->m_controllers) {
+                            item.second->SetVideoMinMaxSeconds(cut_min_duration, cut_max_duration);
+                        }
+                    }
+                } else if (itor_json.name() == "cam_edit") {
+                    string cam_idx = cfg_val["str_cam_dev_id"].asString();
+                    if (pthis->m_GB28181_Enable) {
+                        if (pthis->m_controllers_videoCapElem.find(cam_idx) !=
+                            pthis->m_controllers_videoCapElem.end()) {
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->addCamera(cam_idx, "GB28181");
+                        }
+                    } else {
+                        if (pthis->m_controllers.find(cam_idx) != pthis->m_controllers.end()) {
+                            std::string rtsp_url = rtspAddrBuild(cfg_val["str_ip"].asString(), 554,
+                                                                 cfg_val["str_username"].asString(),
+                                                                 cfg_val["str_password"].asString(),
+                                                                 cfg_val["str_brand"].asString());
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->addCamera(cam_idx, rtsp_url);
+                        }
+                    }
+                } else if (itor_json.name() == "cam_del") {
+                    string cam_idx = cfg_val["str_cam_dev_id"].asString();
+                    pthis->removeCamera(cam_idx);
+                } else if (itor_json.name() == "cam_startPublish") {
+                    string cam_idx = cfg_val["str_cam_dev_id"].asString();
+                    if (pthis->m_GB28181_Enable) {
+                        if (pthis->m_controllers_videoCapElem.find(cam_idx) !=
+                            pthis->m_controllers_videoCapElem.end()) {
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->m_controllers_videoCapElem[cam_idx]->startPublishVideo();
+                        }
+                    } else {
+                        if (pthis->m_controllers.find(cam_idx) != pthis->m_controllers.end()) {
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->m_controllers[cam_idx]->startPublishVideo();
+                        }
+                    }
+                } else if (itor_json.name() == "cam_stopPublish") {
+                    string cam_idx = cfg_val["str_cam_dev_id"].asString();
+                    if (pthis->m_GB28181_Enable) {
+                        if (pthis->m_controllers_videoCapElem.find(cam_idx) !=
+                            pthis->m_controllers_videoCapElem.end()) {
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->m_controllers_videoCapElem[cam_idx]->stopPublishVideo();
+                        }
+                    } else {
+                        if (pthis->m_controllers.find(cam_idx) != pthis->m_controllers.end()) {
+                            INFO("cam add is " << cfg_val["str_addr"].asString());
+                            pthis->m_controllers[cam_idx]->stopPublishVideo();
+                        }
+                    }
+                }
+            } else {
+                DBG("itor value is not string ");
+                continue;
+            }
+        }
+    } else {
+        ERR("Error Message!!");
+    }
+}
+
+
 //鍒濆鍖栧嚱鏁�
 void RtspAnalysManager::init() {
-    INFO("MYH DEBUG HERE")
+//    INFO("MYH DEBUG HERE")
+
+    gRtspAnalysManagerCamera = this;
+    DBG("gRtspAnalysManagerCamera ADDR:" << gRtspAnalysManagerCamera);
 
     m_GB28181_Enable = appPref.getIntData("GB28181_Enable");
     //#todo GB28181
@@ -127,14 +156,33 @@
         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) {
-            for (auto item : lst) {
-                std::string t_camIdex = item.str_cam_dev_id.toStdString();
+            int startCamNO = appPref.getIntData("CamStartNO");
+            int CamCount = appPref.getIntData("CamEndNO") - appPref.getIntData("CamStartNO") + 1;
+            INFO("StartCamNO: " << startCamNO << " CamCount: " << CamCount);
+            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;
+                }
+                std::string t_camIdex = itor->str_cam_dev_id.toStdString();
                 std::string rtsp_url = "GB28181";
-//            rtspAddrBuild(item.str_ip.toStdString(), 554, item.str_username.toStdString(),
-//                                                 item.str_password.toStdString(), item.str_brand.toStdString());
-                INFO("cam add is " << item.str_addr.toStdString());
+
+                INFO("cam add is " << itor->str_addr.toStdString());
+
                 addCamera(t_camIdex, rtsp_url);
                 sleep(1);
+                itor++;
             }
         } else {
             ERR("searchCamDevTableByType size is 0");
@@ -229,7 +277,7 @@
 
         //#todo end
         if (m_controllers_videoCapElem.find(index) == m_controllers_videoCapElem.end()) {
-            INFO("MYH DEBUG HERE");
+//            INFO("MYH DEBUG HERE");
             if (m_currentCount >= m_maxCount) {
                 ERR("addCamera faild, camera's num is full!")
                 return -1;
@@ -247,7 +295,8 @@
 //                25, 0, 0, this);
 
             m_controllers_videoCapElem[index] = new PushStreamAppPipeController(const_cast<string &>(index),
-                                                                                25, 0, appPref.getIntData("gpu.index"), this);
+                                                                                25, 0, appPref.getIntData("gpu.index"),
+                                                                                this);
             m_controllers_videoCapElem[index]->SetVideoMinMaxSeconds(lst_dev.n_cut_min_duration,
                                                                      lst_dev.n_cut_max_duration);
             m_controllers_videoCapElem[index]->start();
@@ -331,7 +380,7 @@
  * @return
  */
 int RtspAnalysManager::removeAll() {
-    INFO("MYH DEBUG HERE");
+//    INFO("MYH DEBUG HERE");
 
     if (m_GB28181_Enable) {
         for (auto controller: m_controllers_videoCapElem) {
@@ -352,7 +401,7 @@
         }
         m_controllers.clear();
     }
-    INFO("MYH DEBUG HERE");
+//    INFO("MYH DEBUG HERE");
 
     m_imgRedisCRwLock.wrlock();
     for (auto controller: m_imgRedisControllers) {
@@ -367,7 +416,7 @@
 
 
     m_currentCount = 0;
-    INFO("MYH DEBUG HERE");
+//    INFO("MYH DEBUG HERE");
     return 0;
 }
 
@@ -418,7 +467,7 @@
 //        INFO("Save Succeed Cam: " << camId << " ImageKey: " << imageName);
         item->second->SaveImage(imageName, img);
     } else {
-        ERR("Save Failed Cam: " << camId << " ImageKey: " << imageName);
+//        ERR("Save Failed Cam: " << camId << " ImageKey: " << imageName);
     }
 //    m_imgRedisCRwLock.unlock();
 //    INFO("MYH DEBUG HERE");

--
Gitblit v1.8.0