+use std::sync::atomic::Ordering;
use std::collections::LinkedList;
use std::sync::{Arc, Mutex};
use std::io::Write;
use crate::datastore::{Store, AddressState, U64Setting, RegexSetting};
+use crate::START_SHUTDOWN;
+
pub enum Stat {
HeaderCount(u64),
NewConnection,
let mut out = stdout.lock();
let stats = thread_arc.lock().unwrap();
+ if START_SHUTDOWN.load(Ordering::Relaxed) && stats.connection_count == 0 {
+ break;
+ }
out.write_all(b"\x1b[2J\x1b[;H\n").expect("stdout broken?");
for line in stats.lines.iter() {
}
}
- pub fn add_line(&self, line: String, _err: bool) {
+ pub fn add_line(&self, line: String, err: bool) {
let mut stats = self.stats.lock().unwrap();
- stats.lines.push_back(line);
+ if err {
+ stats.lines.push_back("\x1b[31m".to_string() + &line + "\x1b[0m");
+ } else {
+ stats.lines.push_back(line);
+ }
if stats.lines.len() > 50 {
stats.lines.pop_front();
}