@mx-sose-front/mx-sose-graph 1.1.8 → 1.1.9

package/src/index.ts

@@ -14,7 +14,7 @@ export { useGraphStore, eventBus } from './store'
 export { ContextMenuUtils } from './utils/contextMenuUtils'
 
 // 导出类型
-export type { Shape, Bounds, Style, NameObject, TaggedValueLabel, Comparent, GraphEvents } from './types'
+export type { Shape, Bounds, Style, NameObject, TaggedValueLabel, Comparent, GraphEvents, GanttData, GanttColumn, GanttContainsModel } from './types'
 /**
  * 可选的安装参数：允许使用方主动把 pinia 传进来
  */

package/src/render/shape-renderer.ts

@@ -1,6 +1,6 @@
 import type { CSSProperties } from "vue";
 import type { Shape } from "@/types";
-import { ShapeKeyMap, DiagramKeyMap, PinKeyMap, EdgeKeyMap } from "../constants";
+import { ShapeKeyMap, DiagramKeyMap, PinKeyMap, EdgeKeyMap, GanttKeyMap, TableKeyMap, MatrixKeyMap } from "../constants";
 import { getShapeComponentByKey } from "./shape-registry";
 
 // 根据 shapeKey 匹配具体组件 (shapeType: 'shape')
@@ -21,6 +21,24 @@ function getComponentByEdgeKey(shapeKey: string) {
   return getShapeComponentByKey(componentName) || componentName;
 }
 
+// 根据 shapeKey 匹配 Gantt 组件 (shapeType: 'gantt')
+export function getComponentByGanttKey(shapeKey: string) {
+  const componentName = (GanttKeyMap as any)[shapeKey] || shapeKey;
+  return getShapeComponentByKey(componentName) || componentName;
+}
+
+// 根据 shapeKey 匹配 Table 组件 (shapeType: 'table')
+export function getComponentByTableKey(shapeKey: string) {
+  const componentName = (TableKeyMap as any)[shapeKey] || shapeKey;
+  return getShapeComponentByKey(componentName) || componentName;
+}
+
+// 根据 shapeKey 匹配 Matrix 组件 (shapeType: 'matrix')
+export function getComponentByMatrixKey(shapeKey: string) {
+  const componentName = (MatrixKeyMap as any)[shapeKey] || shapeKey;
+  return getShapeComponentByKey(componentName) || componentName;
+}
+
 // 根据 shapeKey 匹配 Edge 组件 (shapeType: 'diagram')
 export function getComponentByDiagramKey(shapeKey: string) {
   const componentName = (DiagramKeyMap as any)[shapeKey] || shapeKey;
@@ -28,6 +46,7 @@ export function getComponentByDiagramKey(shapeKey: string) {
 }
 
 // 根据 shape 读取组件
+// 注意：后端可能对矩阵/表格/甘特图也传 shapeType='diagram'，需按 shapeKey 优先匹配 Matrix/Table/Gantt
 export const getShapeComponent = (shape: Shape) => {
   const { shapeType, shapeKey } = shape;
   switch (shapeType) {
@@ -36,9 +55,18 @@ export const getShapeComponent = (shape: Shape) => {
     case "shape":
       return getComponentByShapeKey(shapeKey);
     case "diagram":
+      if ((MatrixKeyMap as any)[shapeKey]) return getComponentByMatrixKey(shapeKey);
+      if ((TableKeyMap as any)[shapeKey]) return getComponentByTableKey(shapeKey);
+      if ((GanttKeyMap as any)[shapeKey]) return getComponentByGanttKey(shapeKey);
       return getComponentByDiagramKey(shapeKey);
     case "edge":
       return getComponentByEdgeKey(shapeKey);
+    case "gantt":
+      return getComponentByGanttKey(shapeKey);
+    case "table":
+      return getComponentByTableKey(shapeKey);
+    case "matrix":
+      return getComponentByMatrixKey(shapeKey);
     default:
       return "ShapeComponent";
   }
@@ -108,11 +136,43 @@ export function clearEdgeStyleCache(): void {
 }
 
 // 根据 shape 的 bounds 生成样式
+// shapeType 为 diagram 但 shapeKey 对应 Matrix/Table/Gantt 时也使用铺满样式
 export const getShapeStyle = (shape: Shape): CSSProperties => {
-  const { bounds, shapeType } = shape
+  const { bounds, shapeType, shapeKey } = shape
 
   if (shapeType === "edge") return getEdgeShapeStyle(shape)
 
+  const isDiagramAsMatrixOrTableOrGantt =
+    shapeType === "diagram" &&
+    ((MatrixKeyMap as any)[shapeKey] || (TableKeyMap as any)[shapeKey] || (GanttKeyMap as any)[shapeKey])
+  if (isDiagramAsMatrixOrTableOrGantt) {
+    return {
+      position: "relative",
+      width: "100%",
+      height: "100%",
+    }
+  }
+  if (shapeType === "gantt") {
+    return {
+      position: "relative",
+      width: "100%",
+      height: "100%",
+    }
+  }
+  if (shapeType === "table") {
+    return {
+      position: "relative",
+      width: "100%",
+      height: "100%",
+    }
+  }
+  if (shapeType === "matrix") {
+    return {
+      position: "relative",
+      width: "100%",
+      height: "100%",
+    }
+  }
   return {
     position: "absolute",
     left: `${bounds?.x || 0}px`,

package/src/store/graphStore.ts

@@ -20,13 +20,57 @@ import {
   syncShowComparentsByReparent,
 } from '../utils/compartment'
 import { EdgeUtils } from '../utils/edgeUtils'
+import { snapPinToParentEdge } from '../utils/pinUtils'
 import { expandParentByChild } from '../utils/autoExpandParent'
 import { batchAutoExpandParents } from '../utils/batchAutoExpand'
 import { adjustCanvasToFitAllShapes } from '../utils/diagram'
 import { applyReparentAndClone, autoExpandMovedCompartmentsAfterDrag, buildDragEndPayloads, buildPrevParentMap, collectAffectedShapeIds, createOnNestDoneCallback } from '../utils/graphDragService'
 import { createShapeOperator, type ShapeOp, type ShapeId } from "../utils/shapeOps/shapeOps"
 type Rect = { x: number; y: number; width: number; height: number };
+
+const GRAPH_SELECTION_STORAGE_KEY = 'mx-sose-graph:selected-state'
+
+interface PersistedSelectionState {
+  diagramId: string | null
+  selectedIds: string[]
+}
+
+function loadPersistedSelectionState(): PersistedSelectionState {
+  if (typeof window === 'undefined') {
+    return { diagramId: null, selectedIds: [] }
+  }
+
+  try {
+    const raw = window.sessionStorage.getItem(GRAPH_SELECTION_STORAGE_KEY)
+    if (!raw) {
+      return { diagramId: null, selectedIds: [] }
+    }
+
+    const parsed = JSON.parse(raw) as Partial<PersistedSelectionState>
+    return {
+      diagramId: typeof parsed.diagramId === 'string' ? parsed.diagramId : null,
+      selectedIds: Array.isArray(parsed.selectedIds)
+        ? parsed.selectedIds.filter((id): id is string => typeof id === 'string')
+        : [],
+    }
+  } catch {
+    return { diagramId: null, selectedIds: [] }
+  }
+}
+
+function savePersistedSelectionState(state: PersistedSelectionState) {
+  if (typeof window === 'undefined') return
+
+  window.sessionStorage.setItem(GRAPH_SELECTION_STORAGE_KEY, JSON.stringify(state))
+}
+
+function clearPersistedSelectionState() {
+  if (typeof window === 'undefined') return
+
+  window.sessionStorage.removeItem(GRAPH_SELECTION_STORAGE_KEY)
+}
 export const useGraphStore = defineStore('graph', () => {
+  const persistedSelectionState = loadPersistedSelectionState()
   // 状态
   const shapes = ref<Shape[]>([])
   // 在 store 初始化时创建一次
@@ -67,7 +111,8 @@ export const useGraphStore = defineStore('graph', () => {
   //当前激活的画布id
   const activeDiagramId = ref<string | null>(null);
   // 当前被选中的多个 id
-  const selectedIds = ref<string[]>([])
+  const selectedIds = ref<string[]>(persistedSelectionState.selectedIds)
+  const selectionDiagramId = ref<string | null>(persistedSelectionState.diagramId)
   // 外部拖拽创建中的形状 id(用于隐藏实际形状,只显示 ghost)
   const externalCreatingId = ref<string | null>(null)
   // 剪切状态的图元 ID 集合（用于 SVG 遮盖层渲染）
@@ -112,6 +157,29 @@ export const useGraphStore = defineStore('graph', () => {
     return map;
   })
 
+  /**
+   * 连线索引映射
+   * 用于快速查找某个节点关联的所有连线
+   * key: 节点 ID (sourceId 或 targetId), value: 关联的 edge 数组
+   * 性能优化: 避免 updateRelatedEdges / initializeAllEdgeEndpoints 中的全量 filter + find
+   */
+  const edgesByNodeId = computed(() => {
+    const map = new Map<string, Shape[]>()
+    shapes.value.forEach(s => {
+      if (s.shapeType === 'edge') {
+        if (s.sourceId) {
+          if (!map.has(s.sourceId)) map.set(s.sourceId, [])
+          map.get(s.sourceId)!.push(s)
+        }
+        if (s.targetId) {
+          if (!map.has(s.targetId)) map.set(s.targetId, [])
+          map.get(s.targetId)!.push(s)
+        }
+      }
+    })
+    return map
+  })
+
   //图元shapeKey
   const taggedValueLabels = ref<string[]>([])   // 隔间组件的图元类型
   const packagesTypes = ref<string[]>([])           // 需要展示线的隔间组件的图元类型
@@ -150,6 +218,9 @@ export const useGraphStore = defineStore('graph', () => {
     // 通过事件总线发送事件
     // eventBus.emit('shape-added', shape)
 
+    // 图元新增后调整画布大小
+    nextTick(adjustCanvasToFitAllShapes)
+
     // 自动扩父逻辑：如果图元有父元素且需要自动扩父，触发扩父逻辑
     const shouldAutoExpand = options?.autoExpandParent !== false // 默认为 true
     if (shouldAutoExpand && shape.parenShapeId && shape.bounds) {
@@ -163,9 +234,19 @@ export const useGraphStore = defineStore('graph', () => {
   const removeShape = (shapeId: string) => {
     const index = shapes.value.findIndex(s => s.id == shapeId)
     if (index > -1) {
+      const wasPrimarySelected = selectedShape.value?.id === shapeId
+      shapes.value.splice(index, 1)
+      diagrams.value = diagrams.value.filter(diagram => diagram.id !== shapeId)
+      selectedIds.value = selectedIds.value.filter(id => id !== shapeId)
+      pendingNestedIds.value = pendingNestedIds.value.filter(id => id !== shapeId)
+      if (cutShapeIds.value.has(shapeId)) {
+        const nextCutShapeIds = new Set(cutShapeIds.value)
+        nextCutShapeIds.delete(shapeId)
+        cutShapeIds.value = nextCutShapeIds
+      }
       // 如果删除的是当前选中的形状，清除选中状态
-      if (selectedShape.value?.id === shapeId) {
-        selectedShape.value = null
+      syncSelectedState(wasPrimarySelected)
+      if (wasPrimarySelected && !selectedShape.value) {
         eventBus.emit('shape-deselected')
       }
     }
@@ -195,7 +276,9 @@ export const useGraphStore = defineStore('graph', () => {
   })
   // 更新单个图元（不处理父子关系 / comparents，同步由外层 updateShape 负责）
   const updateShapeRaw = (shapeId: string, updates: Partial<Shape>, id: 'id' | 'modelId' = 'id') => {
-    const shape = shapes.value.find(s => s[id] === shapeId)
+    const shape = id === 'id'
+      ? shapeMap.value.get(shapeId) ?? null
+      : shapes.value.find(s => s[id] === shapeId)
     if (shape) {
       // 对 Pin 做强保护：永不允许把 parenShapeId 置空/undefined
       if (shape.shapeType === 'pin' && 'parenShapeId' in updates) {
@@ -218,18 +301,23 @@ export const useGraphStore = defineStore('graph', () => {
   }
   // 更新图元数据
   const updateShape = (shapeId: string, updates: Partial<Shape>, id: 'id' | 'modelId' = 'id') => {
-    const before = shapes.value.find(s => s[id] === shapeId) || null
+    const before = id === 'id'
+      ? shapeMap.value.get(shapeId) ?? null
+      : shapes.value.find(s => s[id] === shapeId) ?? null
     const oldPid = before?.parenShapeId ?? null
     // 先真正写入
     updateShapeRaw(shapeId, updates, id)
-    const after = shapes.value.find(s => s[id] === shapeId) || null
+    // updateShapeRaw 是就地修改，before 和 after 是同一个引用（id 未变时）
+    const after = before
     if (!after) return
 
     // 如果更新了bounds，需要重新计算相关连线的位置
+    // 在高频场景（如拖拽）中使用同步版本保证即时性，
+    // 异步版本留给 endDragShape / endResizeShape 等"结束"时刻
     if (updates.bounds) {
       EdgeUtils.updateRelatedEdges(shapes.value, [after.id], (shape) => {
         updateShape(shape.id, shape);
-      });
+      }, edgesByNodeId.value, shapeMap.value);
     }
 
     const newPid = after.parenShapeId ?? null
@@ -259,23 +347,71 @@ export const useGraphStore = defineStore('graph', () => {
   }
   const sameId = (a?: string | null, b?: string | null) =>
     (a ?? null) === (b ?? null)
+  const getLoadedDiagramId = () => shapes.value[0]?.diagramId ?? null
+  const persistSelectionState = () => {
+    if (!selectedIds.value.length) {
+      selectionDiagramId.value = null
+      clearPersistedSelectionState()
+      return
+    }
+
+    const selected = selectedIds.value
+      .map(id => shapeMap.value.get(id))
+      .find((shape): shape is Shape => !!shape)
+
+    selectionDiagramId.value = selected?.diagramId ?? selectionDiagramId.value ?? getLoadedDiagramId()
+    savePersistedSelectionState({
+      diagramId: selectionDiagramId.value,
+      selectedIds: [...selectedIds.value],
+    })
+  }
+  const syncSelectedState = (emit = false) => {
+    const loadedDiagramId = getLoadedDiagramId()
+    const ids = selectionDiagramId.value && loadedDiagramId && selectionDiagramId.value !== loadedDiagramId
+      ? []
+      : selectedIds.value.filter(id => shapeMap.value.has(id))
+    const nextSelectedShape = ids.length
+      ? (shapeMap.value.get(ids[0]) ?? null)
+      : null
+
+    selectedIds.value = ids
+    selectedShape.value = nextSelectedShape
+
+    if (nextSelectedShape) {
+      selectionDiagramId.value = nextSelectedShape.diagramId ?? loadedDiagramId
+      persistSelectionState()
+    } else {
+      selectionDiagramId.value = null
+      clearPersistedSelectionState()
+    }
+
+    if (emit) {
+      eventBus.emit('shape-selected', nextSelectedShape)
+    }
+  }
   //选中图元
   const selectShape = (shape: Shape | null) => {
     const prevId = selectedShape.value?.id ?? null
     const nextId = shape?.id ?? null
-    if (sameId(prevId, nextId)) return
+    const nextSelectedIds = shape ? [shape.id] : []
+    const hasSameSelection =
+      selectedIds.value.length === nextSelectedIds.length &&
+      selectedIds.value.every((id, index) => id === nextSelectedIds[index])
+    if (sameId(prevId, nextId) && hasSameSelection) return
     selectedShape.value = shape
-    selectedIds.value = shape ? [shape.id] : []
+    selectedIds.value = nextSelectedIds
+    selectionDiagramId.value = shape?.diagramId ?? null
+    persistSelectionState()
     // 通过事件总线发送事件
     eventBus.emit('shape-selected', shape)
   }
   // 被框选中的多个 shape/映射 selectedIds.value中的数据
   const marqueeShapes = computed(() => {
-    const byId = (id: string) => shapes.value.find(s => s.id === id)
+    const map = shapeMap.value
     const isShape = (x: Shape | undefined): x is Shape => !!x
     const pending = new Set(pendingNestedIds.value)
     return selectedIds.value
-      .map(byId)
+      .map(id => map.get(id))
       .filter(isShape)
       .filter(s => !pending.has(s.id))
   })
@@ -304,18 +440,24 @@ export const useGraphStore = defineStore('graph', () => {
   })
   // 选择一组（用于框选/Shift 叠加）
   const selectMany = (ids: string[]) => {
-    selectedIds.value = Array.from(new Set(ids))
-    selectedShape.value = ids.length
-      ? (shapes.value.find(s => s.id === ids[0]) ?? null)
+    const nextIds = Array.from(new Set(ids))
+    selectedIds.value = nextIds
+    selectedShape.value = nextIds.length
+      ? (shapeMap.value.get(nextIds[0]) ?? null)
       : null
+    selectionDiagramId.value = selectedShape.value?.diagramId ?? getLoadedDiagramId()
+    persistSelectionState()
   }
   // 清空选择
   const clearSelection = () => selectShape(null)
   // 全选图元
   const selectAll = () => {
-    // 获取所有非diagram类型的图元
+    // 获取所有非diagram、非edge类型的图元
     const allShapeIds = shapes.value
-      .filter(shape => shape.shapeType?.toLowerCase() !== 'diagram')
+      .filter(shape => {
+        const t = shape.shapeType?.toLowerCase()
+        return t !== 'diagram' && t !== 'edge'
+      })
       .map(shape => shape.id)
     selectMany(allShapeIds)
   }
@@ -346,6 +488,7 @@ export const useGraphStore = defineStore('graph', () => {
       op,
     })
     pendingNestedIds.value = []
+    syncSelectedState(op === 'replace' || op === 'delete')
     // eventBus.emit('shapes-updated', newShapes)
     // hydrateAllComparents()
 
@@ -356,6 +499,9 @@ export const useGraphStore = defineStore('graph', () => {
         batchAutoExpandParents(changedShapes, updateShape)
       })
     }
+
+    // 图元增删后调整画布大小（用 nextTick 合并同一 tick 内的多次调用）
+    nextTick(adjustCanvasToFitAllShapes)
   }
   // 获取当前图元数据
   const getShapes = () => {
@@ -366,12 +512,15 @@ export const useGraphStore = defineStore('graph', () => {
     shapes.value = []
     diagrams.value = []
     selectedShape.value = null
+    selectedIds.value = []
+    selectionDiagramId.value = null
     pendingNestedIds.value = []
+    clearPersistedSelectionState()
     eventBus.emit('shapes-cleared')
   }
   //拖动元素相关操作
   // 开始拖动已有元素（仅对 shape 生效）
-  const byId = (id: string) => shapes.value.find(s => s.id === id) || null
+  const byId = (id: string) => shapeMap.value.get(id) ?? null
   const startDragShape = (shapeId: string, pointer: { x: number; y: number }) => {
     startDrag([shapeId], pointer)
   }
@@ -621,7 +770,7 @@ export const useGraphStore = defineStore('graph', () => {
     }
     // —— Pin 终极兜底：不允许在本次拖拽后变成“无父” ——
     for (const id of changedIds) {
-      const node = shapes.value.find(s => s.id === id)
+      const node = shapeMap.value.get(id) ?? null
       if (node?.shapeType === 'pin') {
         const prevPid = prevParentById[id]
         const nowPid = node.parenShapeId ?? null
@@ -632,9 +781,9 @@ export const useGraphStore = defineStore('graph', () => {
     }
 
     // 更新相关 edge 的 waypoints
-    EdgeUtils.updateRelatedEdges(shapes.value, changedIds, (shape) => {
+    await EdgeUtils.updateRelatedEdgesAsync(shapes.value, changedIds, (shape) => {
       updateShape(shape.id, shape)
-    })
+    }, edgesByNodeId.value, shapeMap.value)
     //对“本次直接拖动的隔间”做一次自动扩容检查
     autoExpandMovedCompartmentsAfterDrag(shapes.value, changedIds, updateShape)
     // 把 clone 也并入“需要同步 comparents 的变更集合”
@@ -655,7 +804,7 @@ export const useGraphStore = defineStore('graph', () => {
     // 在生成 payloads 之前，把父扩容先做掉
     const shapeId = ownerPayload?.shapeId
     if (shapeId != null) {
-      const child = shapes.value.find(s => s.id == shapeId)
+      const child = shapeMap.value.get(shapeId) ?? null
       if (child) {
         expandParentByChild({
           shapes: shapes.value,
@@ -696,16 +845,45 @@ export const useGraphStore = defineStore('graph', () => {
     cleanupAfterDrag()
   }
   // 处理缩放结束后的事件发射
-  const endResizeShape = (id: string) => {
+  const endResizeShape = async (id: string) => {
     // 更新相关edge的waypoints
-    EdgeUtils.updateRelatedEdges(shapes.value, [id], (shape) => {
+    await EdgeUtils.updateRelatedEdgesAsync(shapes.value, [id], (shape) => {
       updateShape(shape.id, shape);
-    });
+    }, edgesByNodeId.value, shapeMap.value);
+
+    // 重新计算子栓/端口的位置（父图元缩放后，栓需要贴合新的边缘）
+    const parentShape = shapeMap.value.get(id);
+    if (parentShape) {
+      const childIds = parentChildMap.value.get(id) ?? [];
+      for (const childId of childIds) {
+        const childShape = shapeMap.value.get(childId);
+        if (!childShape) continue;
+        if (
+          pinsTypes.value.includes(childShape.shapeKey) ||
+          portsTypes.value.includes(childShape.shapeKey)
+        ) {
+          // 用栓的中心点作为参考点，重新吸附到父图元最近的边
+          const pb = childShape.bounds ?? {};
+          const pinCenterX = (pb.x ?? 0) + (pb.width ?? 0) / 2;
+          const pinCenterY = (pb.y ?? 0) + (pb.height ?? 0) / 2;
+          const newPos = snapPinToParentEdge(
+            { x: pinCenterX, y: pinCenterY },
+            parentShape,
+            childShape
+          );
+          updateShape(childId, {
+            bounds: { ...childShape.bounds, x: newPos.x, y: newPos.y },
+          });
+        }
+      }
+    }
 
     // 收集受影响的元素
     const affected = new Set<string>();
     affected.add(id);
-    const parentId = shapes.value.find(s => s.id === id)?.parenShapeId ?? null;
+    // 将缩放图元的所有子孙（包括栓/端口）加入受影响集合
+    collectDescendantIds(shapes.value, id).forEach(cid => affected.add(cid));
+    const parentId = shapeMap.value.get(id)?.parenShapeId ?? null;
     if (parentId) {
       affected.add(parentId);
       collectDescendantIds(shapes.value, parentId).forEach(cid => affected.add(cid));
@@ -716,8 +894,8 @@ export const useGraphStore = defineStore('graph', () => {
       let cur: string | null = affectedId;
       let guard = 0;
       while (cur && guard++ < 1000) {
-        const s = shapes.value.find(x => x.id === cur);
-        const p = s ? (s as any).parenShapeId ?? null : null;
+        const s: Shape | null = shapeMap.value.get(cur!) ?? null;
+        const p: string | null = s ? (s as any).parenShapeId ?? null : null;
         if (!p) break;
         affected.add(p);
         cur = p;
@@ -726,9 +904,10 @@ export const useGraphStore = defineStore('graph', () => {
 
     // 组装并派发shape-drag-end事件
     const payloads = Array.from(affected)
-      .map(affectedId => shapes.value.find(s => s.id === affectedId))
+      .map(affectedId => shapeMap.value.get(affectedId))
       .filter(Boolean)
       .map(s => _.cloneDeep(s!)) as Shape[];
+    adjustCanvasToFitAllShapes()
     eventBus.emit('shape-drag-end', payloads,);
   };
 
@@ -793,10 +972,10 @@ export const useGraphStore = defineStore('graph', () => {
    * 初始化所有连线的端点
    * 用于在初始加载后端数据后，自动计算所有连线的合适端点，避免线横跨图元
    */
-  const initializeAllEdgeEndpoints = () => {
-    EdgeUtils.initializeAllEdgeEndpoints(shapes.value, (shape) => {
+  const initializeAllEdgeEndpoints = async () => {
+    await EdgeUtils.initializeAllEdgeEndpointsAsync(shapes.value, (shape) => {
       updateShape(shape.id, shape);
-    });
+    }, shapeMap.value);
   }
   /**
  * 由“子元素数据”触发的扩父入口：
@@ -806,12 +985,13 @@ export const useGraphStore = defineStore('graph', () => {
  */
   const collectAncestors = (shapes: Shape[], id: string) => {
     const out: string[] = []
-    let cur = shapes.find(s => s.id === id)
+    const map = shapeMap.value
+    let cur = map.get(id)
     while (cur) {
       out.push(cur.id)
       const pid = cur.parenShapeId
       if (!pid) break
-      cur = shapes.find(s => s.id === pid)
+      cur = map.get(pid)
     }
     return out
   }
@@ -827,7 +1007,7 @@ export const useGraphStore = defineStore('graph', () => {
     const before = new Map(
       chainIds.map(id => [
         id,
-        JSON.stringify(shapes.value.find(s => s.id === id)?.bounds ?? {}),
+        JSON.stringify(shapeMap.value.get(id)?.bounds ?? {}),
       ]),
     )
 
@@ -840,7 +1020,7 @@ export const useGraphStore = defineStore('graph', () => {
 
     const payload: ShapeSizeUpdatePayload[] = chainIds
       .filter(id => id !== child.id)
-      .map(id => shapes.value.find(s => s.id === id))
+      .map(id => shapeMap.value.get(id))
       .filter((s): s is Shape => !!s)
       .filter(s => before.get(s.id) !== JSON.stringify(s.bounds ?? {})) //  diff
       .map(s => ({ id: s.id, bounds: JSON.stringify(s.bounds ?? {}) }))
@@ -905,6 +1085,7 @@ export const useGraphStore = defineStore('graph', () => {
     hasSelectedShape,
     parentChildMap, // 父子关系索引映射
     shapeMap,       // 图元ID索引映射 (性能优化)
+    edgesByNodeId,  // 连线索引映射 (性能优化)
     marqueeShapes,
     ghostShadow,
     scales,
@@ -950,12 +1131,4 @@ export const useGraphStore = defineStore('graph', () => {
     clearCutShapeIds,
     setCopiedShapesCount,
   }
-},
-  //配置持久化
-  {
-    //@ts-ignore
-    persist: {
-      paths: ['shapes', 'selectedShape']
-    } as any,
-  }
-)
+})