Refactored

Author: buraksenyurt

router/src/handlers.rs

@@ -0,0 +1,39 @@
+use std::{fs::OpenOptions, io::Write};
+
+use crate::request::Request;
+
+pub trait RouteHandler {
+    fn handle(&self, request: &Request) -> anyhow::Result<()>;
+}
+
+pub struct FileStorageHandler {
+    pub file_name: String,
+}
+
+impl RouteHandler for FileStorageHandler {
+    fn handle(&self, request: &Request) -> anyhow::Result<()> {
+        let mut file = OpenOptions::new()
+            .append(true)
+            .create(true)
+            .open(self.file_name.clone())?;
+        file.write_all(&request.data)?;
+        Ok(())
+    }
+}
+
+pub struct PassToRemoteHandler {
+    pub target_uri: String,
+}
+
+impl RouteHandler for PassToRemoteHandler {
+    fn handle(&self, request: &Request) -> anyhow::Result<()> {
+        println!(
+            "Pretending to POST to {}{} with {} bytes",
+            self.target_uri,
+            request.path,
+            request.data.len()
+        );
+
+        Ok(())
+    }
+}

router/src/main.rs

@@ -1,10 +1,16 @@
-use std::{fs::OpenOptions, io::Write};
+mod handlers;
+mod request;
+mod router;
+
+use handlers::*;
+use request::Request;
+use router::Router;
 
 fn main() -> anyhow::Result<()> {
     // File Storage kullanımı
 
     let fs_handler = Box::new(FileStorageHandler {
-        base_path: ".".into(),
+        file_name: "route_datas.log".into(),
     });
 
     let mut router = Router::new(fs_handler);
@@ -29,68 +35,3 @@ fn main() -> anyhow::Result<()> {
 
     Ok(())
 }
-
-pub struct Request {
-    pub path: String,
-    pub data: Vec<u8>,
-}
-
-pub trait RouteHandler {
-    fn handle(&self, request: &Request) -> anyhow::Result<()>;
-}
-
-pub struct FileStorageHandler {
-    pub base_path: String,
-}
-
-impl RouteHandler for FileStorageHandler {
-    fn handle(&self, request: &Request) -> anyhow::Result<()> {
-        let mut file = OpenOptions::new()
-            .append(true)
-            .create(true)
-            .open("request_logs.dat")?;
-        file.write_all(&request.data)?;
-        Ok(())
-    }
-}
-
-pub struct PassToRemoteHandler {
-    pub target_uri: String,
-}
-
-impl RouteHandler for PassToRemoteHandler {
-    fn handle(&self, request: &Request) -> anyhow::Result<()> {
-        println!(
-            "Pretending to POST to {}{} with {} bytes",
-            self.target_uri,
-            request.path,
-            request.data.len()
-        );
-
-        Ok(())
-    }
-}
-
-pub struct Router {
-    requests: Vec<Request>,
-    handler: Box<dyn RouteHandler>,
-}
-
-impl Router {
-    pub fn new(handler: Box<dyn RouteHandler>) -> Self {
-        Router {
-            requests: Vec::new(),
-            handler,
-        }
-    }
-    pub fn add(&mut self, request: Request) {
-        self.requests.push(request);
-    }
-
-    pub fn apply(&self) -> anyhow::Result<()> {
-        for r in self.requests.iter() {
-            self.handler.handle(&r)?;
-        }
-        Ok(())
-    }
-}

router/src/request.rs

@@ -0,0 +1,4 @@
+pub struct Request {
+    pub path: String,
+    pub data: Vec<u8>,
+}

router/src/router.rs

@@ -0,0 +1,25 @@
+use crate::{handlers::RouteHandler, request::Request};
+
+pub struct Router {
+    requests: Vec<Request>,
+    pub handler: Box<dyn RouteHandler>,
+}
+
+impl Router {
+    pub fn new(handler: Box<dyn RouteHandler>) -> Self {
+        Router {
+            requests: Vec::new(),
+            handler,
+        }
+    }
+    pub fn add(&mut self, request: Request) {
+        self.requests.push(request);
+    }
+
+    pub fn apply(&self) -> anyhow::Result<()> {
+        for r in self.requests.iter() {
+            self.handler.handle(r)?;
+        }
+        Ok(())
+    }
+}