src/reporter.rs

   1 // Copyright 2021 Ian Jackson and contributors to Hippotat
   2 // SPDX-License-Identifier: GPL-3.0-or-later
   3 // There is NO WARRANTY.
   4
   5 use crate::prelude::*;
   6
   7 pub struct Reporter<'r> {
   8   ic: &'r InstanceConfig,
   9   successes: u64,
  10   last_report: Option<Report>,
  11 }
  12
  13 #[derive(Debug)]
  14 struct Report {
  15   when: Instant,
  16   ok: Result<(),()>,
  17 }
  18
  19 // Reporting strategy
  20 //   - report all errors
  21 //   - report first success after a period of lack of messages
  22 //   - if error, report last success
  23
  24 impl<'r> Reporter<'r> {
  25   pub fn new(ic: &'r InstanceConfig) -> Self { Reporter {
  26     ic,
  27     successes: 0,
  28     last_report: None,
  29   } }
  30
  31   pub fn success(&mut self) {
  32     self.successes += 1;
  33     let now = Instant::now();
  34     if let Some(rep) = &self.last_report {
  35       if now - rep.when < match rep.ok {
  36         Ok(()) => Duration::from_secs(3600),
  37         Err(()) => Duration::from_secs(30),
  38       } {
  39         return
  40       }
  41     }
  42
  43     info!(target:"hippotat", "{} ({}ok): running", self.ic, self.successes);
  44     self.last_report = Some(Report { when: now, ok: Ok(()) });
  45   }
  46
  47   pub fn filter<T>(&mut self, req_num: Option<ReqNum>, r: Result<T,AE>)
  48                    -> Option<T> {
  49     let now = Instant::now();
  50     match r {
  51       Ok(t) => {
  52         Some(t)
  53       },
  54       Err(e) => {
  55         let m = (||{
  56           let mut m = self.ic.to_string();
  57           if let Some(req_num) = req_num {
  58             write!(m, " #{}", req_num)?;
  59           }
  60           if self.successes > 0 {
  61             write!(m, " ({}ok)", self.successes)?;
  62             self.successes = 0;
  63           }
  64           write!(m, ": {:?}", e)?;
  65           Ok::<_,fmt::Error>(m)
  66         })().unwrap();
  67         warn!(target:"hippotat", "{}", m);
  68         self.last_report = Some(Report { when: now, ok: Err(()) });
  69         None
  70       },
  71     }
  72   }
  73 }