facebook/sapling
1
1
Fork 0
mirror of https://github.com/facebook/sapling.git synced 2024-10-07 15:27:13 +03:00
Code Issues Projects Releases Wiki Activity

bundle2: add reply awareness to unbundlerecords

Browse Source 
We need an efficient way to handle bundle replies. The unbundle records class is
extended to carry such data.
This commit is contained in:
Pierre-Yves David 2014-04-11 08:24:59 -07:00
parent 311fe283fd
commit 90cfb42a5c
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mercurial/bundle2.py
View File

@ -205,14 +205,21 @@ class unbundlerecords(object):
def __init__(self):
self._categories = {}
self._sequences = []
self._replies = {}
def add(self, category, entry):
def add(self, category, entry, inreplyto=None):
"""add a new record of a given category.
The entry can then be retrieved in the list returned by
self['category']."""
self._categories.setdefault(category, []).append(entry)
self._sequences.append((category, entry))
if inreplyto is not None:
self.getreplies(inreplyto).add(category, entry)
def getreplies(self, partid):
"""get the subrecords that replies to a specific part"""
return self._replies.setdefault(partid, unbundlerecords())
def __getitem__(self, cat):
return tuple(self._categories.get(cat, ()))