Prototype of the first automation stull ugly

This commit is contained in:
Valère Plantevin
2026-05-12 14:00:12 -04:00
parent 20d59ed0ba
commit 7f54aea439
9 changed files with 152 additions and 4 deletions

View File

@@ -267,6 +267,48 @@
"legendFormat": "{{type}} {{direction}}" "legendFormat": "{{type}} {{direction}}"
} }
] ]
},
{
"id": 11,
"title": "Machine State (Relay)",
"type": "stat",
"gridPos": { "h": 8, "w": 6, "x": 18, "y": 8 },
"datasource": { "type": "prometheus", "uid": "${datasource}" },
"options": {
"colorMode": "background",
"graphMode": "none",
"justifyMode": "auto"
},
"fieldConfig": {
"defaults": {
"mappings": [
{
"options": { "0": { "color": "green", "index": 0, "text": "RUNNING" } },
"type": "value"
},
{
"options": { "1": { "color": "red", "index": 1, "text": "STOPPED" } },
"type": "value"
}
]
}
},
"targets": [
{
"expr": "sensor_aggregate{type=\"relay\", stat=\"max\"}",
"refId": "A"
}
]
},
{
"id": 12,
"title": "Manual Control",
"type": "text",
"gridPos": { "h": 8, "w": 6, "x": 18, "y": 16 },
"options": {
"mode": "html",
"content": "<div style=\"text-align: center; margin-top: 20px;\">\n <button onclick=\"fetch('http://localhost:9002/trigger', {method: 'POST'})\" style=\"padding: 15px 30px; font-size: 20px; background: #e24d42; color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: bold; width: 100%;\">\n 🚨 TRIGGER PRESENCE\n </button>\n</div>"
}
} }
] ]
} }

View File

@@ -43,6 +43,7 @@ services:
- GF_SECURITY_ADMIN_USER=admin - GF_SECURITY_ADMIN_USER=admin
- GF_SECURITY_ADMIN_PASSWORD=admin - GF_SECURITY_ADMIN_PASSWORD=admin
- GF_USERS_DEFAULT_THEME=dark - GF_USERS_DEFAULT_THEME=dark
- GF_PANELS_DISABLE_SANITIZE_HTML=true
volumes: volumes:
- ./grafana/provisioning:/etc/grafana/provisioning:ro - ./grafana/provisioning:/etc/grafana/provisioning:ro
- ../dashboards:/var/lib/grafana/dashboards:ro - ../dashboards:/var/lib/grafana/dashboards:ro

View File

@@ -10,7 +10,7 @@ use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::time::{Duration, SystemTime, UNIX_EPOCH}; use std::time::{Duration, SystemTime, UNIX_EPOCH};
use anyhow::Context; use anyhow::Context;
use substrate::transport::QuicMessage; use substrate::transport::{QuicMessage, SensorType};
use tokio::time::MissedTickBehavior; use tokio::time::MissedTickBehavior;
use crate::profile::{SensorSlot, generate_value}; use crate::profile::{SensorSlot, generate_value};
@@ -97,6 +97,7 @@ pub async fn run_t3_emitter(
ticker.set_missed_tick_behavior(MissedTickBehavior::Skip); ticker.set_missed_tick_behavior(MissedTickBehavior::Skip);
let mut sent: u64 = 0; let mut sent: u64 = 0;
let mut timeouts: u64 = 0; let mut timeouts: u64 = 0;
let mut last_relay_state = 0.0;
loop { loop {
ticker.tick().await; ticker.tick().await;
@@ -115,9 +116,20 @@ pub async fn run_t3_emitter(
slot.seq = slot.seq.wrapping_add(1); slot.seq = slot.seq.wrapping_add(1);
match tokio::time::timeout(timeout, t3_one_request(&conn, &cmd)).await { match tokio::time::timeout(timeout, t3_one_request(&conn, &cmd)).await {
Ok(Ok(_ack)) => { Ok(Ok(ack)) => {
sent += 1; sent += 1;
sent_counter.store(sent, Ordering::Relaxed); sent_counter.store(sent, Ordering::Relaxed);
if ack.sensor_type == SensorType::Relay.as_u8() {
let is_on = ack.raw_value > 0.5;
let was_on = last_relay_state > 0.5;
if is_on && !was_on {
tracing::info!(device = %ack.device_id, "Relay triggered ON (machine stopped)!");
} else if !is_on && was_on {
tracing::info!(device = %ack.device_id, "Relay turned OFF.");
}
last_relay_state = ack.raw_value;
}
} }
Ok(Err(e)) => { Ok(Err(e)) => {
tracing::warn!(error = %e, "T3 request failed"); tracing::warn!(error = %e, "T3 request failed");

View File

@@ -214,6 +214,50 @@ async fn main() -> anyhow::Result<()> {
None None
}; };
let presence_slot_opt = slots.iter().find(|s| s.sensor_type == SensorType::Presence).cloned();
let conn_clone = client.conn.clone();
if let Some(presence_slot) = presence_slot_opt {
tokio::spawn(async move {
if let Ok(listener) = tokio::net::TcpListener::bind("0.0.0.0:9002").await {
tracing::info!("Simulator HTTP trigger API listening on 0.0.0.0:9002");
while let Ok((mut socket, _)) = listener.accept().await {
let conn = conn_clone.clone();
let slot = presence_slot.clone();
tokio::spawn(async move {
let mut buf = [0; 1024];
use tokio::io::{AsyncReadExt, AsyncWriteExt};
if let Ok(n) = socket.read(&mut buf).await {
let req = String::from_utf8_lossy(&buf[..n]);
if req.starts_with("OPTIONS") {
let res = "HTTP/1.1 204 No Content\r\nAccess-Control-Allow-Origin: *\r\nAccess-Control-Allow-Methods: POST, OPTIONS\r\n\r\n";
let _ = socket.write_all(res.as_bytes()).await;
} else if req.starts_with("POST /trigger") {
if let Ok(mut send) = conn.open_uni().await {
let msg = QuicMessage {
device_id: slot.device_id,
sensor_id: slot.sensor_id,
raw_value: 0.0,
timestamp_us: now_us(),
sequence_number: 0,
sensor_type: slot.sensor_type.as_u8(),
};
let _ = send.write_all(&msg.to_bytes()).await;
let _ = send.finish();
tracing::info!("HTTP API triggered: pushed Presence=0.0 over T2");
}
let res = "HTTP/1.1 200 OK\r\nAccess-Control-Allow-Origin: *\r\n\r\nTriggered";
let _ = socket.write_all(res.as_bytes()).await;
} else {
let res = "HTTP/1.1 404 Not Found\r\nAccess-Control-Allow-Origin: *\r\n\r\n";
let _ = socket.write_all(res.as_bytes()).await;
}
}
});
}
}
});
}
let started = Instant::now(); let started = Instant::now();
let mut t1_sent: u64 = 0; let mut t1_sent: u64 = 0;
let mut send_errors: u64 = 0; let mut send_errors: u64 = 0;

View File

@@ -57,6 +57,8 @@ pub fn build_slots(
(2, SensorType::Pressure), (2, SensorType::Pressure),
(3, SensorType::Voltage), (3, SensorType::Voltage),
(4, SensorType::Current), (4, SensorType::Current),
(5, SensorType::Presence),
(6, SensorType::Relay),
] { ] {
slots.push(SensorSlot { slots.push(SensorSlot {
device_id, device_id,
@@ -83,6 +85,8 @@ pub fn generate_value(t: SensorType, seq: u32) -> f64 {
SensorType::Pressure => 1013.0 + 5.0 * (t_phase / 20.0).cos(), SensorType::Pressure => 1013.0 + 5.0 * (t_phase / 20.0).cos(),
SensorType::Voltage => 230.0 + 0.5 * (t_phase / 3.0).sin(), SensorType::Voltage => 230.0 + 0.5 * (t_phase / 3.0).sin(),
SensorType::Current => 10.0 + 2.0 * (t_phase / 5.0).cos(), SensorType::Current => 10.0 + 2.0 * (t_phase / 5.0).cos(),
SensorType::Presence => 2.0 + 1.5 * (t_phase / 5.0).sin(), // Drops below 1.0 occasionally
SensorType::Relay => 0.0, // Relay always sends 0.0 as its command (a pure read request)
SensorType::Generic => t_phase.sin(), SensorType::Generic => t_phase.sin(),
} }
} }

View File

@@ -19,6 +19,8 @@ pub enum SensorType {
Pressure = 3, Pressure = 3,
Voltage = 4, Voltage = 4,
Current = 5, Current = 5,
Presence = 6,
Relay = 7,
} }
impl SensorType { impl SensorType {
@@ -29,6 +31,8 @@ impl SensorType {
3 => Self::Pressure, 3 => Self::Pressure,
4 => Self::Voltage, 4 => Self::Voltage,
5 => Self::Current, 5 => Self::Current,
6 => Self::Presence,
7 => Self::Relay,
_ => Self::Generic, _ => Self::Generic,
} }
} }
@@ -46,6 +50,8 @@ impl SensorType {
Self::Pressure => "pressure", Self::Pressure => "pressure",
Self::Voltage => "voltage", Self::Voltage => "voltage",
Self::Current => "current", Self::Current => "current",
Self::Presence => "presence",
Self::Relay => "relay",
} }
} }
@@ -58,6 +64,8 @@ impl SensorType {
Self::Pressure => "hPa", Self::Pressure => "hPa",
Self::Voltage => "V", Self::Voltage => "V",
Self::Current => "A", Self::Current => "A",
Self::Presence => "s",
Self::Relay => "state",
} }
} }
} }

View File

@@ -91,7 +91,9 @@ pub(super) fn threshold_for(t: SensorType) -> f64 {
SensorType::Temperature => 22.0, // °C — simulator oscillates 15..25 SensorType::Temperature => 22.0, // °C — simulator oscillates 15..25
SensorType::Humidity => 55.0, // % — 30..70 SensorType::Humidity => 55.0, // % — 30..70
SensorType::Pressure => 1014.0, // hPa — 1008..1018 SensorType::Pressure => 1014.0, // hPa — 1008..1018
SensorType::Voltage => 230.2, // V — 229.5..230.5 SensorType::Voltage => 230.5, // V — 229..231
SensorType::Current => 10.5, // A — 8..12 SensorType::Current => 10.5, // A — 8..12
SensorType::Presence => 1.0, // s — Trigger threshold
SensorType::Relay => f64::INFINITY, // Actuator state, no threshold
} }
} }

View File

@@ -41,7 +41,10 @@ impl Plugin for WorldPlugin {
PreUpdate, PreUpdate,
systems::ingest_system.run_if(in_state(ServerState::Started)), systems::ingest_system.run_if(in_state(ServerState::Started)),
) )
.add_systems(Update, systems::simulation_system) .add_systems(
Update,
(systems::simulation_system, systems::automation_system).chain(),
)
.add_systems( .add_systems(
PostUpdate, PostUpdate,
(systems::export_system, systems::diagnostics_system).chain(), (systems::export_system, systems::diagnostics_system).chain(),

View File

@@ -144,6 +144,38 @@ fn upsert_reading(
registry.map.insert(key, entity); registry.map.insert(key, entity);
} }
/// Closed-loop automation triggered by T1/T2 sensor data, affecting a T3 actuator.
pub(super) fn automation_system(
mut registry: ResMut<SensorRegistry>,
mut commands: Commands,
mut p: ParamSet<(
Query<(&DeviceId, &SensorTypeTag, &RawSensorData), Changed<RawSensorData>>,
Query<&mut RawSensorData>,
)>,
) {
let mut triggers = Vec::new();
for (dev_id, tag, data) in p.p0().iter() {
if tag.0 == SensorType::Presence {
// Trigger threshold: 1.0 seconds
let relay_state = if data.raw_value < 1.0 { 1.0 } else { 0.0 };
triggers.push((dev_id.0, relay_state));
}
}
let mut q = p.p1();
for (device_id, relay_state) in triggers {
let msg = QuicMessage {
device_id,
sensor_id: 6, // Relay is always 6 in our industrial profile
raw_value: relay_state,
timestamp_us: now_us(),
sequence_number: 0,
sensor_type: SensorType::Relay.as_u8(),
};
upsert_reading(&mut registry, &mut commands, &mut q, msg);
}
}
/// Per-sensor digital-twin transform. Pulls each entity's latest /// Per-sensor digital-twin transform. Pulls each entity's latest
/// `RawSensorData` into a sliding-window mean (`SmoothedValue`), and emits /// `RawSensorData` into a sliding-window mean (`SmoothedValue`), and emits