parse /proc/meminfo for memory usage (#7)

* fix: calculate memory usage via /proc/meminfo

* refactor: move sysinfo call into get_current

No longer used by anything but uptime.
This commit is contained in:
Ryan 2024-08-15 16:14:27 +08:00 committed by GitHub
parent 4d509839f5
commit 1cf4f754ac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 26 additions and 17 deletions

View file

@ -9,23 +9,20 @@ use crate::desktop::get_desktop_info;
use crate::release::{get_os_pretty_name, get_system_info}; use crate::release::{get_os_pretty_name, get_system_info};
use crate::system::{get_memory_usage, get_root_disk_usage, get_shell, get_username_and_hostname}; use crate::system::{get_memory_usage, get_root_disk_usage, get_shell, get_username_and_hostname};
use crate::uptime::get_current; use crate::uptime::get_current;
use std::io;
use color_eyre::Report; use color_eyre::Report;
use nix::sys::sysinfo::sysinfo;
fn main() -> Result<(), Report> { fn main() -> Result<(), Report> {
color_eyre::install()?; color_eyre::install()?;
let info = sysinfo().map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
let fields = Fields { let fields = Fields {
user_info: get_username_and_hostname(), user_info: get_username_and_hostname(),
os_name: get_os_pretty_name()?, os_name: get_os_pretty_name()?,
kernel_version: get_system_info()?, kernel_version: get_system_info()?,
shell: get_shell(), shell: get_shell(),
desktop: get_desktop_info(), desktop: get_desktop_info(),
uptime: get_current(&info)?, uptime: get_current()?,
memory_usage: get_memory_usage(&info), memory_usage: get_memory_usage()?,
storage: get_root_disk_usage()?, storage: get_root_disk_usage()?,
colors: print_dots(), colors: print_dots(),
}; };

View file

@ -1,6 +1,5 @@
use color_eyre::Result; use color_eyre::Result;
use nix::sys::statvfs::statvfs; use nix::sys::statvfs::statvfs;
use nix::sys::sysinfo::SysInfo;
use std::env; use std::env;
use std::io::{self}; use std::io::{self};
@ -40,21 +39,33 @@ pub fn get_root_disk_usage() -> Result<String, io::Error> {
)) ))
} }
pub fn get_memory_usage(info: &SysInfo) -> String { pub fn get_memory_usage() -> Result<String, io::Error> {
#[inline(always)] #[inline(always)]
fn parse_memory_info(info: &SysInfo) -> (f64, f64) { fn parse_memory_info() -> Result<(f64, f64), io::Error> {
let total_memory_kb = (info.ram_total() / 1024) as f64; let mut total_memory_kb = 0.0;
let available_memory_kb = (info.ram_unused() / 1024) as f64; let mut available_memory_kb = 0.0;
let total_memory_gb = total_memory_kb / (1024.0 * 1024.0); for line in std::fs::read_to_string("/proc/meminfo")?.lines() {
let available_memory_gb = available_memory_kb / (1024.0 * 1024.0); let mut split = line.split_whitespace();
let key = split.next().unwrap_or("");
if key == "MemTotal:" {
total_memory_kb = split.next().unwrap_or("0").parse().unwrap_or(0.0);
} else if key == "MemAvailable:" {
available_memory_kb = split.next().unwrap_or("0").parse().unwrap_or(0.0);
}
}
let total_memory_gb = total_memory_kb / 1024.0 / 1024.0;
let available_memory_gb = available_memory_kb / 1024.0 / 1024.0;
let used_memory_gb = total_memory_gb - available_memory_gb; let used_memory_gb = total_memory_gb - available_memory_gb;
(used_memory_gb, total_memory_gb) Ok((used_memory_gb, total_memory_gb))
} }
let (used_memory, total_memory) = parse_memory_info(info); let (used_memory, total_memory) = parse_memory_info()?;
let percentage_used = (used_memory / total_memory * 100.0).round() as u64; let percentage_used = (used_memory / total_memory * 100.0).round() as u64;
format!("{used_memory:.2} GiB / {total_memory:.2} GiB ({CYAN}{percentage_used}%{RESET})") Ok(format!(
"{used_memory:.2} GiB / {total_memory:.2} GiB ({CYAN}{percentage_used}%{RESET})"
))
} }

View file

@ -1,8 +1,9 @@
use color_eyre::Result; use color_eyre::Result;
use nix::sys::sysinfo::SysInfo; use nix::sys::sysinfo::sysinfo;
use std::io; use std::io;
pub fn get_current(info: &SysInfo) -> Result<String, io::Error> { pub fn get_current() -> Result<String, io::Error> {
let info = sysinfo().map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
let uptime_seconds = info.uptime().as_secs_f64(); let uptime_seconds = info.uptime().as_secs_f64();
let total_minutes = (uptime_seconds / 60.0).round() as u64; let total_minutes = (uptime_seconds / 60.0).round() as u64;