mirror of
https://github.com/jomjol/AI-on-the-edge-device.git
synced 2025-12-06 19:46:54 +03:00
313 lines
9.4 KiB
C++
313 lines
9.4 KiB
C++
#include <sstream>
|
|
#include "ClassFlowMQTT.h"
|
|
#include "Helper.h"
|
|
#include "connect_wlan.h"
|
|
|
|
#include "time_sntp.h"
|
|
#include "interface_mqtt.h"
|
|
#include "ClassFlowPostProcessing.h"
|
|
#include "ClassLogFile.h"
|
|
|
|
#include <time.h>
|
|
|
|
#define __HIDE_PASSWORD
|
|
|
|
static const char *TAG = "class_flow_MQTT";
|
|
|
|
void ClassFlowMQTT::SetInitialParameter(void)
|
|
{
|
|
uri = "";
|
|
topic = "";
|
|
topicError = "";
|
|
topicRate = "";
|
|
topicTimeStamp = "";
|
|
maintopic = "";
|
|
lwt = "";
|
|
|
|
topicUptime = "";
|
|
topicFreeMem = "";
|
|
|
|
clientname = "AIOTED-" + getMac();
|
|
|
|
OldValue = "";
|
|
flowpostprocessing = NULL;
|
|
user = "";
|
|
password = "";
|
|
SetRetainFlag = 0;
|
|
previousElement = NULL;
|
|
ListFlowControll = NULL;
|
|
disabled = false;
|
|
MQTTenable = false;
|
|
keepAlive = 600; // TODO This must be greater than the Flow Interval!
|
|
}
|
|
|
|
ClassFlowMQTT::ClassFlowMQTT()
|
|
{
|
|
SetInitialParameter();
|
|
}
|
|
|
|
ClassFlowMQTT::ClassFlowMQTT(std::vector<ClassFlow*>* lfc)
|
|
{
|
|
SetInitialParameter();
|
|
|
|
ListFlowControll = lfc;
|
|
for (int i = 0; i < ListFlowControll->size(); ++i)
|
|
{
|
|
if (((*ListFlowControll)[i])->name().compare("ClassFlowPostProcessing") == 0)
|
|
{
|
|
flowpostprocessing = (ClassFlowPostProcessing*) (*ListFlowControll)[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
ClassFlowMQTT::ClassFlowMQTT(std::vector<ClassFlow*>* lfc, ClassFlow *_prev)
|
|
{
|
|
SetInitialParameter();
|
|
|
|
previousElement = _prev;
|
|
ListFlowControll = lfc;
|
|
|
|
for (int i = 0; i < ListFlowControll->size(); ++i)
|
|
{
|
|
if (((*ListFlowControll)[i])->name().compare("ClassFlowPostProcessing") == 0)
|
|
{
|
|
flowpostprocessing = (ClassFlowPostProcessing*) (*ListFlowControll)[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
bool ClassFlowMQTT::ReadParameter(FILE* pfile, string& aktparamgraph)
|
|
{
|
|
std::vector<string> zerlegt;
|
|
|
|
aktparamgraph = trim(aktparamgraph);
|
|
|
|
if (aktparamgraph.size() == 0)
|
|
if (!this->GetNextParagraph(pfile, aktparamgraph))
|
|
return false;
|
|
|
|
if (toUpper(aktparamgraph).compare("[MQTT]") != 0) // Paragraph passt nich zu MakeImage
|
|
return false;
|
|
|
|
while (this->getNextLine(pfile, &aktparamgraph) && !this->isNewParagraph(aktparamgraph))
|
|
{
|
|
zerlegt = this->ZerlegeZeile(aktparamgraph);
|
|
if ((toUpper(zerlegt[0]) == "USER") && (zerlegt.size() > 1))
|
|
{
|
|
this->user = zerlegt[1];
|
|
}
|
|
if ((toUpper(zerlegt[0]) == "PASSWORD") && (zerlegt.size() > 1))
|
|
{
|
|
this->password = zerlegt[1];
|
|
}
|
|
if ((toUpper(zerlegt[0]) == "URI") && (zerlegt.size() > 1))
|
|
{
|
|
this->uri = zerlegt[1];
|
|
}
|
|
if ((toUpper(zerlegt[0]) == "SETRETAINFLAG") && (zerlegt.size() > 1))
|
|
{
|
|
if (toUpper(zerlegt[1]) == "TRUE")
|
|
SetRetainFlag = 1;
|
|
}
|
|
|
|
|
|
if ((toUpper(zerlegt[0]) == "CLIENTID") && (zerlegt.size() > 1))
|
|
{
|
|
this->clientname = zerlegt[1];
|
|
}
|
|
|
|
if (((toUpper(zerlegt[0]) == "TOPIC") || (toUpper(zerlegt[0]) == "MAINTOPIC")) && (zerlegt.size() > 1))
|
|
{
|
|
maintopic = zerlegt[1];
|
|
}
|
|
else { // Main topic not set, use the hostname
|
|
maintopic = hostname;
|
|
}
|
|
}
|
|
|
|
#ifdef __HIDE_PASSWORD
|
|
ESP_LOGD(TAG, "Init Read with uri: %s, clientname: %s, user: %s, password: XXXXXX, maintopic: %s", uri.c_str(), clientname.c_str(), user.c_str(), maintopic.c_str());
|
|
#else
|
|
ESP_LOGD(TAG, "Init Read with uri: %s, clientname: %s, user: %s, password: %s, maintopic: %s", uri.c_str(), clientname.c_str(), user.c_str(), password.c_str(), maintopic.c_str());
|
|
#endif
|
|
|
|
if (!MQTTisConnected() && (uri.length() > 0) && (maintopic.length() > 0))
|
|
{
|
|
ESP_LOGD(TAG, "InitMQTTInit");
|
|
lwt = maintopic + "/connection";
|
|
#ifdef __HIDE_PASSWORD
|
|
ESP_LOGD(TAG, "Init MQTT with uri: %s, clientname: %s, user: %s, password: XXXXXXXX, maintopic: %s", uri.c_str(), clientname.c_str(), user.c_str(), maintopic.c_str());
|
|
#else
|
|
ESP_LOGD(TAG, "Init MQTT with uri: %s, clientname: %s, user: %s, password: %s, maintopic: %s", uri.c_str(), clientname.c_str(), user.c_str(), password.c_str(), maintopic.c_str());
|
|
#endif
|
|
if (!MQTTInit(uri, clientname, user, password, lwt, keepAlive))
|
|
{ // Failed
|
|
MQTTenable = false;
|
|
return true; // We need to return true despite we failed, else it will retry 5x and then reboot!
|
|
}
|
|
}
|
|
|
|
// Try sending LWT. If it fails, re-run init
|
|
if (!MQTTPublish(lwt, "connected", SetRetainFlag))
|
|
{ // Failed
|
|
LogFile.WriteToFile(ESP_LOG_WARN, "MQTT - Re-running init...!");
|
|
if (!MQTTInit(this->uri, this->clientname, this->user, this->password, this->lwt, keepAlive))
|
|
{ // Failed
|
|
MQTTenable = false;
|
|
return false;
|
|
}
|
|
|
|
// Try again sending LWT and quit if it fails
|
|
if (!MQTTPublish(lwt, "connected", SetRetainFlag))
|
|
{ // Failed
|
|
MQTTenable = false;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
MQTTenable = true;
|
|
return true;
|
|
}
|
|
|
|
|
|
string ClassFlowMQTT::GetMQTTMainTopic()
|
|
{
|
|
return maintopic;
|
|
}
|
|
|
|
|
|
bool ClassFlowMQTT::doFlow(string zwtime)
|
|
{
|
|
std::string result;
|
|
std::string resulterror = "";
|
|
std::string resultraw = "";
|
|
std::string resultrate = "";
|
|
std::string resulttimestamp = "";
|
|
std::string resultchangabs = "";
|
|
string zw = "";
|
|
string namenumber = "";
|
|
|
|
zw = maintopic + "/" + "uptime";
|
|
char uptimeStr[11];
|
|
sprintf(uptimeStr, "%ld", (long)getUpTime());
|
|
|
|
// Try sending uptime. If it fails, re-run init
|
|
if (!MQTTPublish(zw, uptimeStr, SetRetainFlag))
|
|
{ // Failed
|
|
LogFile.WriteToFile(ESP_LOG_WARN, "MQTT - Re-running init...!");
|
|
if (!MQTTInit(this->uri, this->clientname, this->user, this->password, this->lwt, keepAlive))
|
|
{ // Failed
|
|
MQTTenable = false;
|
|
return true; // We need to return true despite we failed, else it will retry 5x and then reboot!
|
|
}
|
|
|
|
// Try again and quit if it fails
|
|
if (!MQTTPublish(zw, uptimeStr, SetRetainFlag))
|
|
{ // Failed
|
|
MQTTenable = false;
|
|
return true; // We need to return true despite we failed, else it will retry 5x and then reboot!
|
|
}
|
|
}
|
|
|
|
zw = maintopic + "/" + "freeMem";
|
|
char freeheapmem[11];
|
|
sprintf(freeheapmem, "%zu", esp_get_free_heap_size());
|
|
if (!MQTTPublish(zw, freeheapmem, SetRetainFlag))
|
|
{ // Failed, skip other topics
|
|
return true; // We need to return true despite we failed, else it will retry 5x and then reboot!
|
|
}
|
|
|
|
zw = maintopic + "/" + "wifiRSSI";
|
|
char rssi[11];
|
|
sprintf(rssi, "%d", get_WIFI_RSSI());
|
|
MQTTPublish(zw, rssi, SetRetainFlag);
|
|
|
|
zw = maintopic + "/" + "CPUtemp";
|
|
std::string cputemp = std::to_string(temperatureRead());
|
|
MQTTPublish(zw, cputemp, SetRetainFlag);
|
|
|
|
if (flowpostprocessing)
|
|
{
|
|
std::vector<NumberPost*>* NUMBERS = flowpostprocessing->GetNumbers();
|
|
|
|
for (int i = 0; i < (*NUMBERS).size(); ++i)
|
|
{
|
|
result = (*NUMBERS)[i]->ReturnValue;
|
|
resultraw = (*NUMBERS)[i]->ReturnRawValue;
|
|
resulterror = (*NUMBERS)[i]->ErrorMessageText;
|
|
resultrate = (*NUMBERS)[i]->ReturnRateValue;
|
|
resultchangabs = (*NUMBERS)[i]->ReturnChangeAbsolute;
|
|
resulttimestamp = (*NUMBERS)[i]->timeStamp;
|
|
|
|
namenumber = (*NUMBERS)[i]->name;
|
|
if (namenumber == "default")
|
|
namenumber = maintopic + "/";
|
|
else
|
|
namenumber = maintopic + "/" + namenumber + "/";
|
|
|
|
zw = namenumber + "value";
|
|
if (result.length() > 0)
|
|
MQTTPublish(zw, result, SetRetainFlag);
|
|
|
|
zw = namenumber + "error";
|
|
if (resulterror.length() > 0)
|
|
MQTTPublish(zw, resulterror, SetRetainFlag);
|
|
|
|
zw = namenumber + "rate";
|
|
if (resultrate.length() > 0)
|
|
MQTTPublish(zw, resultrate, SetRetainFlag);
|
|
|
|
zw = namenumber + "changeabsolut";
|
|
if (resultchangabs.length() > 0)
|
|
MQTTPublish(zw, resultchangabs, SetRetainFlag);
|
|
|
|
zw = namenumber + "raw";
|
|
if (resultraw.length() > 0)
|
|
MQTTPublish(zw, resultraw, SetRetainFlag);
|
|
|
|
zw = namenumber + "timestamp";
|
|
if (resulttimestamp.length() > 0)
|
|
MQTTPublish(zw, resulttimestamp, SetRetainFlag);
|
|
|
|
|
|
std::string json = "";
|
|
|
|
if (result.length() > 0)
|
|
json += "{\"value\":"+result;
|
|
else
|
|
json += "{\"value\":\"\"";
|
|
|
|
json += ",\"raw\":\""+resultraw;
|
|
json += "\",\"error\":\""+resulterror;
|
|
if (resultrate.length() > 0)
|
|
json += "\",\"rate\":"+resultrate;
|
|
else
|
|
json += "\",\"rate\":\"\"";
|
|
json += ",\"timestamp\":\""+resulttimestamp+"\"}";
|
|
|
|
zw = namenumber + "json";
|
|
MQTTPublish(zw, json, SetRetainFlag);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (int i = 0; i < ListFlowControll->size(); ++i)
|
|
{
|
|
zw = (*ListFlowControll)[i]->getReadout();
|
|
if (zw.length() > 0)
|
|
{
|
|
if (result.length() == 0)
|
|
result = zw;
|
|
else
|
|
result = result + "\t" + zw;
|
|
}
|
|
}
|
|
MQTTPublish(topic, result, SetRetainFlag);
|
|
}
|
|
|
|
OldValue = result;
|
|
|
|
return true;
|
|
}
|