add publish flag to SensorData and conditionally process MQTT messages

Author: psytraxx

src/domain.rs

@@ -17,6 +17,7 @@ const MOISTURE_DRY_THRESHOLD: f32 = 0.15;
 #[derive(Default, Debug)]
 pub struct SensorData {
     pub data: Vec<Sensor, 7>,
+    pub publish: bool,
 }
 
 impl Display for SensorData {

src/sensors_task.rs

@@ -199,10 +199,11 @@ pub async fn sensor_task(
                 .data
                 .push(Sensor::BatteryVoltage(avg_battery_voltage))
                 .expect("Too many samples");
-        } else {
-            println!("Error measuring battery voltage");
         }
 
+        // no battery samples - no publish!
+        sensor_data.publish = !battery_voltage_samples.is_empty();
+
         if battery_voltage_samples.is_empty() {
             println!(
                 "No battery voltage samples collected - skipping this cycle {:?}",

src/update_task.rs

@@ -146,6 +146,20 @@ async fn handle_sensor_data(
     client: &mut MqttClientImpl<'_>,
     display: &mut Display<'static, Delay>,
     sensor_data: SensorData,
+) -> Result<(), Error> {
+    if sensor_data.publish {
+        process_mqtt(client, &sensor_data).await?;
+    } else {
+        println!("skipping publishing to MQTT");
+    }
+
+    process_display(display, &sensor_data).await?;
+    Ok(())
+}
+
+async fn process_mqtt(
+    client: &mut MqttClientImpl<'_>,
+    sensor_data: &SensorData,
 ) -> Result<(), Error> {
     let discovery_messages_sent = unsafe { DISCOVERY_MESSAGES_SENT };
     if !discovery_messages_sent {
@@ -179,14 +193,12 @@ async fn handle_sensor_data(
         println!("Discovery messages already sent");
     }
 
-    // act on sensor data - turn pump on/off
+    // Act on sensor data - turn pump on/off
     sensor_data.data.iter().for_each(|entry| {
         if let Sensor::WaterLevel(WaterLevel::Full) = entry {
-            // Water level is full, stop the pump in any case if it's running
             println!("Water level is full, stopping pump");
-            update_pump_state(false);
+            ENABLE_PUMP.signal(false);
         } else if let Sensor::PumpTrigger(enabled) = entry {
-            // Pump trigger is enabled, start the pump
             if *enabled {
                 println!("Soil moisture is low, starting pump");
                 update_pump_state(true);
@@ -216,8 +228,6 @@ async fn handle_sensor_data(
             .await?;
     }
 
-    display.write_multiline(&format!("{}", sensor_data))?;
-
     let pump_topic = format!("{}/pump/state", DEVICE_ID);
     let message = "OFF";
     println!(
@@ -234,11 +244,16 @@ async fn handle_sensor_data(
             false,
         )
         .await?;
+    Ok(())
+}
 
+async fn process_display(
+    display: &mut Display<'static, Delay>,
+    sensor_data: &SensorData,
+) -> Result<(), Error> {
+    display.write_multiline(&format!("{}", sensor_data))?;
     Timer::after(Duration::from_secs(AWAKE_DURATION_SECONDS)).await;
-
     display.enable_powersave()?;
-
     Ok(())
 }