替换前缀

2025-03-10 20:07:43 +08:00
parent ffbb85dfff
commit 9674dfa5dc
1 changed files with 10 additions and 1 deletions
@@ -103,7 +103,16 @@ where
                            if let Ok(redirect) = v.to_str() {
                                log::info!("url重定向响应头 {:?}", res.headers());
                                log::info!("url重定向地址 {}", redirect);
-                                url = redirect.to_string();
+                                // 替换协议前缀
+                                if redirect.starts_with("http://") {
+                                    url = redirect.replacen("http://", "ws://", 1);
+                                } else if redirect.starts_with("https://") {
+                                    url = redirect.replacen("https://", "wss://", 1);
+                                } else {
+                                    url = redirect.to_string();
+                                }
+                                println!("Location：{}", url);
+                                log::info!("修改后的重定向地址: {}", url);
                                continue;
                            }
                        }