refactor for more generic charge controller driver

.gitignore

@@ -1,3 +1,4 @@
 /target
 /test-config
 .trigger
+~*.xlsx

src/charge_controllers/gauge_names.rs

@@ -0,0 +1,10 @@
+pub const CHARGE_CONTROLLER_LABEL: &str = "charge_controller";
+pub const PL_LABEL: &str = "pl_device";
+
+pub const BATTERY_VOLTAGE: &str = "pl_battery_voltage";
+pub const TARGET_VOLTAGE: &str = "pl_target_voltage";
+pub const INPUT_CURRENT: &str = "pl_internal_charge_current";
+pub const CHARGE_STATE: &str = "pl_regulator";
+
+pub const PL_DUTY_CYCLE: &str = "pl_duty_cycle";
+pub const PL_LOAD_CURRENT: &str = "pl_internal_load_current";

src/charge_controllers/mod.rs

@@ -0,0 +1,17 @@
+use metrics::describe_gauge;
+
+mod gauge_names;
+pub mod pl;
+
+pub fn register_metrics() {
+    describe_gauge!(gauge_names::BATTERY_VOLTAGE, "Battery voltage");
+    describe_gauge!(gauge_names::TARGET_VOLTAGE, "Target voltage");
+    describe_gauge!(gauge_names::INPUT_CURRENT, "Internal charge current");
+    describe_gauge!(gauge_names::CHARGE_STATE, "Regulator state");
+    register_pl_metrics();
+}
+
+fn register_pl_metrics() {
+    describe_gauge!(gauge_names::PL_DUTY_CYCLE, "Duty cycle");
+    describe_gauge!(gauge_names::PL_LOAD_CURRENT, "Internal load current");
+}

src/charge_controllers/pl.rs

@@ -4,13 +4,15 @@ use std::{
     time::Duration,
 };
 
-use metrics::{describe_gauge, gauge, Gauge};
+use metrics::{gauge, Gauge, Label};
 use serde::{Deserialize, Serialize};
 use serialport::SerialPort;
 use termcolor::WriteColor;
 
 use crate::errors::{PliError, PrintErrors};
 
+use super::gauge_names;
+
 pub struct Pli {
     pub state: Arc<RwLock<PlState>>,
     port: Box<dyn SerialPort>,
@@ -73,12 +75,35 @@ struct RegulatorGauges {
 }
 
 impl RegulatorGauges {
-    fn new() -> Self {
-        describe_gauge!("pl_regulator", "Regulator state");
-        let boost = gauge!("pl_regulator", "state" => "boost");
-        let equalise = gauge!("pl_regulator", "state" => "equalise");
-        let absorption = gauge!("pl_regulator", "state" => "absorption");
-        let float = gauge!("pl_regulator", "state" => "float");
+    fn new(labels: Vec<Label>) -> Self {
+        let boost = gauge!(
+            gauge_names::CHARGE_STATE,
+            [
+                labels.clone(),
+                vec![Label::new("state", String::from("boost"))]
+            ]
+            .concat()
+        );
+        let equalise = gauge!(
+            gauge_names::CHARGE_STATE,
+            [
+                labels.clone(),
+                vec![Label::new("state", String::from("equalise"))]
+            ]
+            .concat()
+        );
+        let absorption = gauge!(
+            gauge_names::CHARGE_STATE,
+            [
+                labels.clone(),
+                vec![Label::new("state", String::from("absorption"))]
+            ]
+            .concat()
+        );
+        let float = gauge!(
+            gauge_names::CHARGE_STATE,
+            [labels, vec![Label::new("state", String::from("float"))]].concat()
+        );
         Self {
             boost,
             equalise,
@@ -128,20 +153,20 @@ impl Pli {
         baud_rate: u32,
         timeout: u64,
     ) -> Result<Self, serialport::Error> {
-        let port = serialport::new(serial_port, baud_rate)
+        let port = serialport::new(serial_port.clone(), baud_rate)
             .timeout(Duration::from_millis(timeout))
             .open()?;
 
-        describe_gauge!("pl_battery_voltage", "Battery voltage");
-        let voltage_gauge = gauge!("pl_battery_voltage");
-        describe_gauge!("pl_target_voltage", "Target voltage");
-        let target_voltage_gauge = gauge!("pl_target_voltage");
-        describe_gauge!("pl_duty_cycle", "Duty cycle");
-        let duty_cycle = gauge!("pl_duty_cycle");
-        describe_gauge!("pl_internal_charge_current", "Internal charge current");
-        let internal_charge_current = gauge!("pl_internal_charge_current");
-        describe_gauge!("pl_internal_load_current", "Internal load current");
-        let internal_load_current = gauge!("pl_internal_load_current");
+        let device_labels = vec![
+            Label::new(gauge_names::CHARGE_CONTROLLER_LABEL, serial_port.clone()),
+            Label::new(gauge_names::PL_LABEL, serial_port),
+        ];
+
+        let voltage_gauge = gauge!(gauge_names::BATTERY_VOLTAGE, device_labels.clone());
+        let target_voltage_gauge = gauge!(gauge_names::TARGET_VOLTAGE, device_labels.clone());
+        let duty_cycle = gauge!(gauge_names::PL_DUTY_CYCLE, device_labels.clone());
+        let internal_charge_current = gauge!(gauge_names::INPUT_CURRENT, device_labels.clone());
+        let internal_load_current = gauge!(gauge_names::PL_LOAD_CURRENT, device_labels.clone());
 
         Ok(Self {
             state: Arc::new(RwLock::new(Default::default())),
@@ -151,7 +176,7 @@ impl Pli {
             duty_cycle,
             internal_charge_current,
             internal_load_current,
-            regulator_gauges: RegulatorGauges::new(),
+            regulator_gauges: RegulatorGauges::new(device_labels),
         })
     }
 

src/main.rs

@@ -4,17 +4,17 @@
 extern crate rocket;
 
 use api_interface::TeslaInterface;
+use charge_controllers::pl::Pli;
 use clap::{Parser, Subcommand};
 use errors::PrintErrors;
-use pl_interface::Pli;
 use std::path::PathBuf;
 
 use crate::config::Config;
 
 mod api_interface;
+mod charge_controllers;
 mod config;
 mod errors;
-mod pl_interface;
 mod server;
 mod types;
 
@@ -61,6 +61,7 @@ async fn main() {
                 let (api_requests, api_receiver) = async_channel::unbounded();
                 // and to the pli thread
                 let (pli_requests, pli_receiver) = async_channel::unbounded();
+                charge_controllers::register_metrics();
 
                 // try to spawn the pli loop
                 let pl_state = match Pli::new(

src/server/mod.rs

@@ -10,9 +10,9 @@ use rocket::{
 
 use crate::{
     api_interface::InterfaceRequest,
+    charge_controllers::pl::{PlState, PliRequest},
     config::Config,
     errors::ServerError,
-    pl_interface::{PlState, PliRequest},
     types::{CarState, ChargeState, ClimateState},
 };