unbundle20: retrieve unbundler instances through a factory function
authorPierre-Yves David <pierre-yves.david@fb.com>
Mon, 06 Apr 2015 16:04:33 -0700
changeset 24641 60fecc5b14a4
parent 24640 685639f9430d
child 24642 54e5c239c2d9
unbundle20: retrieve unbundler instances through a factory function To support multiple bundle2 formats, we will need a function returning the proper unbundler according to the header. We introduce such aa function and change the usage in the code base. The function will get smarter in later changesets. This is somewhat similar to the dispatching we do for 'HG10' and 'HG11'. The main target is to allow HG2Y support in an extension to ease transition of companies using the experimental protocol in production (yeah...) But I've no doubt this will be useful when playing with a future HG21.
mercurial/bundle2.py
mercurial/exchange.py
mercurial/localrepo.py
mercurial/wireproto.py
tests/test-bundle2-format.t
--- a/mercurial/bundle2.py	Mon Apr 06 15:40:12 2015 -0700
+++ b/mercurial/bundle2.py	Mon Apr 06 16:04:33 2015 -0700
@@ -521,6 +521,10 @@
         if util.safehasattr(self._fp, 'close'):
             return self._fp.close()
 
+def getunbundler(ui, fp, header=None):
+    """return a valid unbundler object for a given header"""
+    return unbundle20(ui, fp, header)
+
 class unbundle20(unpackermixin):
     """interpret a bundle2 stream
 
--- a/mercurial/exchange.py	Mon Apr 06 15:40:12 2015 -0700
+++ b/mercurial/exchange.py	Mon Apr 06 16:04:33 2015 -0700
@@ -33,7 +33,7 @@
             alg = changegroup.readexactly(fh, 2)
         return changegroup.cg1unpacker(fh, alg)
     elif version == '2Y':
-        return bundle2.unbundle20(ui, fh, header=magic + version)
+        return bundle2.getunbundler(ui, fh, header=magic + version)
     else:
         raise util.Abort(_('%s: unknown bundle version %s') % (fname, version))
 
--- a/mercurial/localrepo.py	Mon Apr 06 15:40:12 2015 -0700
+++ b/mercurial/localrepo.py	Mon Apr 06 16:04:33 2015 -0700
@@ -114,7 +114,7 @@
             # When requesting a bundle2, getbundle returns a stream to make the
             # wire level function happier. We need to build a proper object
             # from it in local peer.
-            cg = bundle2.unbundle20(self.ui, cg)
+            cg = bundle2.getunbundler(self.ui, cg)
         return cg
 
     # TODO We might want to move the next two calls into legacypeer and add
@@ -132,7 +132,7 @@
                 # This little dance should be dropped eventually when the API
                 # is finally improved.
                 stream = util.chunkbuffer(ret.getchunks())
-                ret = bundle2.unbundle20(self.ui, stream)
+                ret = bundle2.getunbundler(self.ui, stream)
             return ret
         except error.PushRaced, exc:
             raise error.ResponseError(_('push failed:'), str(exc))
--- a/mercurial/wireproto.py	Mon Apr 06 15:40:12 2015 -0700
+++ b/mercurial/wireproto.py	Mon Apr 06 16:04:33 2015 -0700
@@ -364,7 +364,7 @@
         f = self._callcompressable("getbundle", **opts)
         bundlecaps = kwargs.get('bundlecaps')
         if bundlecaps is not None and 'HG2Y' in bundlecaps:
-            return bundle2.unbundle20(self.ui, f)
+            return bundle2.getunbundler(self.ui, f)
         else:
             return changegroupmod.cg1unpacker(f, 'UN')
 
@@ -401,7 +401,7 @@
         else:
             # bundle2 push. Send a stream, fetch a stream.
             stream = self._calltwowaystream('unbundle', cg, heads=heads)
-            ret = bundle2.unbundle20(self.ui, stream)
+            ret = bundle2.getunbundler(self.ui, stream)
         return ret
 
     def debugwireargs(self, one, two, three=None, four=None, five=None):
--- a/tests/test-bundle2-format.t	Mon Apr 06 15:40:12 2015 -0700
+++ b/tests/test-bundle2-format.t	Mon Apr 06 16:04:33 2015 -0700
@@ -157,7 +157,7 @@
   >         lock = repo.lock()
   >         tr = repo.transaction('processbundle')
   >         try:
-  >             unbundler = bundle2.unbundle20(ui, sys.stdin)
+  >             unbundler = bundle2.getunbundler(ui, sys.stdin)
   >             op = bundle2.processbundle(repo, unbundler, lambda: tr)
   >             tr.close()
   >         except error.BundleValueError, exc:
@@ -183,7 +183,7 @@
   > @command('statbundle2', [], '')
   > def cmdstatbundle2(ui, repo):
   >     """print statistic on the bundle2 container read from stdin"""
-  >     unbundler = bundle2.unbundle20(ui, sys.stdin)
+  >     unbundler = bundle2.getunbundler(ui, sys.stdin)
   >     try:
   >         params = unbundler.params
   >     except error.BundleValueError, exc: