templog: rust/src/main.rs comparison

NotTooOften

comparison

equal deleted inserted replaced

-:3c1d37d78415
+:282fae1c12e4
 extern crate toml;
 extern crate docopt;
 use std::io;
+use std::time::{Duration,Instant};
+use std::ops::Not;
+use std::cell::Cell;
 use tokio_core::reactor::Core;
 use futures::{Stream,Sink,Future};
 use futures::sync::{mpsc};
 use sensor::Sensor;
 }
 run(&config, args.flag_nowait, args.flag_test);
 }
+/// Call closures with a rate limit. Useful for log message ratelimiting
+pub struct NotTooOften {
+last: Cell<Instant>,
+limit: Duration,
+}
+impl NotTooOften {
+pub fn new(limit_secs: u64) -> Self {
+NotTooOften {
+limit: Duration::new(limit_secs, 0),
+last: Cell::new(Instant::now() - Duration::new(limit_secs+1, 0)),
+}
+}
+pub fn and_then<F>(&self, op: F)
+where F: Fn() {
+let now = Instant::now();
+if now - self.last.get() > self.limit {
+self.last.set(now);
+op();
+}
+}
+}

Mercurial > templog