This is an archive of the discontinued Mercurial Phabricator instance.

Differential D718

infinitepush: store received treepack parts
ClosedPublic

Authored by durham on Sep 14 2017, 7:50 PM.
Tags
None
Subscribers
None

Details

Reviewers
stash
Group Reviewers
Restricted Project
Commits
rFBHGXadd3cc24e736: infinitepush: store received treepack parts
Summary

Makes the infinitepush server store the received treepack parts in the bundle.
Future patches will enable serving the trees from the bundle, and sending the
trees from the client.

Diff Detail

Repository
rFBHGX Facebook Mercurial Extensions
Lint
Automatic diff as part of commit; lint not applicable.
Unit
Automatic diff as part of commit; unit tests not applicable.

Event Timeline

durham created this revision.Sep 14 2017, 7:50 PM
Herald added a reviewer: Restricted Project. · View Herald TranscriptSep 14 2017, 7:50 PM
stash accepted this revision.Sep 15 2017, 1:25 PM
This revision is now accepted and ready to land.Sep 15 2017, 1:25 PM
Closed by commit rFBHGXadd3cc24e736: infinitepush: store received treepack parts (authored by durham). · Explain WhySep 22 2017, 1:33 PM
This revision was automatically updated to reflect the committed changes.

Revision Contents
Changeset List

PathPackages
Minfinitepush/__init__.py (9 lines)

Diff 2034

infinitepush/__init__.py

return logger return logger
def processparts(orig, repo, op, unbundler): def processparts(orig, repo, op, unbundler):
if unbundler.params.get('infinitepush') != 'True': if unbundler.params.get('infinitepush') != 'True':
return orig(repo, op, unbundler) return orig(repo, op, unbundler)
handleallparts = repo.ui.configbool('infinitepush', 'storeallparts') handleallparts = repo.ui.configbool('infinitepush', 'storeallparts')
partforwardingwhitelist = []
try:
treemfmod = extensions.find('treemanifest')
partforwardingwhitelist.append(treemfmod.TREEGROUP_PARTTYPE2)
except KeyError:
pass
bundler = bundle2.bundle20(repo.ui) bundler = bundle2.bundle20(repo.ui)
cgparams = None cgparams = None
scratchbookpart = None scratchbookpart = None
with bundle2.partiterator(repo, op, unbundler) as parts: with bundle2.partiterator(repo, op, unbundler) as parts:
for part in parts: for part in parts:
bundlepart = None bundlepart = None
if part.type == 'replycaps': if part.type == 'replycaps':
# This configures the current operation to allow reply parts. # This configures the current operation to allow reply parts.
# If we're not dumping all parts into the new bundle, we need to # If we're not dumping all parts into the new bundle, we need to
# alert the future pushkey handler to skip the part. # alert the future pushkey handler to skip the part.
if not handleallparts: if not handleallparts:
op.records.add(scratchbranchparttype + '_skippushkey', True) op.records.add(scratchbranchparttype + '_skippushkey', True)
elif part.type == scratchbookmarksparttype: elif part.type == scratchbookmarksparttype:
# Save this for later processing. Details below. # Save this for later processing. Details below.
scratchbookpart = part scratchbookpart = part
else: else:
if handleallparts: if handleallparts or part.type in partforwardingwhitelist:
# Ideally we would not process any parts, and instead just # Ideally we would not process any parts, and instead just
# forward them to the bundle for storage, but since this # forward them to the bundle for storage, but since this
# differs from previous behavior, we need to put it behind a # differs from previous behavior, we need to put it behind a
# config flag for incremental rollout. # config flag for incremental rollout.
bundlepart = bundle2.bundlepart(part.type, data=part.read()) bundlepart = bundle2.bundlepart(part.type, data=part.read())
for key, value in part.params.iteritems(): for key, value in part.params.iteritems():
bundlepart.addparam(key, value) bundlepart.addparam(key, value)