chenshijun
2019-04-16 f946a62d3921e86b44ff8e2973138304b9cd53cd
QiaoJiaSystem/VideoToImageMulth/RtspAnalysManager.cpp
@@ -7,109 +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) {
                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_videoCapElem[t_camIdex]->SetVideoMinMaxSeconds(cut_min_duration, cut_max_duration);
                    itor++;
                }
            } 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) {
@@ -132,9 +30,143 @@
    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");
    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!!");
}
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)) {
                DBG("itor_json.name():" << itor_json.name());
                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 = "";
                    if(pthis->m_GB28181_Enable){
                        cam_idx = cfg_val["equipCode"].asString();
                    }else{
                        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 " << cam_idx);
                            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 = "";
                    if(pthis->m_GB28181_Enable){
                        cam_idx = cfg_val["equipCode"].asString();
                    }else{
                        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["id"].asString();
                    if (pthis->m_GB28181_Enable) {
                        if (pthis->m_controllers_videoCapElem.find(cam_idx) !=
                            pthis->m_controllers_videoCapElem.end()) {
                            INFO("cam add is " << cam_idx);
                            pthis->m_controllers_videoCapElem[cam_idx]->startPublishVideo();
                        }
                    } else {
                        if (pthis->m_controllers.find(cam_idx) != pthis->m_controllers.end()) {
                            INFO("cam add is " << cam_idx);
                            pthis->m_controllers[cam_idx]->startPublishVideo();
                        }
                    }
                } else if (itor_json.name() == "cam_stopPublish") {
                    string cam_idx = cfg_val["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
@@ -267,7 +299,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;
@@ -285,7 +317,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();
@@ -303,7 +336,7 @@
        //    auto lst = m_lDBTool->searchCamDevTableAll();
        Record_Config lst_dev = m_lDBTool->searchConfigTableWithinServerInfo();
        if (m_controllers.find(index) == m_controllers.end()) {
            INFO("MYH DEBUG HERE");
//            INFO("MYH DEBUG HERE");
            if (m_currentCount >= m_maxCount) {
                ERR("addCamera faild, camera's num is full!")
                return -1;
@@ -334,7 +367,7 @@
 * @return 总是0
 */
int RtspAnalysManager::removeCamera(const std::string &index) {
    INFO("MYH DEBUG HERE");
//    INFO("MYH DEBUG HERE");
    if (m_GB28181_Enable) {
        if (m_controllers_videoCapElem.find(index) == m_controllers_videoCapElem.end())return -1;
        auto controller = m_controllers_videoCapElem[index];
@@ -360,7 +393,7 @@
    delete imgRedis;
    m_imgRedisControllers.erase(index);
    m_imgRedisCRwLock.unlock();
    INFO("MYH DEBUG HERE");
//    INFO("MYH DEBUG HERE");
    return 0;
}
@@ -369,7 +402,7 @@
 * @return
 */
int RtspAnalysManager::removeAll() {
    INFO("MYH DEBUG HERE");
//    INFO("MYH DEBUG HERE");
    if (m_GB28181_Enable) {
        for (auto controller: m_controllers_videoCapElem) {
@@ -390,7 +423,7 @@
        }
        m_controllers.clear();
    }
    INFO("MYH DEBUG HERE");
//    INFO("MYH DEBUG HERE");
    m_imgRedisCRwLock.wrlock();
    for (auto controller: m_imgRedisControllers) {
@@ -405,7 +438,7 @@
    m_currentCount = 0;
    INFO("MYH DEBUG HERE");
//    INFO("MYH DEBUG HERE");
    return 0;
}
@@ -421,7 +454,7 @@
//录取视频的RPC的接口函数
::std::string RtspAnalysManager::recordVideo(const ::std::string &name, const ::Ice::Current &) {
    INFO("Record Video For: " << name);
//    INFO("Record Video For: " << name);
    ImageName_s_t nameSt = ImageName_s_t::fromString(name);
    if (nameSt.Valid()) {
        if (m_GB28181_Enable) {
@@ -456,7 +489,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");