GB28181集成完成,集成推流模块

pansen

2019-03-07 d3b7bbe7102cd089680a828f5d8f6402c8cf6342

 QiaoJiaSystem/StructureApp/HiredisTool.cpp

@@ -4,42 +4,38 @@
#include <jsoncpp/json/json.h>
#include <basic/debug/Debug.h>
#include <basic/util/app/AppConfig.h>
HiredisTool::HiredisTool():
m_redis(nullptr),
m_nImgListLen(0)
{

HiredisTool::HiredisTool() :
    m_redis(nullptr),
    m_nImgListLen(0) {
    init();
}
HiredisTool::~HiredisTool()
{
    if(m_redis != nullptr)
    {

HiredisTool::~HiredisTool() {
    if (m_redis != nullptr) {
        redisFree(m_redis);//析构函数释放资源
        DBG("~RedisTool :: free redis connection ") ;
        DBG("~RedisTool :: free redis connection ");
    }

}

void HiredisTool::init()
{
       struct timeval timeout = { 1, 500000 }; // 1.5 seconds 设置连接等待时间
void HiredisTool::init() {
    struct timeval timeout = {1, 500000}; // 1.5 seconds 设置连接等待时间

     //  m_nImgListLen=appConfig.getIntProperty("redis_buf_len");
       std::string ip=appConfig.getStringProperty("redis_ip");
    //  m_nImgListLen=appConfig.getIntProperty("redis_buf_len");
    std::string ip = appConfig.getStringProperty("redis_ip");

        int port=appConfig.getIntProperty("redis_port");
        m_redis = redisConnectWithTimeout(ip.c_str(), port, timeout);//建立连接
        if (m_redis->err) {
    int port = appConfig.getIntProperty("redis_port");
    m_redis = redisConnectWithTimeout(ip.c_str(), port, timeout);//建立连接
    if (m_redis->err) {

            DBG("RedisTool : Connection error: %s"<< m_redis->errstr);
            redisFree(m_redis);
            m_redis= nullptr;
        }
        else
        {
                DBG( "init redis tool success " );
        DBG("RedisTool : Connection error: %s" << m_redis->errstr);
        redisFree(m_redis);
        m_redis = nullptr;
    } else {
        DBG("init redis tool success ");

        }
    }
}


@@ -81,42 +77,41 @@



bool HiredisTool::pushImageBuf(const std::string& file_name,const cv::Mat& img)
{


       std::vector<uchar> buf;
      Json::Value json;

      CvUtil::cvMat2Buffer(img,buf);
//      json["time"]=info.time;
      json["img"]=std::string(buf.begin(),buf.end());
  //    buf.clear();
      std::string strBuf=std::string(buf.begin(),buf.end());
      if(!listLpush(file_name,json.toStyledString()));
      {

          return false;
      }

      //DBG(json.toStyledString());

      return true;
}

bool HiredisTool::setKeyImage(const std::string &file_name, const cv::Mat &img)
{
bool HiredisTool::pushImageBuf(const std::string &file_name, const cv::Mat &img) {


    std::vector<uchar> buf;
    Json::Value json;

    CvUtil::cvMat2Buffer(img,buf);
    CvUtil::cvMat2Buffer(img, buf);
//      json["time"]=info.time;
    json["img"]=std::string(buf.begin(),buf.end());
    json["img"] = std::string(buf.begin(), buf.end());
    //    buf.clear();
    std::string strBuf = std::string(buf.begin(), buf.end());
    if (!listLpush(file_name, json.toStyledString()));
    {

        return false;
    }

    //DBG(json.toStyledString());

    return true;
}

bool HiredisTool::setKeyImage(const std::string &file_name, const cv::Mat &img) {


    std::vector<uchar> buf;
    Json::Value json;

    CvUtil::cvMat2Buffer(img, buf);
//      json["time"]=info.time;
    json["img"] = std::move(m_base64.Encode(buf.data(), buf.size()));
//    json["img"] = std::string(buf.begin(), buf.end());
    //    buf.clear();
//    std::string strBuf=std::string(buf.begin(),buf.end());
    if(!setKeyValue(file_name,json.toStyledString()));
    if (!setKeyValue(file_name, json.toStyledString()));
    {

        return false;
@@ -128,62 +123,59 @@
}


void  HiredisTool::getImage(const std::string& file_name, cv::Mat& img)
{
void HiredisTool::getImage(const std::string &file_name, cv::Mat &img) {


        std::string content;
        cv::Mat imgTep;

        listRpop(file_name,content);


        Json::Reader reader;
        Json::Value value;

        if(!reader.parse(content,value))
        {
           return ;
        }

        std::string str=value["img"].asString();
        std::vector<uchar> data;

        data.resize(str.size());

        data.assign(str.begin(),str.end());

        CvUtil::buffer2CvMat(data,imgTep);

        imgTep.copyTo(img);
    // return img;
}


void HiredisTool::getKeyImage(const std::string &imageName, cv::Mat &img)
{

    std::string content;
    cv::Mat imgTep;

    getKeyValue(imageName,content);
    listRpop(file_name, content);


    Json::Reader reader;
    Json::Value value;

    if(!reader.parse(content,value))
    {
        return ;
    if (!reader.parse(content, value)) {
        return;
    }

    std::string str=value["img"].asString();
    std::string str = value["img"].asString();
    std::vector<uchar> data;

    data.resize(str.size());

    data.assign(str.begin(),str.end());
    data.assign(str.begin(), str.end());

    CvUtil::buffer2CvMat(data,imgTep);
    CvUtil::buffer2CvMat(data, imgTep);

    imgTep.copyTo(img);
    // return img;
}


void HiredisTool::getKeyImage(const std::string &imageName, cv::Mat &img) {

    std::string content;
    cv::Mat imgTep;

    getKeyValue(imageName, content);

    Json::Reader reader;
    Json::Value value;

    if (!reader.parse(content, value)) {
        return;
    }

    std::string str = value["img"].asString();
    str = std::move(m_base64.Decode(str.c_str(), str.size()));
    std::vector<uchar> data;

    data.resize(str.size());

    data.assign(str.begin(), str.end());

    CvUtil::buffer2CvMat(data, imgTep);

    imgTep.copyTo(img);
    // return img;
@@ -195,25 +187,22 @@
//    DBG("type="<<type);
//    return hashSet(file_list,cam_id,type);
//}
bool HiredisTool::clearAllImageBuf()
{
bool HiredisTool::clearAllImageBuf() {
//    auto cam_map=findAllCamera();
//    for(auto cam:cam_map)
//    {
//        delKey(cam.first);
//    }
}
bool HiredisTool::listLpush(const std::string& key,const std::string& value)
{
    if(!checkParam())
    {

bool HiredisTool::listLpush(const std::string &key, const std::string &value) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"lpush %s %s", key.c_str(),value.c_str());
    if(!checkResult(reply))
    {
    reply = (redisReply *) redisCommand(m_redis, "lpush %s %s", key.c_str(), value.c_str());
    if (!checkResult(reply)) {
        return false;
    }

@@ -222,17 +211,14 @@
    return true;
}

bool HiredisTool::setKeyValue(const std::string &key, const std::string &value)
{
    if(!checkParam())
    {
bool HiredisTool::setKeyValue(const std::string &key, const std::string &value) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"set %s %s", key.c_str(),value.c_str());
    if(!checkResult(reply))
    {
    reply = (redisReply *) redisCommand(m_redis, "set %s %s", key.c_str(), value.c_str());
    if (!checkResult(reply)) {
        return false;
    }

@@ -241,64 +227,54 @@
    return true;
}

bool HiredisTool::getKeyValue(const std::string &key, std::string &value)
{
    if(!checkParam())
    {
bool HiredisTool::getKeyValue(const std::string &key, std::string &value) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;
    //1.find all filename
    reply = (redisReply*)redisCommand(m_redis,"get %s  ", key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "get %s  ", key.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    if(reply->type == REDIS_REPLY_STRING)
    {
        value=reply->str;
    if (reply->type == REDIS_REPLY_STRING) {
        value = reply->str;
    }
    freeReplyObject(reply);

    return true;
}

bool HiredisTool::listRpop(const std::string& key,std::string& value)
{
    if(!checkParam())
    {
bool HiredisTool::listRpop(const std::string &key, std::string &value) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;
    //1.find all filename
    reply = (redisReply*)redisCommand(m_redis,"rpop %s  ", key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "rpop %s  ", key.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    if(reply->type == REDIS_REPLY_STRING)
    {
        value=reply->str;
    if (reply->type == REDIS_REPLY_STRING) {
        value = reply->str;
    }
    freeReplyObject(reply);

    return true;
}
bool HiredisTool::hashSet(const std::string& tab,const std::string& key,const int& value)
{
    if(!checkParam())
    {

bool HiredisTool::hashSet(const std::string &tab, const std::string &key, const int &value) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;


    reply = (redisReply*)redisCommand(m_redis,"hset %s %s %d", tab.c_str(),key.c_str(),value);
    reply = (redisReply *) redisCommand(m_redis, "hset %s %s %d", tab.c_str(), key.c_str(), value);

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    freeReplyObject(reply);
@@ -307,176 +283,154 @@
    return true;
}

int  HiredisTool::getSize(const std::string& key)
{
    int size=0;
    if(!checkParam())
    {
int HiredisTool::getSize(const std::string &key) {
    int size = 0;
    if (!checkParam()) {
        return size;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"llen %s", key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "llen %s", key.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return size;
    }
    size=reply->integer;
    size = reply->integer;
    freeReplyObject(reply);
    return size;
}
bool HiredisTool::listLindex(const std::string& key,std::string& value)
{
    int size=0;
    if(!checkParam())
    {

bool HiredisTool::listLindex(const std::string &key, std::string &value) {
    int size = 0;
    if (!checkParam()) {
        return size;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"lindex %s -1", key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "lindex %s -1", key.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    if(reply->type == REDIS_REPLY_STRING)
    {
        value=reply->str;
    if (reply->type == REDIS_REPLY_STRING) {
        value = reply->str;
    }
    freeReplyObject(reply);
}
bool HiredisTool::delKey(const std::string& key)
{
    if(!checkParam())
    {

bool HiredisTool::delKey(const std::string &key) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"del %s", key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "del %s", key.c_str());


    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    freeReplyObject(reply);
    return true;
}
bool HiredisTool::checkParam()
{
     if(m_redis == nullptr || m_redis->err)
     {
         if(m_redis)
         {
             redisFree(m_redis);
             m_redis= nullptr;
         }
         DBG("Redis init Error !!!");
         init();
         return false;
     }
     return true;

bool HiredisTool::checkParam() {
    if (m_redis == nullptr || m_redis->err) {
        if (m_redis) {
            redisFree(m_redis);
            m_redis = nullptr;
        }
        DBG("Redis init Error !!!");
        init();
        return false;
    }
    return true;
}
bool HiredisTool::checkResult(redisReply *reply)
{
    if (reply==nullptr)
    {
         DBG("reply nullptr !!!");
         return false;

bool HiredisTool::checkResult(redisReply *reply) {
    if (reply == nullptr) {
        DBG("reply nullptr !!!");
        return false;
    }

    if (reply->type == REDIS_REPLY_ERROR)
    {
        ERR("redisCommand Error: "<<reply->str);
    if (reply->type == REDIS_REPLY_ERROR) {
        ERR("redisCommand Error: " << reply->str);
        freeReplyObject(reply);
        return false;
    }
    return true;
}
void HiredisTool::addFileInfo(const std::string& filename,const int& file_status)
{

      hashSet(file_list,filename,file_status);
 }
std::queue<cv::Mat> HiredisTool::getImgQue(const std::string& filename)
{
void HiredisTool::addFileInfo(const std::string &filename, const int &file_status) {

    hashSet(file_list, filename, file_status);
}

std::queue<cv::Mat> HiredisTool::getImgQue(const std::string &filename) {
    std::queue<cv::Mat> que;
    if(!checkParam())
    {
    if (!checkParam()) {
        return que;
    }
    redisReply *reply;

    reply = (redisReply*)redisCommand(m_redis,"lrange %s 0 -1", filename.c_str());
    reply = (redisReply *) redisCommand(m_redis, "lrange %s 0 -1", filename.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return que;
    }
    if(reply->type == REDIS_REPLY_ARRAY)
    {
        for(int i=0;i<reply->elements;++i)
        {
            std::string str=reply->element[i]->str;
    if (reply->type == REDIS_REPLY_ARRAY) {
        for (int i = 0; i < reply->elements; ++i) {
            std::string str = reply->element[i]->str;
            std::vector<uchar> data;
            cv::Mat img;
            data.resize(str.size());
            data.assign(str.begin(),str.end());
            CvUtil::buffer2CvMat(data,img);
            data.assign(str.begin(), str.end());
            CvUtil::buffer2CvMat(data, img);
            que.push(img);
        }
    }
    freeReplyObject(reply);
}
std::map<std::string,int> HiredisTool::findAllFileStatus()
{
     std::map<std::string,int> fileMap;
     if(!checkParam())
     {
         return fileMap;
     }

     redisReply *reply;
std::map<std::string, int> HiredisTool::findAllFileStatus() {
    std::map<std::string, int> fileMap;
    if (!checkParam()) {
        return fileMap;
    }

     reply = (redisReply*)redisCommand(m_redis,"hgetall %s", file_list);
    redisReply *reply;

     if(!checkResult(reply))
     {
         return fileMap;
     }
    reply = (redisReply *) redisCommand(m_redis, "hgetall %s", file_list);

     if(reply->type == REDIS_REPLY_ARRAY)
     {
    if (!checkResult(reply)) {
        return fileMap;
    }

         for (int i= 0; i < reply->elements; i+=2)
         {
             std::string key=reply->element[i]->str;
             std::string value=reply->element[i+1]->str;
    if (reply->type == REDIS_REPLY_ARRAY) {

             if(key.empty()||value.empty()) continue;
        for (int i = 0; i < reply->elements; i += 2) {
            std::string key = reply->element[i]->str;
            std::string value = reply->element[i + 1]->str;

             fileMap.insert(std::pair<std::string,int>(key,atoi(value.c_str())));
         }
     }
     freeReplyObject(reply);
     return fileMap;
            if (key.empty() || value.empty()) continue;

            fileMap.insert(std::pair<std::string, int>(key, atoi(value.c_str())));
        }
    }
    freeReplyObject(reply);
    return fileMap;
}
bool HiredisTool::hashDel(const std::string& tab,const std::string &key)
{
    if(!checkParam())
    {

bool HiredisTool::hashDel(const std::string &tab, const std::string &key) {
    if (!checkParam()) {
        return false;
    }
    redisReply *reply;


    reply = (redisReply*)redisCommand(m_redis,"hdel %s %s", tab.c_str(),key.c_str());
    reply = (redisReply *) redisCommand(m_redis, "hdel %s %s", tab.c_str(), key.c_str());

    if(!checkResult(reply))
    {
    if (!checkResult(reply)) {
        return false;
    }
    freeReplyObject(reply);