REFACTOR: json, mqtt, websocket + CACHE

This commit is contained in:
Patrick Haßel 2025-01-16 15:54:32 +01:00
parent 8bae83ea32
commit 7b5d0b30f4
15 changed files with 206 additions and 143 deletions

View File

@ -7,7 +7,7 @@
#include "patrix/sensor/DHT22.h" #include "patrix/sensor/DHT22.h"
#include "patrix/sensor/Max6675Sensor.h" #include "patrix/sensor/Max6675Sensor.h"
class NodeHeizung final : public Node { class NodeHeizung final : public Node<100, 14> {
const unsigned long MAX_AGE_SECONDS = 10; const unsigned long MAX_AGE_SECONDS = 10;

View File

@ -1,12 +1,14 @@
#ifndef NODE_TEST_H #ifndef NODE_TEST_H
#define NODE_TEST_H #define NODE_TEST_H
#include <AsyncWebSocket.h>
#include "patrix/Node.h" #include "patrix/Node.h"
#include "patrix/sensor/Dallas.h" #include "patrix/sensor/Dallas.h"
#include "patrix/sensor/DallasSensor.h" #include "patrix/sensor/DallasSensor.h"
#include "patrix/sensor/DHT22.h" #include "patrix/sensor/DHT22.h"
class NodeTest final : public Node { class NodeTest final : public Node<100, 4> {
const unsigned long MAX_AGE_SECONDS = 10; const unsigned long MAX_AGE_SECONDS = 10;

82
src/patrix/Cache.h Normal file
View File

@ -0,0 +1,82 @@
#ifndef CACHE_H
#define CACHE_H
#include "log.h"
template<size_t ENTRY_COUNT, size_t VALUE_COUNT>
class Cache {
public:
struct Entry {
time_t timestamp;
double data[VALUE_COUNT];
};
private:
Entry entries[ENTRY_COUNT];
Entry *r = entries;
Entry *w = entries;
void clearWriteEntry() {
for (double& value: r->data) {
value = 0;
}
}
Entry *findEntry(time_t timestamp) {
if (w->timestamp == timestamp) {
return w;
}
auto wNext = (w + 1 - entries) % ENTRY_COUNT + entries;
if (w->timestamp == 0) {
w->timestamp = timestamp;
clearWriteEntry();
} else if (wNext != r) {
w = wNext;
w->timestamp = timestamp;
clearWriteEntry();
} else {
return nullptr;
}
return w;
}
public:
void put(const time_t timestamp, const int valueIndex, const double value) {
if (valueIndex >= VALUE_COUNT) {
error("Cache.put: valueIndex to big for valueCount: valueIndex=%d, valueCount=%d", valueIndex, VALUE_COUNT);
return;
}
Entry *entry = findEntry(timestamp);
if (entry == nullptr) {
warn("Cache full!");
return;
}
entry->data[valueIndex] = value;
}
Entry *read() {
if (w != r || r->timestamp != 0) {
return r;
}
return nullptr;
}
void skip() {
if (w != r || w->timestamp != 0) {
r->timestamp = 0;
r = (r + 1 - entries) % ENTRY_COUNT + entries;
}
}
};
#endif

View File

@ -1,33 +1,38 @@
#ifndef NODE_H #ifndef NODE_H
#define NODE_H #define NODE_H
#include "Cache.h"
#include "clock.h"
#include "sensor/Sensor.h" #include "sensor/Sensor.h"
template<size_t ENTRY_COUNT, size_t VALUE_COUNT>
class Node { class Node {
Cache<ENTRY_COUNT, VALUE_COUNT> cache;
public: public:
Node() : cache() {
//
}
virtual ~Node() = default; virtual ~Node() = default;
void toJson(const JsonObject& json) {
auto index = 0;
Sensor *sensor;
while ((sensor = getSensor(index++)) != nullptr) {
sensor->toJson(json[sensor->getName()].to<JsonObject>());
}
}
void send(const bool mqtt, const bool websocket) {
JsonDocument json;
toJson(json.to<JsonObject>());
mqttPublish(String(WiFiClass::getHostname()) + "/json", json, mqtt, websocket);
}
void loop() { void loop() {
loopBeforeSensors(); loopBeforeSensors();
const auto changed = loopSensors(); loopSensors();
if (changed) { cacheSend();
send(true, true); }
void websocketSendOne(AsyncWebSocketClient *client) {
auto sensorIndex = 0;
Sensor *sensor;
while ((sensor = this->getSensor(sensorIndex++)) != nullptr) {
auto valueIndex = 0;
Value *value;
while ((value = sensor->getValue(valueIndex++)) != nullptr) {
client->text(value->toJson(true));
}
} }
} }
@ -39,16 +44,48 @@ public:
private: private:
bool loopSensors() { void loopSensors() {
auto changed = false; const auto timestamp = time(nullptr);
auto index = 0; auto cacheIndex = 0;
auto sensorIndex = 0;
Sensor *sensor; Sensor *sensor;
while ((sensor = getSensor(index++)) != nullptr) { while ((sensor = getSensor(sensorIndex++)) != nullptr) {
if (sensor->loop()) { auto valueIndex = 0;
changed = true; Value *value;
while ((value = sensor->getValue(valueIndex++)) != nullptr) {
if (value->loop()) {
cache.put(timestamp, cacheIndex++, value->getCurrentValue());
} }
} }
return changed; }
}
void cacheSend() {
const auto entry = cache.read();
if (entry == nullptr) {
return;
}
auto cacheIndex = 0;
auto sensorIndex = 0;
Sensor *sensor;
while ((sensor = getSensor(sensorIndex++)) != nullptr) {
auto valueIndex = 0;
Value *value;
while ((value = sensor->getValue(valueIndex++)) != nullptr) {
const auto v = entry->data[cacheIndex++];
const auto t = entry->timestamp;
if (isCorrectTime(t) && !isnan(v)) {
auto json = value->toJson(t, v, false);
if (mqttPublish(value->name + "/persist", json, NO_RETAIN)) {
entry->data[cacheIndex++] = NAN;
} else {
return;
}
}
}
}
cache.skip();
} }
}; };

View File

@ -2,10 +2,10 @@
#include <ESPAsyncWebServer.h> #include <ESPAsyncWebServer.h>
#include "INDEX_HTML.h"
#include "log.h" #include "log.h"
#include "main.h" #include "main.h"
#include "system.h" #include "system.h"
#include "INDEX_HTML.h"
AsyncWebServer server(80); AsyncWebServer server(80);
@ -31,7 +31,7 @@ void httpSetup() {
switch (type) { switch (type) {
case WS_EVT_CONNECT: case WS_EVT_CONNECT:
t = "CONNECT"; t = "CONNECT";
node.send(false, true); // TODO this currently sends to ALL node.websocketSendOne(client);
break; break;
case WS_EVT_DISCONNECT: case WS_EVT_DISCONNECT:
t = "DISCONNECT"; t = "DISCONNECT";
@ -62,6 +62,6 @@ void httpLoop() {
ws.cleanupClients(); ws.cleanupClients();
} }
void httpPublish(char *payload) { void websocketSendAll(const String& payload) {
ws.textAll(payload); ws.textAll(payload);
} }

View File

@ -1,10 +1,12 @@
#ifndef HTTP_H #ifndef HTTP_H
#define HTTP_H #define HTTP_H
#include <Arduino.h>
void httpSetup(); void httpSetup();
void httpLoop(); void httpLoop();
void httpPublish(char *payload); void websocketSendAll(const String& payload);
#endif #endif

View File

@ -7,9 +7,13 @@
#include "mqtt.h" #include "mqtt.h"
#include "wifi.h" #include "wifi.h"
// ReSharper disable CppUnusedIncludeDirective
#include "sensor/DallasSensor.h" #include "sensor/DallasSensor.h"
#include "sensor/DHT22.h" #include "sensor/DHT22.h"
#include "sensor/Max6675Sensor.h" #include "sensor/Max6675Sensor.h"
// ReSharper restore CppUnusedIncludeDirective
// ReSharper disable CppUseAuto
#ifdef NODE_TEST #ifdef NODE_TEST
NodeTest node = NodeTest(); NodeTest node = NodeTest();
@ -19,6 +23,8 @@ NodeTest node = NodeTest();
NodeHeizung node = NodeHeizung(); NodeHeizung node = NodeHeizung();
#endif #endif
// ReSharper restore CppUseAuto
void setup() { void setup() {
logSetup(); logSetup();
bootDelay(); bootDelay();

View File

@ -1,8 +1,10 @@
#ifndef MAIN_H #ifndef MAIN_H
#define MAIN_H #define MAIN_H
#include "NodeTest.h" // ReSharper disable CppUnusedIncludeDirective
#include "NodeHeizung.h" #include "NodeHeizung.h"
#include "NodeTest.h"
// ReSharper restore CppUnusedIncludeDirective
#ifdef NODE_TEST #ifdef NODE_TEST
extern NodeTest node; extern NodeTest node;

View File

@ -3,7 +3,6 @@
#include <PubSubClient.h> #include <PubSubClient.h>
#include <WiFi.h> #include <WiFi.h>
#include "http.h"
#include "log.h" #include "log.h"
#include "wifi.h" #include "wifi.h"
@ -47,19 +46,6 @@ void mqttLoop() {
} }
} }
bool mqttPublish(const String& topic, const double value, const bool retained) { bool mqttPublish(const String& topic, const String& payload, const Retain retain) {
return mqtt.publish(topic.c_str(), String(value).c_str(), retained); return mqtt.publish(topic.c_str(), payload.c_str(), retain == RETAIN);
}
bool mqttPublish(const String& topic, const JsonDocument& json, const bool mqttSend, const bool websocketSend) {
char buffer[512];
serializeJson(json, buffer);
if (websocketSend) {
httpPublish(buffer);
}
if (mqttSend) {
return mqtt.publish(topic.c_str(), buffer);
}
return true;
} }

View File

@ -1,12 +1,14 @@
#ifndef MQTT_H #ifndef MQTT_H
#define MQTT_H #define MQTT_H
#include <ArduinoJson.h> #include <AsyncWebSocket.h>
enum Retain {
NO_RETAIN, RETAIN
};
void mqttLoop(); void mqttLoop();
bool mqttPublish(const String& topic, const double value, bool retained); bool mqttPublish(const String& topic, const String& payload, Retain retain);
bool mqttPublish(const String& topic, const JsonDocument& json, bool mqtt, bool websocketSend);
#endif #endif

View File

@ -19,7 +19,7 @@ class DHT22 final : public Sensor {
public: public:
DHT22(const int pin, DHT22(const int pin,
const char *name, const String& name,
const double temperatureThreshold, const double temperatureThreshold,
const double humidityRelativeThreshold, const double humidityRelativeThreshold,
const double humidityAbsoluteThreshold, const double humidityAbsoluteThreshold,
@ -27,9 +27,9 @@ public:
const unsigned long overdueSeconds const unsigned long overdueSeconds
) : Sensor(name), ) : Sensor(name),
sensor(pin, DHT_TYPE_22), sensor(pin, DHT_TYPE_22),
temperature(name, "temperature", temperatureThreshold, maxAgeSeconds, overdueSeconds), temperature(name + "/temperature", temperatureThreshold, maxAgeSeconds, overdueSeconds),
humidityRelative(name, "humidity/relative", humidityRelativeThreshold, maxAgeSeconds, overdueSeconds), humidityRelative(name + "/humidity/relative", humidityRelativeThreshold, maxAgeSeconds, overdueSeconds),
humidityAbsolute(name, "humidity/absolute", humidityAbsoluteThreshold, maxAgeSeconds, overdueSeconds) { humidityAbsolute(name + "/humidity/absolute", humidityAbsoluteThreshold, maxAgeSeconds, overdueSeconds) {
// //
} }

View File

@ -19,14 +19,14 @@ public:
DallasSensor( DallasSensor(
Dallas& sensors, Dallas& sensors,
const uint64_t address, const uint64_t address,
const char *name, const String& name,
const double threshold, const double threshold,
const unsigned long maxAgeSeconds, const unsigned long maxAgeSeconds,
const unsigned long overdueSeconds const unsigned long overdueSeconds
) : Sensor(name), ) : Sensor(name),
sensors(sensors), sensors(sensors),
address(address), address(address),
temperature(name, "temperature", threshold, maxAgeSeconds, overdueSeconds) { temperature(name + "/temperature", threshold, maxAgeSeconds, overdueSeconds) {
// //
} }

View File

@ -20,13 +20,13 @@ public:
const int8_t pinMISO, const int8_t pinMISO,
const int8_t pinCS, const int8_t pinCS,
const int8_t pinCLK, const int8_t pinCLK,
const char *name, const String& name,
const double threshold, const double threshold,
const unsigned long maxAgeSeconds, const unsigned long maxAgeSeconds,
const unsigned long overdueSeconds const unsigned long overdueSeconds
) : Sensor(name), ) : Sensor(name),
sensor(pinCLK, pinCS, pinMISO), sensor(pinCLK, pinCS, pinMISO),
temperature(name, "temperature", threshold, maxAgeSeconds, overdueSeconds) { temperature(name + "/temperature", threshold, maxAgeSeconds, overdueSeconds) {
// //
} }

View File

@ -6,63 +6,20 @@
class Sensor { class Sensor {
protected:
const char *name;
public: public:
explicit Sensor(const char *name): name(name) { const String& name;
explicit Sensor(const String& name): name(name) {
// //
} }
virtual ~Sensor() = default; virtual ~Sensor() = default;
const char *getName() const {
return name;
}
bool loop() {
loopBeforeValues();
const auto changed = loopValues();
if (changed) {
send();
}
return changed;
}
void send() {
JsonDocument json;
toJson(json.to<JsonObject>());
mqttPublish(String(name) + "/json", json, true, false);
}
void toJson(const JsonObject& json) {
auto index = 0;
Value *value;
while ((value = getValue(index++)) != nullptr) {
value->toJson(json[value->getName()].to<JsonObject>());
}
}
virtual void loopBeforeValues() {} virtual void loopBeforeValues() {}
virtual Value *getValue(int index) = 0; virtual Value *getValue(int index) = 0;
private:
bool loopValues() {
auto changed = false;
auto index = 0;
Value *value;
while ((value = getValue(index++)) != nullptr) {
if (value->loop()) {
changed = true;
}
}
return changed;
}
}; };
#endif #endif

View File

@ -1,16 +1,13 @@
#ifndef VALUE_H #ifndef VALUE_H
#define VALUE_H #define VALUE_H
#include <WiFi.h> #include <ArduinoJson.h>
#include <patrix/http.h>
#include <patrix/log.h> #include <patrix/log.h>
#include <patrix/mqtt.h> #include <patrix/mqtt.h>
class Value { class Value {
const char *parent;
const char *name;
double threshold; double threshold;
unsigned long maxAgeMillis; unsigned long maxAgeMillis;
@ -29,17 +26,17 @@ class Value {
public: public:
const String name;
Value( Value(
const char *parent, const String& name,
const char *name,
const double threshold, const double threshold,
const unsigned long maxAgeSeconds, const unsigned long maxAgeSeconds,
const unsigned long overdueSeconds const unsigned long overdueSeconds
) : parent(parent), ) : threshold(threshold),
name(name),
threshold(threshold),
maxAgeMillis(maxAgeSeconds * 1000), maxAgeMillis(maxAgeSeconds * 1000),
overdueSeconds(overdueSeconds) { overdueSeconds(overdueSeconds),
name(name) {
// //
} }
@ -49,24 +46,9 @@ public:
currentEpoch = time(nullptr); currentEpoch = time(nullptr);
} }
void toJson(const JsonObject& json) const {
json["value"] = currentValue;
json["time"] = currentEpoch;
}
void send() {
mqttPublish(String(parent) + "/" + name + "/plain", currentValue, true);
JsonDocument json;
toJson(json.to<JsonObject>());
mqttPublish(String(parent) + "/" + name + "/json", json, true, false);
markSent();
}
bool loop() { bool loop() {
if (!isnan(currentValue) && millis() - currentMillis > maxAgeMillis) { if (!isnan(currentValue) && millis() - currentMillis > maxAgeMillis) {
warn("Value too old: %s/%s", parent, name); warn("Value too old: %s", name.c_str());
update(NAN); update(NAN);
} }
@ -75,28 +57,33 @@ public:
const auto dueToTime = sentInterval != 0 && sentInterval != time(nullptr) / overdueSeconds; const auto dueToTime = sentInterval != 0 && sentInterval != time(nullptr) / overdueSeconds;
const auto changed = dueToNAN || dueToThreshold || dueToTime; const auto changed = dueToNAN || dueToThreshold || dueToTime;
if (changed) { if (changed) {
send(); mqttPublish(name + "/retain", String(currentValue), RETAIN);
websocketSendAll(toJson(false));
sentValue = currentValue;
sentInterval = time(nullptr) / overdueSeconds;
} }
return changed; return changed;
} }
const char *getName() const {
return name;
}
double getCurrentValue() const { double getCurrentValue() const {
return currentValue; return currentValue;
} }
time_t getCurrentEpoch() const { String toJson(const bool addNameField) const {
return currentEpoch; return toJson(currentEpoch, currentValue, addNameField);
} }
private: String toJson(const time_t timestamp, const double value, const bool addNameField) const {
JsonDocument json;
if (addNameField) {
json["name"] = name.c_str();
}
json["timestamp"] = timestamp;
json["value"] = value;
void markSent() { char buffer[256];
sentValue = currentValue; serializeJson(json, buffer, sizeof buffer);
sentInterval = time(nullptr) / overdueSeconds; return String(buffer);
} }
}; };