Add benchmark mod

Author: fangpenlin

src/benchmarks.rs

@@ -0,0 +1 @@
+pub mod storage;

src/benchmarks/storage.rs

@@ -0,0 +1,64 @@
+use std::cmp::{max, min};
+use std::fs::File;
+use std::io::Write;
+use std::time::{Duration, SystemTime};
+
+#[derive(Debug)]
+pub struct StorageBenchmark {
+    pub path: String,
+    pub file_size: usize,
+    pub chunk_size: usize,
+    pub loop_times: usize,
+}
+
+impl StorageBenchmark {
+    pub fn run_once(&self) -> anyhow::Result<Duration> {
+        log::info!("Running write test with config={self:#?}");
+        let buf: Vec<u8> = vec![1; self.chunk_size];
+        let mut remaining_size = self.file_size;
+
+        let start = SystemTime::now();
+        let mut file = File::open(&self.path)?;
+        while true {
+            let chunk_size = max(self.chunk_size, remaining_size);
+            let written = file.write(&buf[..chunk_size])?;
+            remaining_size -= written;
+        }
+        file.sync_all();
+        let end = SystemTime::now();
+        let duration = end.duration_since(start)?;
+        log::info!("Finish write test with total time {duration}");
+        log::info!(
+            "throughput={:,} MB/s",
+            ((self.file_size as f64) / (1024.0 * 1024.0)) / (duration.as_secs() as f64)
+        );
+        Ok(duration)
+    }
+
+    pub fn run(&self) -> anyhow::Result<()> {
+        let mut durations = Vec::<Duration>::with_capacity(self.loop_times);
+        for idx in (0..self.loop_times) {
+            log::info!(
+                "Running write test for {} out of {} time",
+                idx + 1,
+                self.loop_times
+            );
+            let duration = self.run_once()?;
+            durations.push(duration);
+        }
+
+        let total = durations.iter().sum();
+        log::info!(
+            "Finish {} write test with total time {}",
+            self.loop_times,
+            total
+        );
+        log::info!(
+            "avg throughput={:,} MB/s",
+            ((self.file_size as f64) * self.loop_times / (1024.0 * 1024.0))
+                / (total.as_secs() as f64)
+        );
+
+        Ok(())
+    }
+}

src/main.rs

@@ -3,6 +3,7 @@ mod config;
 mod storage;
 mod usb;
 mod wifi;
+mod benchmarks;
 
 use crate::api::processor::{process_events, DeviceInfo, DeviceInfoProducer};
 use crate::api::websocket::{ConnectionState, SessionEvent, WebSocketSession};