comparison mercurial/exchange.py @ 21002:dda41da069a4

bundle2: lazily generate the changegroup part in exchange.getbundle

Now that we have lazy generation of parts, let's use it.

21001:c93bb6a08fa1

    if bundlecaps is None or 'HG20' not in bundlecaps:
        return cg
    # very crude first implementation,
    # the bundle API will change and the generation will be done lazily.
    bundler = bundle2.bundle20(repo.ui)
    tempname = changegroup.writebundle(cg, None, 'HG10UN')
    data = open(tempname).read()
    part = bundle2.part('changegroup', data=data)
    bundler.addpart(part)
    temp = cStringIO.StringIO()
    for c in bundler.getchunks():
        temp.write(c)
    temp.seek(0)

21002:dda41da069a4

    if bundlecaps is None or 'HG20' not in bundlecaps:
        return cg
    # very crude first implementation,
    # the bundle API will change and the generation will be done lazily.
    bundler = bundle2.bundle20(repo.ui)
    def cgchunks(cg=cg):
        yield 'HG10UN'
        for c in cg.getchunks():
            yield c
    part = bundle2.part('changegroup', data=cgchunks())
    bundler.addpart(part)
    temp = cStringIO.StringIO()
    for c in bundler.getchunks():
        temp.write(c)
    temp.seek(0)