react批量编辑转换输入框过程页面卡顿?

前端取消分页功能 页面默认展示全部数据，点击编辑操作需要等七八秒才会有input框显示，取消也是需要等几秒，请教各位大神这种问题如何优化

const arrEditList: any = useMemo(() => {

if (dataSource.length) {
  return dataSavedOriginal.current?.map( (item: { id: any, monthList: any }) => {
    // 批量获取id
    return item.id
  })
}}, [dataSource])

编辑按钮

<Button
  key="edit"
  icon={<IconFont type="icon-bianji" />}
  onClick={() => {
    console.log('出发编辑')
      setEditableKeys(arrEditList)
  }}
  >
  编辑
</Button>