Make CPU graph better

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 5ba279f..d4faf9f 100755
--- a/src/main.rs
+++ b/src/main.rs
@@ -17,7 +17,7 @@ use std::{
 	os::unix::fs::MetadataExt,
 	str::FromStr,
 	sync::{
-		atomic::{AtomicUsize, Ordering},
+		atomic::{AtomicU8, AtomicUsize, Ordering},
 		Arc,
 	},
 	time::Duration,
@@ -55,7 +55,10 @@ pub struct AwakeState {
 	pub do_statistics: bool,
 	pub database: Arc<Database>,
 	pub cache_path: Utf8PathBuf,
+	/// kbps
 	pub netinfo_upper_bound: Arc<AtomicUsize>,
+	/// whole digit % cpu usage
+	pub cpuinfo_upper_bound: Arc<AtomicUsize>,
 }
 
 #[tokio::main]
@@ -97,6 +100,7 @@ async fn main() {
 		database: Arc::new(database),
 		cache_path: cache.into(),
 		netinfo_upper_bound: Arc::new(AtomicUsize::new(256)),
+		cpuinfo_upper_bound: Arc::new(AtomicUsize::new(100)),
 	};
 
 	match std::env::args().nth(1).as_deref() {
@@ -437,6 +441,9 @@ fn template_content(state: AwakeState, frontmatter: &Frontmatter, marked: String
 
 		let netinfo_upper = state.netinfo_upper_bound.load(Ordering::Relaxed);
 		doc.set("stats.net.max_bound", netinfo_upper);
+
+		let cpuinfo_upper = state.cpuinfo_upper_bound.load(Ordering::Relaxed);
+		doc.set("stats.cpu.max_bound", cpuinfo_upper);
 	}
 
 	doc.compile()