added category field to entity discovery payload

1 files changed, 25 insertions, 0 deletions

diff --git a/examples/sensor.rs b/examples/sensor.rs
index 2fd2d1f..acbb3ab 100644
--- a/examples/sensor.rs
+++ b/examples/sensor.rs
@@ -51,8 +51,25 @@ async fn main_task(spawner: Spawner) {
         },
     );
 
+    let signal_strength_sensor = embassy_ha::create_sensor(
+        &device,
+        "signal-strength-sensor-id",
+        embassy_ha::SensorConfig {
+            common: embassy_ha::EntityCommonConfig {
+                name: Some("Signal Strength"),
+                category: Some(embassy_ha::EntityCategory::Diagnostic),
+                ..Default::default()
+            },
+            class: embassy_ha::SensorClass::SignalStrength,
+            state_class: embassy_ha::StateClass::Measurement,
+            unit: Some(embassy_ha::constants::HA_UNIT_SIGNAL_STRENGTH_DBM),
+            suggested_display_precision: Some(0),
+        },
+    );
+
     spawner.must_spawn(random_temperature_task(temperature_sensor));
     spawner.must_spawn(random_humidity_task(humidity_sensor));
+    spawner.must_spawn(random_signal_strength_task(signal_strength_sensor));
 
     embassy_ha::run(&mut device, &mut stream).await.unwrap();
 }
@@ -73,4 +90,12 @@ async fn random_humidity_task(mut sensor: embassy_ha::Sensor<'static>) {
     }
 }
 
+#[embassy_executor::task]
+async fn random_signal_strength_task(mut sensor: embassy_ha::Sensor<'static>) {
+    loop {
+        sensor.publish(rand::random_range(-90.0..-30.0));
+        Timer::after_secs(1).await;
+    }
+}
+
 example_main!();