fix: Fix process.env.SOCKET_SERVER port is NO_VALIDATE

packages/bundler-webpack/src/server/server.ts

@@ -4,7 +4,7 @@ import type { Stats } from '@umijs/bundler-webpack/compiled/webpack';
 import webpack, {
   Configuration,
 } from '@umijs/bundler-webpack/compiled/webpack';
-import { getDevBanner, lodash, logger } from '@umijs/utils';
+import { getDevBanner, lodash, logger, portfinder } from '@umijs/utils';
 import cors from 'cors';
 import { createReadStream, existsSync } from 'fs';
 import http from 'http';
@@ -235,19 +235,31 @@ export async function createServer(opts: IOpts): Promise<any> {
     return null;
   }
 
-  ws = createWebSocketServer(server);
+  const protocol = userConfig.https ? 'https:' : 'http:';
+  const basePort = opts.port || 8000;
+
+  const parseSocketServerAddress = async () => {
+    if (!process.env.SOCKET_SERVER) return basePort;
+    const {port} = new URL(process.env.SOCKET_SERVER);
+    const startPort = Number(port) || basePort;
+    const hmrPort = await portfinder.getPortPromise({ port: startPort })
+    if (port && startPort !== hmrPort) {
+      console.log(`[SOCKET_SERVER] hmr port changed from ${port} to ${basePort}`)
+      return undefined;
+    }
+    return startPort ===  basePort ? undefined : startPort;
+  };
+
+  ws = createWebSocketServer(server, await parseSocketServerAddress());
 
   ws.wss.on('connection', (socket) => {
     if (stats) {
       sendStats(getStats(stats), false, socket);
     }
   });
 
-  const protocol = userConfig.https ? 'https:' : 'http:';
-  const port = opts.port || 8000;
-
-  server.listen(port, () => {
-    const banner = getDevBanner(protocol, opts.host, port);
+  server.listen(basePort, () => {
+    const banner = getDevBanner(protocol, opts.host, basePort);
 
     console.log(banner.before);
     logger.ready(banner.main);

packages/bundler-webpack/src/server/ws.ts

@@ -7,10 +7,16 @@ import WebSocket from '../../compiled/ws';
 
 export function createWebSocketServer(
   server: HttpServer | HttpsServer | Http2Server | Server,
+  port?: number | undefined,
 ) {
-  const wss = new WebSocket.Server({
-    noServer: true,
-  });
+  let wss: WebSocket.Server;
+  if (port) {
+    wss = new WebSocket.Server({port});
+  } else {
+    wss = new WebSocket.Server({
+      noServer: true,
+    });
+  }
 
   server.on('upgrade', (req, socket, head) => {
     if (req.headers['sec-websocket-protocol'] === 'webpack-hmr') {