Fix failing build after merge (#125)

Co-authored-by: Benoît CORTIER <benoit.cortier@fried-world.eu>
Devolutions · Dec 9, 2020 · 8f610c4 · 8f610c4
1 parent 27b0821
commit 8f610c4
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/websocket_client.rs b/src/websocket_client.rs
@@ -310,18 +310,18 @@ fn process_req(req: &Request<Body>) -> Response<Body> {
             .unwrap();
     };
 
-    let mut builder = Response::builder();
-
-    builder
+    let builder = Response::builder()
         .status(StatusCode::SWITCHING_PROTOCOLS)
         .header(header::UPGRADE, "websocket")
         .header(header::CONNECTION, "upgrade")
         .header(header::SEC_WEBSOCKET_ACCEPT, key.as_str());
 
     // Add the SEC_WEBSOCKET_PROTOCOL header only if it was in the request, otherwise, IIS doesn't like it
-    if let Some(websocket_protocol) = req.headers().get(header::SEC_WEBSOCKET_PROTOCOL) {
-        builder.header(header::SEC_WEBSOCKET_PROTOCOL, websocket_protocol);
-    }
+    let builder = if let Some(websocket_protocol) = req.headers().get(header::SEC_WEBSOCKET_PROTOCOL) {
+        builder.header(header::SEC_WEBSOCKET_PROTOCOL, websocket_protocol)
+    } else {
+        builder
+    };
 
     builder.body(Body::empty()).unwrap()
 }