Line data    Source code

       1              : //! process metrics that the [`::prometheus`] crate doesn't provide.
       2              : 
       3              : // This module has heavy inspiration from the prometheus crate's `process_collector.rs`.
       4              : 
       5              : use crate::UIntGauge;
       6              : 
       7              : pub struct Collector {
       8              :     descs: Vec<prometheus::core::Desc>,
       9              :     vmlck: crate::UIntGauge,
      10              : }
      11              : 
      12              : const NMETRICS: usize = 1;
      13              : 
      14              : impl prometheus::core::Collector for Collector {
      15          625 :     fn desc(&self) -> Vec<&prometheus::core::Desc> {
      16          625 :         self.descs.iter().collect()
      17          625 :     }
      18              : 
      19         1259 :     fn collect(&self) -> Vec<prometheus::proto::MetricFamily> {
      20         1259 :         let Ok(myself) = procfs::process::Process::myself() else {
      21            0 :             return vec![];
      22              :         };
      23         1259 :         let mut mfs = Vec::with_capacity(NMETRICS);
      24         1259 :         if let Ok(status) = myself.status() {
      25         1259 :             if let Some(vmlck) = status.vmlck {
      26         1259 :                 self.vmlck.set(vmlck);
      27         1259 :                 mfs.extend(self.vmlck.collect())
      28            0 :             }
      29            0 :         }
      30         1259 :         mfs
      31         1259 :     }
      32              : }
      33              : 
      34              : impl Collector {
      35          625 :     pub fn new() -> Self {
      36          625 :         let mut descs = Vec::new();
      37          625 : 
      38          625 :         let vmlck =
      39          625 :             UIntGauge::new("libmetrics_process_status_vmlck", "/proc/self/status vmlck").unwrap();
      40          625 :         descs.extend(
      41          625 :             prometheus::core::Collector::desc(&vmlck)
      42          625 :                 .into_iter()
      43          625 :                 .cloned(),
      44          625 :         );
      45          625 : 
      46          625 :         Self { descs, vmlck }
      47          625 :     }
      48              : }
      49              : 
      50              : impl Default for Collector {
      51            0 :     fn default() -> Self {
      52            0 :         Self::new()
      53            0 :     }
      54              : }