Add LDLM stats (#20)

* Add LDLM stats

* Apply suggestions from code review

Co-authored-by: Joe Grund <jgrund@whamcloud.io>

* Use published crate

---------

Co-authored-by: Joe Grund <jgrund@whamcloud.io>

src/lib.rs

@@ -1,13 +1,15 @@
 pub mod brw_stats;
 pub mod jobstats;
 pub mod lnet;
+pub mod service;
 pub mod stats;
 
 use brw_stats::build_target_stats;
 use lnet::build_lnet_stats;
 use lustre_collector::{LNetStat, LNetStatGlobal, Record, TargetStat, TargetVariant};
 use num_traits::Num;
 use prometheus_exporter_base::{prelude::*, Yes};
+use service::build_service_stats;
 use std::{collections::BTreeMap, fmt, ops::Deref, time::Duration};
 
 #[derive(Debug, Clone, Copy)]
@@ -123,6 +125,9 @@ pub fn build_lustre_stats(output: Vec<Record>, time: Duration) -> String {
             lustre_collector::Record::Target(x) => {
                 build_target_stats(x, &mut stats_map, time);
             }
+            lustre_collector::Record::LustreService(x) => {
+                build_service_stats(x, &mut stats_map, time);
+            }
         }
     }
 

src/service.rs

@@ -0,0 +1,49 @@
+use crate::{Metric, StatsMapExt};
+use lustre_collector::LustreServiceStats;
+use prometheus_exporter_base::prelude::*;
+use std::{collections::BTreeMap, ops::Deref, time::Duration};
+
+static LDLM_CANCELD_STATS_SAMPLES: Metric = Metric {
+    name: "lustre_ldlm_canceld_stats",
+    help: "Gives information about LDLM Canceld service.",
+    r#type: MetricType::Counter,
+};
+
+static LDLM_CBD_STATS_SAMPLES: Metric = Metric {
+    name: "lustre_ldlm_cbd_stats",
+    help: "Gives information about LDLM Callback service.",
+    r#type: MetricType::Counter,
+};
+
+pub fn build_service_stats(
+    x: LustreServiceStats,
+    stats_map: &mut BTreeMap<&'static str, PrometheusMetric<'static>>,
+    time: Duration,
+) {
+    match x {
+        LustreServiceStats::LdlmCanceld(xs) => {
+            for s in xs {
+                stats_map
+                    .get_mut_metric(LDLM_CANCELD_STATS_SAMPLES)
+                    .render_and_append_instance(
+                        &PrometheusInstance::new()
+                            .with_label("operation", s.name.deref())
+                            .with_value(s.samples)
+                            .with_timestamp(time.as_millis()),
+                    );
+            }
+        }
+        LustreServiceStats::LdlmCbd(xs) => {
+            for s in xs {
+                stats_map
+                    .get_mut_metric(LDLM_CBD_STATS_SAMPLES)
+                    .render_and_append_instance(
+                        &PrometheusInstance::new()
+                            .with_label("operation", s.name.deref())
+                            .with_value(s.samples)
+                            .with_timestamp(time.as_millis()),
+                    );
+            }
+        }
+    };
+}