refactor(router): Extract logic for computing restricted headers

V-Wong · Jan 4, 2025 · bc0132f · bc0132f
1 parent f684201
commit bc0132f
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/router.rs b/src/router.rs
@@ -47,17 +47,22 @@ impl Router {
 
             self.thread_pool.exec(move || {
                 let mut response = handler(&request);
-                response.headers.add(
-                    "Content-Length",
-                    &response
-                        .body
-                        .as_ref()
-                        .map(|body| body.len())
-                        .unwrap_or(0)
-                        .to_string(),
-                );
+                compute_restricted_headers(&mut response);
                 stream.write_all(String::from(response).as_bytes()).unwrap();
             });
         }
     }
 }
+
+fn compute_restricted_headers(response: &mut response::Response) -> &response::Response {
+    response.headers.add(
+        "Content-Length",
+        &response
+            .body
+            .as_ref()
+            .map(|body| body.len())
+            .unwrap_or(0)
+            .to_string(),
+    );
+    response
+}