feat(ws): 新增WebSocket实时通信支持与SSE独立服务

重构中间件结构，新增ws模块实现WebSocket流程执行实时推送
将SSE服务拆分为独立端口监听，默认8866
优化前端流式模式切换，支持WS/SSE协议选择
统一流式事件处理逻辑，完善错误处理与取消机制
更新Cargo.toml依赖，添加WebSocket相关库
调整代码组织结构，规范导入分组与注释

frontend/src/flows/services/custom-service.ts

@@ -15,6 +15,7 @@ import { I18n } from '@flowgram.ai/free-layout-editor';
 import api, { type ApiResp } from '../../utils/axios';
 import { stringifyFlowDoc } from '../utils/yaml';
 import { postSSE } from '../../utils/sse';
+import { getToken } from '../../utils/token'
 
 interface RunResult { ok: boolean; ctx: any; logs: string[] }
 
@@ -124,7 +125,126 @@ export class CustomService {
     }
   }
 
-  // 新增：SSE 流式运行，返回取消函数与完成 Promise
+  // 新增：WebSocket 流式运行
+  runStreamWS(
+    input: any = {},
+    handlers?: {
+      onNode?: (e: StreamEvent & { type: 'node' }) => void;
+      onDone?: (e: StreamEvent & { type: 'done' }) => void;
+      onError?: (e: StreamEvent & { type: 'error' }) => void;
+      onFatal?: (err: Error) => void;
+    }
+  ) {
+    const id = getFlowIdFromUrl();
+    if (!id) {
+      const err = new Error(I18n.t('Flow ID is missing, cannot run'));
+      handlers?.onFatal?.(err);
+      return { cancel: () => {}, done: Promise.resolve<RunResult | null>(null) } as const;
+    }
+
+    // 构造 WS URL
+    const base = (api.defaults.baseURL || '') as string; // 可能是 /api 或 http(s)://host/api
+    function toWsUrl(httpUrl: string) {
+      if (httpUrl.startsWith('https://')) return 'wss://' + httpUrl.slice('https://'.length);
+      if (httpUrl.startsWith('http://')) return 'ws://' + httpUrl.slice('http://'.length);
+      // 相对路径：拼 window.location
+      const origin = window.location.origin; // http(s)://host:port
+      const full = origin.replace(/^http/, 'ws') + (httpUrl.startsWith('/') ? httpUrl : '/' + httpUrl);
+      return full;
+    }
+    const path = `/flows/${id}/run/ws`;
+    // 取 token 放到查询参数（WS 握手无法自定义 Authorization 头部）
+    const token = getToken();
+
+    // 新增：WS 使用独立端口，默认 8855，可通过 VITE_WS_PORT 覆盖
+    const wsPort = (import.meta as any).env?.VITE_WS_PORT || '8855';
+    let wsBase: string;
+    if (base.startsWith('http://') || base.startsWith('https://')) {
+      try {
+        const u = new URL(base);
+        const proto = u.protocol === 'https:' ? 'wss:' : 'ws:';
+        u.protocol = proto;
+        u.port = wsPort; // 改为 WS 端口
+        wsBase = `${u.protocol}//${u.host}${u.pathname.replace(/\/$/, '')}`;
+      } catch {
+        const loc = window.location;
+        const proto = loc.protocol === 'https:' ? 'wss:' : 'ws:';
+        wsBase = `${proto}//${loc.hostname}:${wsPort}${base.startsWith('/') ? base : '/' + base}`;
+      }
+    } else {
+      const loc = window.location;
+      const proto = loc.protocol === 'https:' ? 'wss:' : 'ws:';
+      wsBase = `${proto}//${loc.hostname}:${wsPort}${base.startsWith('/') ? base : '/' + base}`;
+    }
+
+    const wsUrl = wsBase + path + (token ? (wsBase.includes('?') ? `&access_token=${encodeURIComponent(token)}` : `?access_token=${encodeURIComponent(token)}`) : '');
+
+    let ws: WebSocket | null = null;
+    let resolveDone: (v: RunResult | null) => void;
+    let rejectDone: (e: any) => void;
+    const done = new Promise<RunResult | null>((resolve, reject) => { resolveDone = resolve; rejectDone = reject; });
+    let finished = false;
+
+    try {
+      ws = new WebSocket(wsUrl);
+    } catch (e: any) {
+      handlers?.onFatal?.(e);
+      return { cancel: () => {}, done: Promise.resolve<RunResult | null>(null) } as const;
+    }
+
+    ws.onopen = () => {
+      try {
+        ws?.send(JSON.stringify({ input }));
+      } catch (e: any) {
+        handlers?.onFatal?.(e);
+      }
+    };
+
+    ws.onmessage = (ev: MessageEvent) => {
+      try {
+        const data = typeof ev.data === 'string' ? ev.data : '' + ev.data;
+        const evt = JSON.parse(data) as StreamEvent;
+        if (evt.type === 'node') {
+          handlers?.onNode?.(evt as any);
+          return;
+        }
+        if (evt.type === 'error') {
+          handlers?.onError?.(evt as any);
+          return;
+        }
+        if (evt.type === 'done') {
+          finished = true;
+          handlers?.onDone?.(evt as any);
+          resolveDone({ ok: evt.ok, ctx: (evt as any).ctx, logs: (evt as any).logs });
+          ws?.close();
+          return;
+        }
+      } catch (e: any) {
+        // 忽略解析错误为致命，仅记录
+        console.warn('WS message parse error', e);
+      }
+    };
+
+    ws.onerror = (ev: Event) => {
+      if (!finished) {
+        handlers?.onFatal?.(new Error('WebSocket error'));
+      }
+    };
+
+    ws.onclose = () => {
+      if (!finished) {
+        resolveDone(null);
+      }
+    };
+
+    const cancel = () => {
+      try { finished = true; ws?.close(); } catch {}
+    };
+
+    return { cancel, done } as const;
+  }
+
+  // 现有：SSE 流式运行
   runStream(input: any = {}, handlers?: { onNode?: (e: StreamEvent & { type: 'node' }) => void; onDone?: (e: StreamEvent & { type: 'done' }) => void; onError?: (e: StreamEvent & { type: 'error' }) => void; onFatal?: (err: Error) => void; }) {
     const id = getFlowIdFromUrl();
     if (!id) {
@@ -134,7 +254,25 @@ export class CustomService {
     }
 
     const base = (api.defaults.baseURL || '') as string;
-    const url = base ? `${base}/flows/${id}/run/stream` : `/flows/${id}/run/stream`;
+    // 参照 WS：SSE 使用独立端口，默认 8866，可通过 VITE_SSE_PORT 覆盖
+    const ssePort = (import.meta as any).env?.VITE_SSE_PORT || '8866';
+    let sseBase: string;
+    if (base.startsWith('http://') || base.startsWith('https://')) {
+      try {
+        const u = new URL(base);
+        // 协议保持与 base 一致，仅替换端口
+        u.port = ssePort;
+        sseBase = `${u.protocol}//${u.host}${u.pathname.replace(/\/$/, '')}`;
+      } catch {
+        const loc = window.location;
+        sseBase = `${loc.protocol}//${loc.hostname}:${ssePort}${base.startsWith('/') ? base : '/' + base}`;
+      }
+    } else {
+      const loc = window.location;
+      sseBase = `${loc.protocol}//${loc.hostname}:${ssePort}${base.startsWith('/') ? base : '/' + base}`;
+    }
+
+    const url = sseBase + `/flows/${id}/run/stream`;
 
     const { cancel, done } = postSSE<RunResult | null>(url, { input }, {
       onMessage: (json: any) => {
@@ -150,12 +288,12 @@ export class CustomService {
           }
           if (evt.type === 'done') {
             handlers?.onDone?.(evt as any)
-            return { ok: evt.ok, ctx: evt.ctx, logs: evt.logs }
+            return { ok: (evt as any).ok, ctx: (evt as any).ctx, logs: (evt as any).logs }
           }
         } catch (_) {}
         return undefined
       },
-      onFatal: (e) => handlers?.onFatal?.(e),
+      onFatal: (e: any) => handlers?.onFatal?.(e),
     })
     return { cancel, done } as const;
   }