export const createData = () => {
  const size = 3 // 每层级节点个数
  const level = 3 // 层级


  const data: Record<string, any> = {
    nodes: [],
    relationships: []
  }

  for(let i = 1; i < size; i++) {
    data.nodes.push(createNode(`${i}`))
    for(let j = 0; j< level; j++) {
      data.nodes.push(createNode(`${i}${j}`))
      data.relationships.push(createRelations(`${i}${j}`, size))
      for(let l = 0; l < level; l++) {
        data.nodes.push(createNode(`${i}${j}${l}`))
        data.relationships.push(createRelations(`${i}${j}${l}`, size))
      }
    }
  }

  return data
}

const createNode = (index: number | string) => {
  return {
    id: index,
    label: `index-${index}`,
    properties: {
      name: `节点${index}`,
    }
  }
}

const createRelations = (index: string, max: number) => {
  const len = index.length
  const start = index.slice(0, len - 1)
  const end = index
  const params = {
    id: `r${index}`,
    "type": "组成", // 关系类型
    "start_node_id": start, // 起始节点id
    "end_node_id": end,
    properties: {
    }
  }
  return params
}