feat(proxy): add rewrite support for ws (#1407)

Co-authored-by: wangwei <wangwei@qdcares.cn>
vitejs · Jan 7, 2021 · fa3bc34 · fa3bc34
1 parent 3321111
commit fa3bc34
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/vite/src/node/server/middlewares/proxy.ts b/packages/vite/src/node/server/middlewares/proxy.ts
@@ -58,6 +58,9 @@ export function proxyMiddleware({
           (opts.ws || opts.target?.toString().startsWith('ws:')) &&
           req.headers['sec-websocket-protocol'] !== HMR_HEADER
         ) {
+          if (opts.rewrite) {
+            req.url = opts.rewrite(url)
+          }
           proxy.ws(req, socket, head)
         }
       }