This is an archive of the discontinued Mercurial Phabricator instance.

Differential D3294

bundlerepo: use command executor for wire protocol commands
ClosedPublic

Authored by indygreg on Apr 12 2018, 3:59 PM.

Details

Reviewers
durin42
Group Reviewers
hg-reviewers
Commits
rHG1aa4d646d0de: bundlerepo: use command executor for wire protocol commands
Summary

Diff Detail

Repository
rHG Mercurial
Lint
Lint Skipped
Unit
Unit Tests Skipped

Event Timeline

indygreg created this revision.Apr 12 2018, 3:59 PM
indygreg updated this revision to Diff 8133.Apr 13 2018, 3:14 PM
durin42 accepted this revision.Apr 13 2018, 5:55 PM
This revision is now accepted and ready to land.Apr 13 2018, 5:55 PM
Closed by commit rHG1aa4d646d0de: bundlerepo: use command executor for wire protocol commands (authored by indygreg). · Explain WhyApr 13 2018, 6:09 PM
This revision was automatically updated to reflect the committed changes.

Revision Contents
Changeset List

PathPackages
Mmercurial/bundlerepo.py (41 lines)

Diff 8087

mercurial/bundlerepo.py

# developer config: devel.legacy.exchange # developer config: devel.legacy.exchange
legexc = ui.configlist('devel', 'legacy.exchange') legexc = ui.configlist('devel', 'legacy.exchange')
forcebundle1 = 'bundle2' not in legexc and 'bundle1' in legexc forcebundle1 = 'bundle2' not in legexc and 'bundle1' in legexc
canbundle2 = (not forcebundle1 canbundle2 = (not forcebundle1
and peer.capable('getbundle') and peer.capable('getbundle')
and peer.capable('bundle2')) and peer.capable('bundle2'))
if canbundle2: if canbundle2:
kwargs = {} with peer.commandexecutor() as e:
kwargs[r'common'] = common fincoming = e.callcommand('getbundle', {
kwargs[r'heads'] = rheads 'source': 'incoming',
kwargs[r'bundlecaps'] = exchange.caps20to10(repo, role='client') 'common': common,
kwargs[r'cg'] = True 'heads': rheads,
b2 = peer.getbundle('incoming', **kwargs) 'bundlecaps': exchange.caps20to10(repo, role='client'),
fname = bundle = changegroup.writechunks(ui, b2._forwardchunks(), 'cg': True,
})
e.sendcommands()
b2 = fincoming.result()
fname = bundle = changegroup.writechunks(ui,
b2._forwardchunks(),
bundlename) bundlename)
else: else:
if peer.capable('getbundle'): if peer.capable('getbundle'):
cg = peer.getbundle('incoming', common=common, heads=rheads) with peer.commandexecutor() as e:
fincoming = e.callcommand('getbundle', {
'source': 'incoming',
'common': common,
'heads': rheads,
})
e.sendcommands()
cg = fincoming.result()
elif onlyheads is None and not peer.capable('changegroupsubset'): elif onlyheads is None and not peer.capable('changegroupsubset'):
# compat with older servers when pulling all remote heads # compat with older servers when pulling all remote heads
with peer.commandexecutor() as e: with peer.commandexecutor() as e:
fcg = e.callcommand('changegroup', { fcg = e.callcommand('changegroup', {
'nodes': incoming, 'nodes': incoming,
'source': 'incoming', 'source': 'incoming',
}) })
# So use an unfiltered version # So use an unfiltered version
# The discovery process probably need cleanup to avoid that # The discovery process probably need cleanup to avoid that
localrepo = localrepo.unfiltered() localrepo = localrepo.unfiltered()
csets = localrepo.changelog.findmissing(common, rheads) csets = localrepo.changelog.findmissing(common, rheads)
if bundlerepo: if bundlerepo:
reponodes = [ctx.node() for ctx in bundlerepo[bundlerepo.firstnewrev:]] reponodes = [ctx.node() for ctx in bundlerepo[bundlerepo.firstnewrev:]]
remotephases = peer.listkeys('phases')
with peer.commandexecutor() as e:
flistkeys = e.callcommand('listkeys', {
'namespace': 'phases',
})
remotephases = flistkeys.result()
pullop = exchange.pulloperation(bundlerepo, peer, heads=reponodes) pullop = exchange.pulloperation(bundlerepo, peer, heads=reponodes)
pullop.trmanager = bundletransactionmanager() pullop.trmanager = bundletransactionmanager()
exchange._pullapplyphases(pullop, remotephases) exchange._pullapplyphases(pullop, remotephases)
def cleanup(): def cleanup():
if bundlerepo: if bundlerepo:
bundlerepo.close() bundlerepo.close()
if bundle: if bundle:
os.unlink(bundle) os.unlink(bundle)
peer.close() peer.close()
return (localrepo, csets, cleanup) return (localrepo, csets, cleanup)