comparison mercurial/exchange.py @ 20476:1180c6ec5695

pull: move obsolescence marker exchange in the exchange module

The obsolescence marker exchange code was already extracted during a previous
cycle. We are moving the extracted functio in this module. This function will
read and write data in the `pulloperation` object and I prefer to have all core
function collaborating through this object in the same place.

This changeset is pure code movement only. Code change for direct consumption of
the `pulloperation` object will come later.

20475:b79b405583af

# GNU General Public License version 2 or any later version.

from i18n import _
from node import hex, nullid
import errno
import util, scmutil, changegroup
import discovery, phases, obsolete, bookmarks


class pushoperation(object):
    """A object that represent a single push operation

        self.heads = heads
        # do we force pull?
        self.force = force

def pull(repo, remote, heads=None, force=False):
    pullop = pulloperation(repo, remote, heads)
    if pullop.remote.local():
        missing = set(pullop.remote.requirements) - pullop.repo.supported
        if missing:
            msg = _("required features are not"
                    " supported in the destination:"

        def gettransaction():
            if tr is None:
                return pullop.repo.transaction(trname)
            return tr

        obstr = obsolete.syncpull(pullop.repo, pullop.remote, gettransaction)
        if obstr is not None:
            tr = obstr

        if tr is not None:
            tr.close()

        if tr is not None:
            tr.release()
        lock.release()

    return result

20476:1180c6ec5695

# GNU General Public License version 2 or any later version.

from i18n import _
from node import hex, nullid
import errno
import util, scmutil, changegroup, base85
import discovery, phases, obsolete, bookmarks


class pushoperation(object):
    """A object that represent a single push operation

        self.heads = heads
        # do we force pull?
        self.force = force

def pull(repo, remote, heads=None, force=False):
    pullop = pulloperation(repo, remote, heads, force)
    if pullop.remote.local():
        missing = set(pullop.remote.requirements) - pullop.repo.supported
        if missing:
            msg = _("required features are not"
                    " supported in the destination:"

        def gettransaction():
            if tr is None:
                return pullop.repo.transaction(trname)
            return tr

        obstr = _pullobsolete(pullop.repo, pullop.remote, gettransaction)
        if obstr is not None:
            tr = obstr

        if tr is not None:
            tr.close()

        if tr is not None:
            tr.release()
        lock.release()

    return result

def _pullobsolete(repo, remote, gettransaction):
    """utility function to pull obsolete markers from a remote

    The `gettransaction` is function that return the pull transaction, creating
    one if necessary. We return the transaction to inform the calling code that
    a new transaction have been created (when applicable).

    Exists mostly to allow overriding for experimentation purpose"""
    tr = None
    if obsolete._enabled:
        repo.ui.debug('fetching remote obsolete markers\n')
        remoteobs = remote.listkeys('obsolete')
        if 'dump0' in remoteobs:
            tr = gettransaction()
            for key in sorted(remoteobs, reverse=True):
                if key.startswith('dump'):
                    data = base85.b85decode(remoteobs[key])
                    repo.obsstore.mergemarkers(tr, data)
            repo.invalidatevolatilesets()
    return tr