comparison hgext/evolve.py @ 854:86b826399dfd

exchange: add progress bar when pushing using wire protocol command

Progress bar on pull was only available for pushkey based push before this
changeset. This implementation is very hacky but we are still at a proof of
concept stage for those new exchange method.

853:b82b49189328

testedwith = '2.7 2.7.1 2.7.2 2.8 2.8.1'
buglink = 'https://bitbucket.org/marmoute/mutable-history/issues'

import sys
import random
from cStringIO import StringIO
import struct

import mercurial
from mercurial import util

def _encodemarkersstream(fp, markers):
    fp.write(_pack('>B', 0))
    for mark in markers:
        fp.write(obsolete._encodeonemarker(mark))

@eh.wrapfunction(exchange, '_pushobsolete')
def _pushobsolete(orig, pushop):
    """utility function to push obsolete markers to a remote"""
    pushop.ui.debug('try to push obsolete markers to remote\n')
    repo = pushop.repo

        repo.ui.status("OBSEXC: computing markers relevant to %i nodes\n"
                       % len(nodes))
        markers = repo.obsstore.relevantmarkers(nodes)
        if remote.capable('_evoext_pushobsmarkers_0'):
            repo.ui.status("OBSEXC: writing %i markers\n" % len(markers))
            obsdata = StringIO()
            _encodemarkersstream(obsdata, markers)
            obsdata.seek(0)
            repo.ui.status("OBSEXC: pushing %i bytes\n"
                           % len(obsdata.getvalue()))
            remote.evoext_pushobsmarkers_0(obsdata)
        else:
            rslts = []
            repo.ui.status("OBSEXC: encoding %i markers\n" % len(markers))
            remotedata = _pushkeyescape(markers).items()
            totalbytes = sum(len(d) for k,d in remotedata)

854:86b826399dfd

testedwith = '2.7 2.7.1 2.7.2 2.8 2.8.1'
buglink = 'https://bitbucket.org/marmoute/mutable-history/issues'

import sys
import random
from StringIO import StringIO
import struct

import mercurial
from mercurial import util

def _encodemarkersstream(fp, markers):
    fp.write(_pack('>B', 0))
    for mark in markers:
        fp.write(obsolete._encodeonemarker(mark))

class pushobsmarkerStringIO(StringIO):
    """hacky string io for progress"""

    @util.propertycache
    def _length(self):
        return len(self.getvalue())

    def read(self, size):
        self.ui.progress('OBSEXC', self.tell(), unit="bytes",
                         total=self._length)
        return StringIO.read(self, size)



@eh.wrapfunction(exchange, '_pushobsolete')
def _pushobsolete(orig, pushop):
    """utility function to push obsolete markers to a remote"""
    pushop.ui.debug('try to push obsolete markers to remote\n')
    repo = pushop.repo

        repo.ui.status("OBSEXC: computing markers relevant to %i nodes\n"
                       % len(nodes))
        markers = repo.obsstore.relevantmarkers(nodes)
        if remote.capable('_evoext_pushobsmarkers_0'):
            repo.ui.status("OBSEXC: writing %i markers\n" % len(markers))
            obsdata = pushobsmarkerStringIO()
            _encodemarkersstream(obsdata, markers)
            obsdata.seek(0)
            obsdata.ui = repo.ui
            repo.ui.status("OBSEXC: pushing %i bytes\n"
                           % len(obsdata.getvalue()))
            remote.evoext_pushobsmarkers_0(obsdata)
            repo.ui.progress('OBSEXC', None)
        else:
            rslts = []
            repo.ui.status("OBSEXC: encoding %i markers\n" % len(markers))
            remotedata = _pushkeyescape(markers).items()
            totalbytes = sum(len(d) for k,d in remotedata)