alighasami/xyflow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(connectionlookup): use maps instead of arrays

Browse Source
This commit is contained in:
moklick
2023-12-06 17:49:34 +01:00
parent b6743a3cf3
commit a58f915697
6 changed files with 76 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/react/src/store/initialState.ts
+1 -1
View File
@@ -25,7 +25,7 @@ const getInitialState = ({
fitView?: boolean;
} = {}): ReactFlowStore => {
const nodeLookup = new Map<string, Node>();
const connectionLookup = updateConnectionLookup(new Map<string, Connection[]>(), edges);
const connectionLookup = updateConnectionLookup(new Map<string, Map<string, Connection>>(), edges);
const nextNodes = updateNodes(nodes, nodeLookup, { nodeOrigin: [0, 0], elevateNodesOnSelect: false });
let transform: Transform = [0, 0, 1];
packages/react/src/store/utils.ts
+5 -6
View File
@@ -44,7 +44,7 @@ export function updateNodesAndEdgesSelections({ changedNodes, changedEdges, get,
}
}
export function updateConnectionLookup(lookup: Map<string, Connection[]>, edges: Edge[]) {
export function updateConnectionLookup(lookup: Map<string, Map<string, Connection>>, edges: Edge[]) {
lookup.clear();
edges.forEach(({ source, target, sourceHandle = null, targetHandle = null }) => {
@@ -52,13 +52,12 @@ export function updateConnectionLookup(lookup: Map<string, Connection[]>, edges:
const sourceKey = `${source}-source-${sourceHandle}`;
const targetKey = `${target}-target-${targetHandle}`;
const prevSource = lookup.get(sourceKey);
const prevTarget = lookup.get(targetKey);
const prevSource = lookup.get(sourceKey) || new Map();
const prevTarget = lookup.get(targetKey) || new Map();
const connection = { source, target, sourceHandle, targetHandle };
lookup.set(sourceKey, prevSource ? [...prevSource, connection] : [connection]);
lookup.set(targetKey, prevTarget ? [...prevTarget, connection] : [connection]);
lookup.set(sourceKey, prevSource.set(`${target}-${targetHandle}`, connection));
lookup.set(targetKey, prevTarget.set(`${source}-${sourceHandle}`, connection));
}
});