@kgdata/visualgraph/src/utils/nodesFns.ts

visualgraph

export const pickNodesByClosedIds = (
  nodes: { id: string }[],
  closedIds: string,
) => {
  return nodes.filter((node) => {
    return !closedIds.includes(node.id);
  });
};
export const pickNodeByNodeId = (nodes: { id: string }[], nodeId: string) => {
  const node = nodes.filter((n) => n.id === nodeId)[0] || null;
  return node;
};

/**
 *
 * @param nodes 图数据的节点数据
 * @param nodeProps 更新节点pageNum pageNum 1: 初始化数据 2: 展开全部数据
 * @returns
 */
export const updateNodePageById = (
  nodes: { id: string }[],
  nodeProps: { nodePageNum: number; id: string },
) => {
  return nodes.map((n) => {
    if (n.id === nodeProps.id) {
      return {
        ...n,
        ...nodeProps,
      };
    }
    return {
      ...n,
    };
  });
};