comparison mercurial/wireproto.py @ 21177:952af771bc17 stable

bundle2: gracefully handle abort during unbundle

Clients expect a bundle2 reply to their bundle2 submission. So we
catch the Abort error and turn it into a bundle2 containing a part
transporting the exception data. The unbundling of this reply will
raise the error again.

21176:70fcb0a71445

    except util.Abort, inst:
        # The old code we moved used sys.stderr directly.
        # We did not change it to minimise code change.
        # This need to be moved to something proper.
        # Feel free to do it.
        sys.stderr.write("abort: %s\n" % inst)
        return pushres(0)
    except exchange.PushRaced, exc:
        return pusherr(str(exc))

21177:952af771bc17

    except util.Abort, inst:
        # The old code we moved used sys.stderr directly.
        # We did not change it to minimise code change.
        # This need to be moved to something proper.
        # Feel free to do it.
        if getattr(inst, 'duringunbundle2', False):
            bundler = bundle2.bundle20(repo.ui)
            manargs = [('message', str(inst))]
            advargs = []
            if inst.hint is not None:
                advargs.append(('hint', inst.hint))
            bundler.addpart(bundle2.bundlepart('B2X:ERROR:ABORT',
                                               manargs, advargs))
            return streamres(bundler.getchunks())
        else:
            sys.stderr.write("abort: %s\n" % inst)
            return pushres(0)
    except exchange.PushRaced, exc:
        return pusherr(str(exc))