🐛 Add 404 page for missing file instead of gateway timeout

src/bin/server.rs

@@ -149,6 +149,30 @@ async fn serve_file (
 	}
 }
 
+async fn serve_error (
+	client: Arc <Client>, 
+	req_id: String,
+	status_code: ptth::http_serde::StatusCode
+) {
+	let headers: HashMap <String, Vec <u8>> = Default::default ();
+	
+	let resp_parts = http_serde::ResponseParts {
+		status_code,
+		headers,
+	};
+	
+	let resp_req = client
+		.post (&format! ("http://127.0.0.1:4000/http_response/{}", req_id))
+		.header ("X-PTTH-2LJYXWC4", base64::encode (rmp_serde::to_vec (&resp_parts).unwrap ()))
+		.body ("404 Not Found")
+	;
+	
+	//println! ("Step 6");
+	if let Err (e) = resp_req.send ().await {
+		println! ("Err: {:?}", e);
+	}
+}
+
 #[tokio::main]
 async fn main () -> Result <(), Box <dyn Error>> {
 	let client = Arc::new (Client::new ());
@@ -221,16 +245,20 @@ async fn main () -> Result <(), Box <dyn Error>> {
 		tokio::spawn (async move {
 			let mut path = PathBuf::from ("/home/user");
 			path.push (&uri [1..]);
-			let f = File::open (path).await.unwrap ();
 			
-			serve_file (
+			match File::open (path).await {
+				Ok (f) => serve_file (
 					client,
 					f,
 					req_id,
 					should_send_body,
 					range_start,
 					range_end
-			).await;
+				).await,
+				Err (_) => {
+					serve_error (client, req_id, ptth::http_serde::StatusCode::NotFound).await;
+				},
+			}
 		});
 	}
 }