O react até que tem uma forma melhor de lidar com isso, que teoricamente seria com o useLayoutEffect, que roda após a a mudança do componente mas antes da renderização. Vale dar uma olhada.