Added support for InfluxDB logging of fertilizer and plant watering events

include/WifiStuff.h

@@ -31,6 +31,8 @@ void wifi_send_status_broadcast(void);
 
 void wifi_send_websocket(String s);
 
+bool wifi_write_database(int duration, const char *type, int id);
+
 #endif
 
 #endif // _WIFI_STUFF_H_

include/config.h

@@ -56,4 +56,14 @@
 #define I2C_BUS_SPEED 400000
 #define I2C_BUF_SIZE 32
 
+//#define ENABLE_GPIO_TEST
+#define GPIO_TEST_INTERVAL 4000
+#define GPIO_TEST_DELAY 200
+
+// InfluxDB settings
+#define ENABLE_INFLUXDB_LOGGING
+#define INFLUXDB_HOST "10.23.42.14"
+#define INFLUXDB_PORT 8086
+#define INFLUXDB_DATABASE "giessomat"
+
 #endif // _CONFIG_H_

platformio.ini

@@ -32,6 +32,7 @@ lib_deps =
     Wire
     https://github.com/Links2004/arduinoWebSockets
     https://github.com/rlogiacco/CircularBuffer
+    https://github.com/tobiasschuerg/InfluxDB-Client-for-Arduino.git
 
 [env:arduino_ui]
 platform = atmelavr

src/Statemachine.cpp

@@ -19,6 +19,7 @@
 
 #include "Plants.h"
 #include "DebugLog.h"
+#include "WifiStuff.h"
 #include "Statemachine.h"
 #include "config.h"
 
@@ -200,7 +201,6 @@ void Statemachine::input(int n) {
                             plants.startPlant(i);
                         }
                     }
-                    selected_plants.clear();
                     
                     selected_time = MAX_AUTO_PLANT_RUNTIME;
                     start_time = millis();
@@ -623,6 +623,37 @@ void Statemachine::switch_to(States s) {
               a.c_str(),
               "Hit any key for menu",
               -1);
+
+#if defined(PLATFORM_ESP)
+        unsigned long runtime = stop_time - start_time;
+        if (old_state == auto_plant_run) {
+            for (int i = 0; i < plants.countPlants(); i++) {
+                if (selected_plants.isSet(i)) {
+                    bool success = wifi_write_database(runtime / 1000, "plant", i + 1);
+                    if (!success) {
+                        debug.print("Error writing to InfluxDB ");
+                        debug.print(INFLUXDB_HOST);
+                        debug.print(":");
+                        debug.print(INFLUXDB_PORT);
+                        debug.print("/");
+                        debug.print(INFLUXDB_DATABASE);
+                        debug.println("/plant");
+                    }
+                }
+            }
+        } else if (old_state == auto_fert_run) {
+            bool success = wifi_write_database(runtime / 1000, "fertilizer", selected_id);
+            if (!success) {
+                debug.print("Error writing to InfluxDB ");
+                debug.print(INFLUXDB_HOST);
+                debug.print(":");
+                debug.print(INFLUXDB_PORT);
+                debug.print("/");
+                debug.print(INFLUXDB_DATABASE);
+                debug.println("/fertilizer");
+            }
+        }
+#endif // PLATFORM_ESP
     } else if (s == menu_pumps) {
         String a = String("(Input 1 to ") + String(plants.countFertilizers()) + String(")");
         
@@ -671,6 +702,20 @@ void Statemachine::switch_to(States s) {
               a.c_str(),
               "Hit any key for menu",
               -1);
+
+#if defined(PLATFORM_ESP)
+        unsigned long runtime = stop_time - start_time;
+        bool success = wifi_write_database(runtime / 1000, "fertilizer", selected_id);
+        if (!success) {
+            debug.print("Error writing to InfluxDB ");
+            debug.print(INFLUXDB_HOST);
+            debug.print(":");
+            debug.print(INFLUXDB_PORT);
+            debug.print("/");
+            debug.print(INFLUXDB_DATABASE);
+            debug.println("/fertilizer");
+        }
+#endif // PLATFORM_ESP
     } else if (s == menu_valves) {
         String a = String("(Input 1 to ") + String(plants.countPlants() + 1) + String(")");
         
@@ -719,6 +764,22 @@ void Statemachine::switch_to(States s) {
               a.c_str(),
               "Hit any key for menu",
               -1);
+
+#if defined(PLATFORM_ESP)
+        unsigned long runtime = stop_time - start_time;
+        if (selected_id <= plants.countPlants()) {
+            bool success = wifi_write_database(runtime / 1000, "plant", selected_id);
+            if (!success) {
+                debug.print("Error writing to InfluxDB ");
+                debug.print(INFLUXDB_HOST);
+                debug.print(":");
+                debug.print(INFLUXDB_PORT);
+                debug.print("/");
+                debug.print(INFLUXDB_DATABASE);
+                debug.println("/plant");
+            }
+        }
+#endif // PLATFORM_ESP
     } else if (s == error) {
         print("------ Error! ------",
               "There is a problem:",

src/WifiStuff.cpp

@@ -56,14 +56,18 @@ String message_buffer_b;
 String message_buffer_c;
 String message_buffer_d;
 
-//#define ENABLE_GPIO_TEST
 #ifdef ENABLE_GPIO_TEST
-
-#define GPIO_TEST_INTERVAL 4000
-#define GPIO_TEST_DELAY 200
 static bool runningGpioTest = false;
 static bool gpioTestState = false;
 unsigned long lastGpioTime = 0;
+#endif // ENABLE_GPIO_TEST
+
+#ifdef ENABLE_INFLUXDB_LOGGING
+#include <InfluxDb.h>
+Influxdb influx(INFLUXDB_HOST, INFLUXDB_PORT);
+#endif // ENABLE_INFLUXDB_LOGGING
+
+#ifdef ENABLE_GPIO_TEST
 
 void runGpioTest(bool state) {
     lastGpioTime = millis();
@@ -95,6 +99,21 @@ void handleGpioTest() {
 
 #endif // ENABLE_GPIO_TEST
 
+bool wifi_write_database(int duration, const char *type, int id) {
+    bool success = false;
+
+#ifdef ENABLE_INFLUXDB_LOGGING
+    InfluxData measurement(type);
+    measurement.addTag("version", FIRMWARE_VERSION);
+    measurement.addTag("device", WiFi.macAddress());
+    measurement.addTag("id", String(id));
+    measurement.addValue("duration", duration);
+    success = influx.write(measurement);
+#endif // ENABLE_INFLUXDB_LOGGING
+
+    return success;
+}
+
 void wifi_set_message_buffer(String a, String b, String c, String d) {
     message_buffer_a = a;
     message_buffer_b = b;
@@ -488,6 +507,20 @@ void handleRoot() {
     
 #endif
 
+    message += F("<p>\n");
+#ifdef ENABLE_INFLUXDB_LOGGING
+    message += F("InfluxDB: ");
+    message += INFLUXDB_DATABASE;
+    message += F(" @ ");
+    message += INFLUXDB_HOST;
+    message += F(":");
+    message += String(INFLUXDB_PORT);
+    message += F("\n");
+#else
+    message += F("InfluxDB logging not enabled!\n");
+#endif
+    message += F("</p>\n");
+
     message += F("<p>Try <a href='/update'>/update</a> for OTA firmware updates!</p>\n");
     message += F("<p>Made by <a href='https://xythobuz.de'>xythobuz</a></p>\n");
     message += F("<p><a href='https://git.xythobuz.de/thomas/giess-o-mat'>Project Repository</a></p>\n");
@@ -647,6 +680,12 @@ void wifi_setup() {
 
 #endif
 
+#ifdef ENABLE_INFLUXDB_LOGGING
+    // Setup InfluxDB Client
+    debug.println("WiFi: set InfluxDB database");
+    influx.setDb(INFLUXDB_DATABASE);
+#endif // ENABLE_INFLUXDB_LOGGING
+
     // Setup HTTP Server
     debug.println("WiFi: initializing HTTP server");
     MDNS.begin(hostname.c_str());