diff hgext/largefiles/proto.py @ 37414:2d965bfeb8f6

wireproto: allow direct stream processing for unbundle

Introduce a new option server.streamunbundle which starts a transaction
immediately to apply a bundle instead of writing it to a temporary file
first. This side steps the need for a large tmp directory at the cost of
preventing concurrent pushes. This is a reasonable trade-off for many
setups as concurrent pushes for the main branch at least are disallowed
anyway. The option defaults to off to preserve existing behavior.

Change the wireproto interface to provide a generator for reading the
payload and make callers responsible for consuming all data.

Differential Revision: https://phab.mercurial-scm.org/D2470

--- a/hgext/largefiles/proto.py	Fri Apr 06 22:22:19 2018 +0200
+++ b/hgext/largefiles/proto.py	Tue Feb 27 02:37:31 2018 +0100
@@ -41,7 +41,8 @@
         tmpfp = util.atomictempfile(path, createmode=repo.store.createmode)
 
         try:
-            proto.forwardpayload(tmpfp)
+            for p in proto.getpayload():
+                tmpfp.write(p)
             tmpfp._fp.seek(0)
             if sha != lfutil.hexsha1(tmpfp._fp):
                 raise IOError(0, _('largefile contents do not match hash'))