Added simple multithreading

src/main.rs

@@ -5,6 +5,8 @@ use std::{
     io::{prelude::*, BufReader, Error, ErrorKind, Result},
     net::{TcpListener, TcpStream},
     path::Path,
+    sync::{Arc, Mutex},
+    thread,
 };
 
 use data_stream::{
@@ -43,12 +45,13 @@ fn fail(mut stream: TcpStream) {
     let _ = writeln!(stream, "Page not found!");
 }
 
+#[derive(Clone)]
 struct Comment {
     name: Box<str>,
     text: Box<str>,
 }
 
-#[derive(Default)]
+#[derive(Default, Clone)]
 struct SiteInfo {
     comments: Vec<Comment>,
     visits: usize,
@@ -111,7 +114,7 @@ impl<S: SizeSettings> FromStream<S> for SiteInfo {
 
 #[derive(Default)]
 struct Context {
-    infos: HashMap<Box<str>, SiteInfo>,
+    infos: HashMap<Box<str>, Arc<Mutex<SiteInfo>>>,
 }
 
 impl Context {
@@ -192,17 +195,21 @@ impl Context {
 
         use Entry::*;
         let info = match self.infos.entry(relative_path.clone().into_boxed_str()) {
-            Occupied(o) => o.into_mut(),
-            Vacant(v) => v.insert(
+            Occupied(o) => o.get().clone(),
+            Vacant(v) => v
+                .insert(Arc::new(Mutex::new(
                     File::open(&data_path)
                         .map(|mut file| {
                             from_stream::<PortableSettings, _, _>(&mut file).unwrap_or_default()
                         })
                         .unwrap_or_default(),
-            ),
+                )))
+                .clone(),
         };
 
-        info.handle_connection(
+        thread::spawn(move || {
+            handle_relative_connection(
+                info,
                 stream,
                 &request.body,
                 &relative_path,
@@ -212,12 +219,12 @@ impl Context {
                 &data_path,
                 start_level,
             )
+        });
     }
 }
 
-impl SiteInfo {
-    fn handle_connection(
-        &mut self,
+fn handle_relative_connection(
+    info: Arc<Mutex<SiteInfo>>,
     mut stream: TcpStream,
     body: &str,
     relative_path: &str,
@@ -226,10 +233,13 @@ impl SiteInfo {
     pki_path: Option<&Path>,
     data_path: &Path,
     start_level: usize,
-    ) {
+) {
     let mut name = None;
     let mut text = None;
 
+    let mut up = false;
+    let mut down = false;
+
     for entry in body.split('&') {
         let Some((key, input)) = entry.split_once('=') else {
             continue;
@@ -263,20 +273,32 @@ impl SiteInfo {
                         .into(),
                 );
             }
-                "up" => self.up += 1,
-                "down" => self.down += 1,
+            "up" => up = true,
+            "down" => down = true,
             _ => (),
         }
     }
 
+    let info = if let Ok(mut info) = info.lock() {
         if let (Some(name), Some(text)) = (name, text) {
-            self.comments.push(Comment { name, text });
+            info.comments.push(Comment { name, text });
+        }
+        if up {
+            info.up += 1;
+        }
+        if down {
+            info.down += 1;
         }
 
-        self.visits += 1;
+        info.visits += 1;
+
+        info.clone()
+    } else {
+        return;
+    };
 
     if let Ok(mut file) = File::create(data_path) {
-            if to_stream::<PortableSettings, _, _>(self, &mut file).is_err() {
+        if to_stream::<PortableSettings, _, _>(&info, &mut file).is_err() {
             eprintln!("Error saving data!");
             eprintln!();
         }
@@ -289,7 +311,7 @@ impl SiteInfo {
     let _ = writeln!(
         stream,
         "<p>👁️{} 💖️{} 💔️{}</p>",
-            self.visits, self.up, self.down
+        info.visits, info.up, info.down
     );
 
     let title = relative_path
@@ -386,11 +408,10 @@ impl SiteInfo {
     };
     let _ = stream.write_all(html.into_string().as_bytes());
 
-        for Comment { name, text } in &self.comments {
+    for Comment { name, text } in &info.comments {
         let _ = writeln!(stream, "<fieldset><legend>{name}</legend>{text}</fieldset>");
     }
 
     let _ = writeln!(stream, "<hr>");
     let _ = writeln!(stream, "<a href=\"/{parent_path}\">&lt;&lt; Back</a>");
-    }
 }