diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/main.rs | 251 |
1 files changed, 142 insertions, 109 deletions
diff --git a/src/main.rs b/src/main.rs index 956285b..60c02e4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,127 +1,143 @@ #[macro_use] extern crate clap; -extern crate num; +extern crate vec_map; +extern crate libc; -use std::path::{Path, PathBuf}; -use std::fs::File; -use std::io::{self, Read}; -use std::{iter, thread, ops}; -use std::time::Duration; - -use clap::{Arg, App, AppSettings, SubCommand}; - -use num::NumCast; - -const SYS_PRESENT: &'static str = "/sys/devices/system/cpu/present"; -static FORMAT: [char; 9] = [' ', '▁', '▂', '▃', '▄', '▅', '▆', '▇', '█']; - -const WINDOW_SIZE: usize = 32; - -#[derive(Debug, Default)] -struct Window<T> { - data: Vec<T>, - idx: usize, - size: usize -} - -impl<T> Window<T> where T: for<'a> iter::Sum<&'a T> + ops::Div<Output=T> + NumCast { - fn new(size: usize) -> Window<T> { - Window { data: Vec::with_capacity(size), idx: 0, size: size } - } - - fn sample(&mut self, value: T) { - if self.data.len() < self.size { self.data.push(value) } - else { self.data[self.idx] = value; } - self.idx = (self.idx + 1) % self.size; - } - - fn elements(&self) -> (&[T], &[T]) { (&self.data[self.idx..], &self.data[..self.idx]) } - fn average(&self) -> T { self.data.iter().sum::<T>() / T::from(self.data.len()).unwrap() } -} - -#[derive(Debug, Default)] -struct Core { - index: u32, - min_freq: u32, - max_freq: u32, - cur_freq: Window<u32> -} +#[macro_use] +extern crate log; +extern crate env_logger; +use std::fs::File; +use std::io::{self, BufRead, BufReader}; +use std::thread; +use std::time::{Instant, Duration}; + +use clap::{Arg, App}; + +use vec_map::VecMap; + +const PROC_STAT: &'static str = "/proc/stat"; +static FORMAT: [char; 8] = ['▁', '▂', '▃', '▄', '▅', '▆', '▇', '█']; + +/// cpu 3357 0 4313 1362393 +/// The amount of time, measured in units of USER_HZ (1/100ths of a +/// second on most architectures, use sysconf(_SC_CLK_TCK) to obtain +/// the right value), that the system spent in various states: +/// +/// user (1) Time spent in user mode. +/// +/// nice (2) Time spent in user mode with low priority (nice). +/// +/// system (3) Time spent in system mode. +/// +/// idle (4) Time spent in the idle task. This value should be +/// USER_HZ times the second entry in the /proc/uptime pseudo- +/// file. +/// +/// iowait (since Linux 2.5.41) +/// (5) Time waiting for I/O to complete. +/// +/// irq (since Linux 2.6.0-test4) +/// (6) Time servicing interrupts. +/// +/// softirq (since Linux 2.6.0-test4) +/// (7) Time servicing softirqs. +/// +/// steal (since Linux 2.6.11) +/// (8) Stolen time, which is the time spent in other operating +/// systems when running in a virtualized environment +/// +/// guest (since Linux 2.6.24) +/// (9) Time spent running a virtual CPU for guest operating +/// systems under the control of the Linux kernel. +/// +/// guest_nice (since Linux 2.6.33) +/// (10) Time spent running a niced guest (virtual CPU for +/// guest operating systems under the control of the Linux ker‐ +/// nel). #[derive(Debug)] -struct System { - cores: Vec<Core>, - load: Window<f32> +struct Stat { + time: Instant, + total: Option<CPU>, + cores: VecMap<CPU> } -impl System { - fn new() -> System { - System { - cores: init_cores(), - load: Window::new(8) +impl Stat { + pub fn read() -> io::Result<Stat> { + let file = try!(File::open(PROC_STAT)); + let reader = BufReader::new(file); + let mut stat = Stat { time: Instant::now(), total: None, cores: VecMap::new() }; + + for line in reader.lines() { + let line = try!(line); + const OFFSET: usize = 3; // "cpu".len() + if line.starts_with("cpu ") { + stat.total = Some(CPU::from_line(&line[OFFSET..])); + } else if line.starts_with("cpu") { + let num: u64 = line[OFFSET..].split_whitespace().next().and_then(|s| s.trim().parse().ok()).unwrap(); + stat.cores.insert(num as usize, CPU::from_line(&line[OFFSET..])); + } } - } - // Direct printing to avoid allocation - fn print_cores(&self) { - for c in &self.cores { - let load = (c.cur_freq.average() - c.min_freq) as f32 / (c.max_freq - c.min_freq) as f32; - print!("{}", FORMAT[(FORMAT.len() as f32 * load) as usize]); - } - println!(""); + Ok(stat) } - fn print_system(&self) { - let (first, last) = self.load.elements(); - for load in first.iter().chain(last.iter()) { - print!("{}", FORMAT[(FORMAT.len() as f32 * load) as usize]); + pub fn load_since(&self, earlier: &Stat) -> Load { + Load { + duration: self.time.duration_since(earlier.time), + total: match (&self.total, &earlier.total) { + (&Some(ref now), &Some(ref old)) => Some(now.diff(old)), + _ => None + }, + cores: self.cores.iter() + .flat_map(|(idx, core)| earlier.cores.get(idx).map(|ec| (idx, core.diff(ec)))) + .collect() } - println!(""); } } -fn read_into_string<P: AsRef<Path>>(path: P) -> io::Result<String> { - let mut output = String::new(); - try!(try!(File::open(path)).read_to_string(&mut output)); - Ok(output) +#[derive(Debug, Clone)] +struct Load { + duration: Duration, + total: Option<CPU>, + cores: VecMap<CPU> } -fn init_cores() -> Vec<Core> { - let present = read_into_string(SYS_PRESENT).expect("Can't read kernel interface to query present cores"); - - // a-b - let mut parts = present.split('-').flat_map(|s| s.trim().parse::<u32>().ok()); - let (a, b) = (parts.next().unwrap(), parts.next().unwrap()); - (a..b + 1).map(|idx| { - let cpu_path: PathBuf = format!("/sys/devices/system/cpu/cpu{}/cpufreq", idx).into(); - Core { - index: idx, - min_freq: read_into_string(cpu_path.join("scaling_min_freq")).ok() - .and_then(|s| s.trim().parse().ok()).expect("Can't read min freq"), - max_freq: read_into_string(cpu_path.join("scaling_max_freq")).ok() - .and_then(|s| s.trim().parse().ok()).expect("Can't read max frequency"), - cur_freq: Window::new(WINDOW_SIZE) - } - }).collect() +#[derive(Debug, Clone)] +struct CPU { + user: u64, + nice: u64, + system: u64, + idle: u64 } -fn update(system: &mut System) { - let mut frame_average = 0.; - for c in &mut system.cores { - let cpu_path: PathBuf = format!("/sys/devices/system/cpu/cpu{}/cpufreq/scaling_cur_freq", c.index).into(); - let cur_freq = read_into_string(cpu_path).ok() - .and_then(|s| s.trim().parse().ok()).expect("Can't read current frequency"); - c.cur_freq.sample(cur_freq); - frame_average += (cur_freq - c.min_freq) as f32 / (c.max_freq - c.min_freq) as f32; +impl CPU { + pub fn from_line(line: &str) -> CPU { + fn parse(s: Option<&str>) -> u64 { s.and_then(|s| s.trim().parse().ok()).expect("Couldn't parse CPU stat") } + let mut tok = line.split_whitespace(); + CPU { + user: parse(tok.next()), + nice: parse(tok.next()), + system: parse(tok.next()), + idle: parse(tok.next()) + } + } + + pub fn diff(&self, other: &CPU) -> CPU { + CPU { + user: self.user - other.user, + nice: self.nice - other.nice, + system: self.system - other.system, + idle: self.idle - other.idle + } } - system.load.sample(frame_average / system.cores.len() as f32); } fn main() { + env_logger::init().unwrap(); let matches = App::new("cpuline") .version(crate_version!()) .author(crate_authors!()) - .about("Display CPU usage per-core or over time (Linux-only)") - .setting(AppSettings::SubcommandRequired) .arg(Arg::with_name("interval") .global(true) .short("i") @@ -129,23 +145,40 @@ fn main() { .value_name("MS") .takes_value(true) .default_value("1000")) - .subcommand(SubCommand::with_name("cores")) - .subcommand(SubCommand::with_name("time")) .get_matches(); - let action = match matches.subcommand_name() { - Some("cores") => System::print_cores, - Some("time") => System::print_system, - _ => unreachable!() - }; - let interval = value_t!(matches, "interval", u64).unwrap(); - let mut system = System::new(); + let mut stat = None; + let user_hz: u64 = unsafe { libc::sysconf(libc::_SC_CLK_TCK) as u64 }; + debug!("user_hz({})", user_hz); + loop { - update(&mut system); - action(&system); + let old = stat; + stat = Stat::read().ok(); + + match (&stat, &old) { + (&Some(ref now), &Some(ref old)) => { + let load = now.load_since(&old); + let duration_ticks = load.duration.as_secs() * user_hz + + (load.duration.subsec_nanos() as f64 / 1E9 * user_hz as f64) as u64; + + for (_, core) in load.cores.iter() { + // How many ticks the core was in use + let used = core.user + core.nice + core.system; + // How long this core was used with 0 (not used) to 1 (fully used) + let used_part = used as f32 / duration_ticks as f32; + + let output = FORMAT[((FORMAT.len() - 1) as f32 * used_part) as usize]; + debug!("used_part({}) = used({}) / duration_ticks({}) => '{}'", used_part, used, duration_ticks, output); + print!("{}", output); + } + println!(""); + }, + _ => () + } + thread::sleep(Duration::from_millis(interval)); } } |