# HG changeset patch # User Gregory Szorc # Date 2018-08-16 19:47:30 # Node ID 858a12846f4feb502c706c7de1cc66058f76a54c # Parent 0e46b92b37b1364bb1c39a74e2469ecbc72642f7 setdiscovery: don't use dagutil for node -> rev conversion The node -> rev conversion is possible using standard storage APIs and doesn't need to involve the dagutil module. Differential Revision: https://phab.mercurial-scm.org/D4306 diff --git a/mercurial/debugcommands.py b/mercurial/debugcommands.py --- a/mercurial/debugcommands.py +++ b/mercurial/debugcommands.py @@ -794,7 +794,7 @@ def debugdiscovery(ui, repo, remoteurl=" cl = repo.changelog clnode = cl.node dag = dagutil.revlogdag(cl) - all = dag.ancestorset(dag.internalizeall(common)) + all = dag.ancestorset(cl.rev(n) for n in common) common = {clnode(r) for r in dag.headsetofconnecteds(all)} else: nodes = None diff --git a/mercurial/setdiscovery.py b/mercurial/setdiscovery.py --- a/mercurial/setdiscovery.py +++ b/mercurial/setdiscovery.py @@ -143,11 +143,11 @@ def findcommonheads(ui, local, remote, roundtrips = 0 cl = local.changelog clnode = cl.node + clrev = cl.rev localsubset = None if ancestorsof is not None: - rev = local.changelog.rev - localsubset = [rev(n) for n in ancestorsof] + localsubset = [clrev(n) for n in ancestorsof] dag = dagutil.revlogdag(cl, localsubset=localsubset) # early exit if we know all the specified remote heads already @@ -175,7 +175,17 @@ def findcommonheads(ui, local, remote, # compatibility reasons) ui.status(_("searching for changes\n")) - srvheads = dag.internalizeall(srvheadhashes, filterunknown=True) + srvheads = [] + for node in srvheadhashes: + if node == nullid: + continue + + try: + srvheads.append(clrev(node)) + # Catches unknown and filtered nodes. + except error.LookupError: + continue + if len(srvheads) == len(srvheadhashes): ui.debug("all remote heads known locally\n") return srvheadhashes, False, srvheadhashes