add channel support 10

siteassist-benjamin · Mar 5, 2024 · 44fd3b8 · 44fd3b8
1 parent 6011102
commit 44fd3b8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -30,8 +30,7 @@ fn handle_fanout_ws(mut req: Request, chan: &str) -> Response {
     resp
 }
 
-fn handle_fanout(req: Request) -> Response {
-    let chan = req.get_header_str("FANOUT").unwrap_or("test");
+fn handle_fanout(req: Request, chan: &str) -> Response {
     match req.get_url().path() {
         "/stream/long-poll" => fanout_util::grip_response("text/plain", "response", chan),
         "/stream/plain" => fanout_util::grip_response("text/plain", "stream", chan),
@@ -54,7 +53,9 @@ fn main() -> Result<(), Error> {
         return Ok(if req.get_header_str("Grip-Sig").is_some() {
             // Request is from Fanout
 
-            handle_fanout(req).send_to_client();
+            let channel = &req.get_header_str("FANOUT").unwrap_or("test");
+
+            handle_fanout(req, channel).send_to_client();
         } else {
             // Not from fanout, hand it off to Fanout to manage
             req.with_header("Access-Control-Allow-Origin","*").handoff_fanout("self")?