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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
use crate::config::CONFIG;
use crate::types::ThreadsData;
use std::fs::read_to_string;
/*
mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl
thanks for htop's developer on stackoverflow for providing this algorithm to
calculate used memory.
*/
pub fn get_memory() -> ThreadsData {
let buf = match read_to_string("/proc/meminfo") {
Ok(data) => data,
_ => return ThreadsData::Memory(String::from("Error Reading memory!")),
};
let mut mem_total: u32 = 0;
let mut shmem: u32 = 0;
let mut mem_free: u32 = 0;
let mut mem_buffers: u32 = 0;
let mut mem_cached: u32 = 0;
let mut mem_srecl: u32 = 0;
for line in buf.lines() {
if mem_total > 0
&& shmem > 0
&& mem_free > 0
&& mem_buffers > 0
&& mem_cached > 0
&& mem_srecl > 0
{
break;
}
if line.starts_with("MemTotal") {
assign_val(line, &mut mem_total);
}
if line.starts_with("SReclaimable") {
assign_val(line, &mut mem_srecl)
}
if line.starts_with("Cached") {
assign_val(line, &mut mem_cached)
}
if line.starts_with("Shmem") {
assign_val(line, &mut shmem);
}
if line.starts_with("MemFree") {
assign_val(line, &mut mem_free);
}
if line.starts_with("Buffers") {
assign_val(line, &mut mem_buffers);
}
}
let mem_used = (mem_total + shmem - mem_free - mem_buffers - mem_cached - mem_srecl) / 1024;
let result: String;
if mem_used > 1000 {
result = format!(
" {} {:.1}G {}",
CONFIG.memory.icon,
mem_used as f32 / 1000.0,
CONFIG.seperator
);
} else {
result = format!(
" {} {}M {}",
CONFIG.memory.icon, mem_used, CONFIG.seperator
);
}
ThreadsData::Memory(result)
}
/*
this helper function will split the line(first argument) by the character(:)
and then parse the right splited item as u32
then assign that to the "assignable"(2nd argument).
*/
fn assign_val(line: &str, assignable: &mut u32) {
let parsed: u32 = line.split(':').collect::<Vec<&str>>()[1]
.trim()
.split(' ')
.collect::<Vec<&str>>()[0]
.parse()
.unwrap();
*assignable = parsed;
}
|
