feat(http): support websocket server

cloudwego · Aug 7, 2024 · 3800ea5 · 3800ea5
1 parent 62cf986
commit 3800ea5
Showing 1 changed file with 11 additions and 14 deletions.
diff --git a/volo-http/src/server/utils/ws.rs b/volo-http/src/server/utils/ws.rs
@@ -624,22 +624,19 @@ mod websocket_tests {
             ws.set_config(
                 WebSocketConfig::new().set_protocols(["graphql-ws", "graphql-transport-ws"]),
             )
-            .on_protocol([(
-                "graphql-ws",
-                |mut socket: WebSocket| async move {
-                    while let Some(Ok(msg)) = socket.next().await {
-                        match msg {
-                            Message::Text(text) => {
-                                socket
-                                    .send(Message::Text(text.add("-graphql-ws")))
-                                    .await
-                                    .unwrap();
-                            }
-                            _ => {}
+            .on_protocol([("graphql-ws", |mut socket: WebSocket| async move {
+                while let Some(Ok(msg)) = socket.next().await {
+                    match msg {
+                        Message::Text(text) => {
+                            socket
+                                .send(Message::Text(text.add("-graphql-ws")))
+                                .await
+                                .unwrap();
                         }
+                        _ => {}
                     }
-                },
-            )])
+                }
+            })])
             .on_upgrade(|_| async {})
         }