rapl-energy 0.2.2

Small library for getting the CPU energy consumption from RAPL
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use std::fs::{File, OpenOptions};
use std::io::{self, Read, Seek, SeekFrom};
use std::str::FromStr;
use std::sync::{Mutex, MutexGuard};

#[derive(Debug)]
pub struct FileHandle {
    handle: Mutex<File>,
}

impl FileHandle {
    pub fn new(path: &str, write: bool) -> io::Result<Self> {
        let file = OpenOptions::new().read(true).write(write).open(path)?;
        Ok(Self { handle: Mutex::new(file) })
    }

    pub fn read<T: FromStr>(&self) -> T where T::Err: std::fmt::Debug {
        // Open file
        let mut file = self.open(0);
        // Read content
        let mut buf = String::new();
        file.read_to_string(&mut buf).unwrap();
        // Parse buffer
        let buf = buf.trim();
        buf.parse::<T>().expect(&format!("Could not parse {}", buf))
    }

    fn open(&self, offset: u64) -> MutexGuard<'_, File> {
        let mut file = self.handle.lock().unwrap();
        file.seek(SeekFrom::Start(offset)).unwrap();
        file
    }
}