lp-data-alarm/EngineAlarm/DataEngine.cpp

#include "DataEngine.h"
#include "DeviceManagerProxy.h"
#include <QJsonArray>
#include <QJsonObject>
#include <QJsonDocument>
#include <fstream>
#include <QtCore/QVariant>
#include <QtCore/QDebug>
#include <QtWidgets/QApplication>

extern std::string loadContent(std::string szFilename);

DataEngine::DataEngine()
    :QObject()
{
    m_pRedis = nullptr;
    m_pTDengine = nullptr;
    m_mpDevControler.clear();
}

void DataEngine::OnData(std::string name,QVariant val)
{
//    if( m_pRedis != nullptr )
//    {
//        m_pRedis->Publish(name, val);
//    }
//    else
//    {
//        qCritical() << __FILE__ << __LINE__ << " " << "no publisher.";
//    }
    if( m_pTDengine != nullptr )
    {
 //       m_pTDengine->Publish(name, val);
    }
    else
    {
        qCritical() << __FILE__ << __LINE__ << " " << "no publisher.";
    }
    if( m_pMqtt != nullptr )
    {
        m_pMqtt->Publish(name, val);
    }
    else
    {
        qCritical() << __FILE__ << __LINE__ << " " << "no publisher.";
    }
}

void DataEngine::sltSubData(const std::string topic, const std::string msg)
{
    if(m_mpDevControler.find(topic) == m_mpDevControler.end())
    {
        return;
    }

    m_mpDevControler[topic]->OnSubData(topic,msg);
}

void loadConfig(std::string szFile,Config& config)
{
    std::string cfg = loadContent(szFile);
    QJsonDocument jsonDoc(QJsonDocument::fromJson(cfg.c_str()));
    QJsonObject json = jsonDoc.object();
    QStringList keys = json.keys();
    if( keys.contains("serverid",Qt::CaseInsensitive))
    {
        config.serverId = json["serverid"].toString().toLocal8Bit().toStdString();
    }
    if( keys.contains("module",Qt::CaseInsensitive))
    {
        config.appName = json["module"].toString().toLocal8Bit().toStdString();
    }
}

void DataEngine::Startup()
{
    // merge file name
    QString szWorkDir = qApp->applicationDirPath();
    if( szWorkDir.right(1).contains("/") == false && szWorkDir.right(1).contains("\\") == false)
    {
        szWorkDir += "/";
    }
    std::string szFile = szWorkDir.toLocal8Bit().toStdString() + "config/server.json";

    // load server config
    std::map<std::string,tagSetup> mpConfig;
    std::string cfg = loadContent(szFile);
    QJsonDocument jsonDoc(QJsonDocument::fromJson(cfg.c_str()));
    QJsonArray ja = jsonDoc.array();
    for(auto i : ja)
    {
        QJsonObject json = i.toObject();
        QStringList items = json.keys();
        if( items.contains("ignore",Qt::CaseInsensitive) && json["ignore"].type() == QJsonValue::Bool && json["ignore"].toBool())
        {
            continue;
        }

        tagSetup ts;
        ts.addr = json["addr"].toString().toLocal8Bit().toStdString();
        ts.user = json["user"].toString().toLocal8Bit().toStdString();
        ts.password = json["password"].toString().toLocal8Bit().toStdString();
        ts.db = json["database"].toString().toLocal8Bit().toStdString();
        ts.port = json["port"].toInt();

        std::string szName = json["name"].toString().toLocal8Bit().toStdString();
        mpConfig[szName] = ts;
    }

    m_pRedis = new RedisAgent();
    m_pRedis->Connect(mpConfig["redis"]);           // redis

    m_pTDengine = new TDAgent();
    m_pTDengine->Connect(mpConfig["tdengine"]);     // td

    m_pMqtt = new MQTTAgent();
    m_pMqtt->Connect(mpConfig["mqtt"]);             // mqtt

    // server.id
    Config config;
    loadConfig(szWorkDir.toLocal8Bit().toStdString() + "config/alarm.json",config);

    DeviceManagerProxy dm(m_pRedis);
    std::list<DeviceInfo> lstDevices = dm.loadDevices(config.serverId, config.appName);
    std::list<DeviceInfo>::iterator itr;
    for( itr = lstDevices.begin();itr!= lstDevices.end();++itr)
    {
        DeviceInfo di = *itr;
        DeviceController* pControler = new DeviceController();
        pControler->regConsumer(this);
        pControler->CreateDevice(di);

        std::list<std::string> lstTopics;
        std::vector<Setting>::iterator itrT;
        for( itrT = di.ModuleInfo.vSettings.begin(); itrT != di.ModuleInfo.vSettings.end(); ++itrT )
        {
            std::string szTopic = "topic_" + itrT->Name;
            lstTopics.push_back(szTopic);
            m_mpDevControler[szTopic] = pControler;
        }
        m_pTDengine->Subscribe(lstTopics);

        qRegisterMetaType<std::string>("std::string");
        connect(m_pTDengine,SIGNAL(sigEvent(const std::string, const std::string)),this,SLOT(sltSubData(const std::string, const std::string)));
    }
}