From 251f373128ef0af66a73957107f08c47d7184806 Mon Sep 17 00:00:00 2001
From: pkkrusty <79770016+pkkrusty@users.noreply.github.com>
Date: Mon, 2 May 2022 13:48:45 +0000
Subject: [PATCH] Add option to specify retention policy

Certain sensors and data flowing to a TSDB like Influx will need different retention policies. Network monitoring, maybe one week, soil moisture for the garden, one month, temperature data, a few years. To avoid growing a large database, it makes sense to assign a retention policy per device if desired.(since most devices won't be doing weather data AND network reporting). Blank retention policy will just use the InfluxDB default (usually infinite).
---
 tasmota/xdrv_59_influxdb.ino | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/tasmota/xdrv_59_influxdb.ino b/tasmota/xdrv_59_influxdb.ino
index 70f9b2803..48adb935c 100644
--- a/tasmota/xdrv_59_influxdb.ino
+++ b/tasmota/xdrv_59_influxdb.ino
@@ -39,8 +39,7 @@
  * IfxToken    - Set Influxdb v2 and token
  * IfxPeriod   - Set Influxdb period. If not set (or 0), use Teleperiod
  * IfxSensor   - Set Influxdb sensor logging off (0) or on (1)
- *
- * Set influxdb update interval with command teleperiod
+ * IfxRP       - Set Influxdb retention policy
  *
  * The following triggers result in automatic influxdb numeric feeds without appended time:
  * - this driver initiated state message
@@ -73,6 +72,9 @@
 #ifndef INFLUXDB_BUCKET
 #define INFLUXDB_BUCKET    "db"          // [IfxDatabase, IfxBucket] Influxdb v1 database or v2 bucket
 #endif
+#ifndef INFLUXDB_RP
+#define INFLUXDB_RP        ""          // [IfxRP] Influxdb v1 retention policy (blank is default, usually autogen infinite)
+#endif
 
 static const char UninitializedMessage[] PROGMEM = "Unconfigured instance";
 // This cannot be put to PROGMEM due to the way how it is used
@@ -132,6 +134,10 @@ bool InfluxDbParameterInit(void) {
     IFDB._writeUrl += "/write?db=";
     IFDB._writeUrl += UrlEncode(SettingsText(SET_INFLUXDB_BUCKET));
     IFDB._writeUrl += InfluxDbAuth();
+    if (strlen(SettingsText(SET_INFLUXDB_RP)) != 0) {
+      IFDB._writeUrl += "&rp=";
+      IFDB._writeUrl += UrlEncode(SettingsText(SET_INFLUXDB_RP));
+    }
   }
   AddLog(LOG_LEVEL_DEBUG, PSTR("IFX: Url %s"), IFDB._writeUrl.c_str());
 
@@ -427,6 +433,7 @@ void InfluxDbLoop(void) {
 #define D_CMND_INFLUXDBBUCKET   "Bucket"
 #define D_CMND_INFLUXDBPERIOD   "Period"
 #define D_CMND_INFLUXDBSENSOR   "Sensor"
+#define D_CMND_INFLUXDBRP "RP"
 
 const char kInfluxDbCommands[] PROGMEM = D_PRFX_INFLUXDB "|"  // Prefix
   "|" D_CMND_INFLUXDBLOG "|"
@@ -434,7 +441,7 @@ const char kInfluxDbCommands[] PROGMEM = D_PRFX_INFLUXDB "|"  // Prefix
   D_CMND_INFLUXDBUSER "|" D_CMND_INFLUXDBORG "|"
   D_CMND_INFLUXDBPASSWORD "|" D_CMND_INFLUXDBTOKEN "|"
   D_CMND_INFLUXDBDATABASE "|" D_CMND_INFLUXDBBUCKET "|"
-  D_CMND_INFLUXDBPERIOD "|" D_CMND_INFLUXDBSENSOR;
+  D_CMND_INFLUXDBPERIOD "|" D_CMND_INFLUXDBSENSOR "|" D_CMND_INFLUXDBRP;
 
 void (* const InfluxCommand[])(void) PROGMEM = {
   &CmndInfluxDbState, &CmndInfluxDbLog,
@@ -442,7 +449,7 @@ void (* const InfluxCommand[])(void) PROGMEM = {
   &CmndInfluxDbUser, &CmndInfluxDbUser,
   &CmndInfluxDbPassword, &CmndInfluxDbPassword,
   &CmndInfluxDbDatabase, &CmndInfluxDbDatabase,
-  &CmndInfluxDbPeriod, &CmndInfluxDbSensor };
+  &CmndInfluxDbPeriod, &CmndInfluxDbSensor, &CmndInfluxDbRP };
 
 void InfluxDbReinit(void) {
   IFDB.init = false;
@@ -532,6 +539,14 @@ void CmndInfluxDbDatabase(void) {
   ResponseCmndChar(SettingsText(SET_INFLUXDB_BUCKET));
 }
 
+void CmndInfluxDbRP(void) {
+  if (XdrvMailbox.data_len > 0) {
+    SettingsUpdateText(SET_INFLUXDB_RP, (SC_CLEAR == Shortcut()) ? "" : (SC_DEFAULT == Shortcut()) ? PSTR(INFLUXDB_RP) : XdrvMailbox.data);
+    InfluxDbReinit();
+  }
+  ResponseCmndChar(SettingsText(SET_INFLUXDB_RP));
+}
+
 void CmndInfluxDbPeriod(void) {
   if ((XdrvMailbox.payload >= 0) && (XdrvMailbox.payload < 3601)) {
     Settings->influxdb_period = XdrvMailbox.payload;
@@ -559,6 +574,7 @@ bool Xdrv59(uint8_t function) {
       SettingsUpdateText(SET_INFLUXDB_ORG, PSTR(INFLUXDB_ORG));
       SettingsUpdateText(SET_INFLUXDB_TOKEN, PSTR(INFLUXDB_TOKEN));
       SettingsUpdateText(SET_INFLUXDB_BUCKET, PSTR(INFLUXDB_BUCKET));
+      SettingsUpdateText(SET_INFLUXDB_RP, PSTR(INFLUXDB_RP));
       Settings->sbflag1.influxdb_default = 1;
     }
   } else if (FUNC_COMMAND == function) {