changegroup: move close() from narrow
More of the same.
Differential Revision: https://phab.mercurial-scm.org/D4064
--- a/hgext/narrow/narrowchangegroup.py Thu Aug 02 09:53:22 2018 -0700
+++ b/hgext/narrow/narrowchangegroup.py Thu Aug 02 09:52:01 2018 -0700
@@ -51,15 +51,6 @@
extensions.wrapfunction(
changegroup.cg1packer, 'generatefiles', generatefiles)
- def close(orig, self):
- getattr(self, 'clrev_to_localrev', {}).clear()
- if getattr(self, 'next_clrev_to_localrev', {}):
- self.clrev_to_localrev = self.next_clrev_to_localrev
- del self.next_clrev_to_localrev
- self.changelog_done = True
- return orig(self)
- extensions.wrapfunction(changegroup.cg1packer, 'close', close)
-
# In a perfect world, we'd generate better ellipsis-ified graphs
# for non-changelog revlogs. In practice, we haven't started doing
# that yet, so the resulting DAGs for the manifestlog and filelogs
--- a/mercurial/changegroup.py Thu Aug 02 09:53:22 2018 -0700
+++ b/mercurial/changegroup.py Thu Aug 02 09:52:01 2018 -0700
@@ -544,6 +544,13 @@
self._verbosenote = lambda s: None
def close(self):
+ # Ellipses serving mode.
+ getattr(self, 'clrev_to_localrev', {}).clear()
+ if getattr(self, 'next_clrev_to_localrev', {}):
+ self.clrev_to_localrev = self.next_clrev_to_localrev
+ del self.next_clrev_to_localrev
+ self.changelog_done = True
+
return closechunk()
def fileheader(self, fname):