in antd v4 clicking a node triggers onDropdownVisibleChange

It's no longer possible to control the open state with treeExpandedKeys={expandedKeys} and onDropdownVisibleChange={onDropdownVisibleChange}

I wish to keep the component open after they click on a parent node
but onSelect is called before onDropdownVisibleChange

Example code

  const [expandedKeys, setExpandedKeys] = useState<string[]>([])
...

  const onDropdownVisibleChange = (visible: boolean) => {
    setDropdownVisible(visible) // this gets triggered on a node click
  }

  const onSelect = async (value: string, node: DataNode) => {
    if (node.isLeaf) {
     // update
      setDropdownVisible(false)
    } else {
      setExpandedKeys(symmetricDifference(expandedKeys, [value]))
      setDropdownVisible(true) // this doesn't work because onDropdownVisibleChange  gets called after
    }
  }

Start by reproducing the example with controlled treeExpandedKeys and the onSelect/onDropdownVisibleChange callbacks, then trace their event order. Done when selecting a parent keeps the component open while selecting a leaf can close it.