mercurial/wireproto.py

30205:b4074417b661

    def groupchunks(self, fh):
        """Generator of chunks to send to the client.

        Some protocols may have compressed the contents.
        """
        raise NotImplementedError()

class remotebatch(peer.batcher):
    '''batches the queued calls; uses as few roundtrips as possible'''

    if not bundle1allowed(repo, 'pull'):
        if not exchange.bundle2requested(opts.get('bundlecaps')):
            return ooberror(bundle2required)

    chunks = exchange.getbundlechunks(repo, 'serve', **opts)
    # TODO avoid util.chunkbuffer() here since it is adding overhead to
    # what is fundamentally a generator proxying operation.
    return streamres(proto.groupchunks(util.chunkbuffer(chunks)))

@wireprotocommand('heads')
def heads(repo, proto):
    h = repo.heads()
    return encodelist(h) + "\n"

30206:d105195436c0

    def groupchunks(self, fh):
        """Generator of chunks to send to the client.

        Some protocols may have compressed the contents.
        """
        raise NotImplementedError()

    def compresschunks(self, chunks):
        """Generator of possible compressed chunks to send to the client.

        This is like ``groupchunks()`` except it accepts a generator as
        its argument.
        """
        raise NotImplementedError()

class remotebatch(peer.batcher):
    '''batches the queued calls; uses as few roundtrips as possible'''

    if not bundle1allowed(repo, 'pull'):
        if not exchange.bundle2requested(opts.get('bundlecaps')):
            return ooberror(bundle2required)

    chunks = exchange.getbundlechunks(repo, 'serve', **opts)
    return streamres(proto.compresschunks(chunks))

@wireprotocommand('heads')
def heads(repo, proto):
    h = repo.heads()
    return encodelist(h) + "\n"