mercurial/changegroup.py

39015:ad9eccedb379

class revisiondelta(object):
    """Describes a delta entry in a changegroup.

    Captured data is sufficient to serialize the delta into multiple
    formats.
    """
    # 20 byte node of this revision.
    node = attr.ib()
    # 20 byte nodes of parent revisions.
    p1node = attr.ib()

    basenode = attr.ib()
    # 20 byte node of changeset revision this delta is associated with.
    linknode = attr.ib()
    # 2 bytes of flags to apply to revision data.
    flags = attr.ib()
    # Iterable of chunks holding raw delta data.
    deltachunks = attr.ib()

def _revisiondeltatochunks(delta, headerfn):
    """Serialize a revisiondelta to changegroup chunks."""
    meta = headerfn(delta)
    l = len(meta) + sum(len(x) for x in delta.deltachunks)
    yield chunkheader(l)
    yield meta
    for x in delta.deltachunks:
        yield x

def _sortnodesnormal(store, nodes, reorder):
    """Sort nodes for changegroup generation and turn into revnums."""
    # for generaldelta revlogs, we linearize the revs; this will both be
    # much quicker and generate a much smaller bundle

    """Construct a revision delta for non-ellipses changegroup generation."""
    node = store.node(rev)
    p1, p2 = store.parentrevs(rev)
    base = deltaparentfn(store, rev, p1, p2, prev)

    prefix = ''
    if store.iscensored(base) or store.iscensored(rev):
        try:
            delta = store.revision(node, raw=True)
        except error.CensoredNodeError as e:
            delta = e.tombstone
        if base == nullrev:
            prefix = mdiff.trivialdiffheader(len(delta))
        else:
            baselen = store.rawsize(base)
            prefix = mdiff.replacediffheader(baselen, len(delta))
    elif base == nullrev:
        delta = store.revision(node, raw=True)
        prefix = mdiff.trivialdiffheader(len(delta))
    else:
        delta = store.revdiff(base, rev)
    p1n, p2n = store.parents(node)

    return revisiondelta(
        node=node,
        p1node=p1n,
        p2node=p2n,
        basenode=store.node(base),
        linknode=linknode,
        flags=store.flags(rev),
        deltachunks=(prefix, delta),
    )

def _revisiondeltanarrow(cl, store, ischangelog, rev, linkrev,
                         linknode, clrevtolocalrev, fullclnodes,
                         precomputedellipsis):

    p1n, p2n = store.node(p1), store.node(p2)
    flags = store.flags(rev)
    flags |= revlog.REVIDX_ELLIPSIS

    # TODO: try and actually send deltas for ellipsis data blocks
    data = store.revision(n)
    diffheader = mdiff.trivialdiffheader(len(data))

    return revisiondelta(
        node=n,
        p1node=p1n,
        p2node=p2n,
        basenode=nullid,
        linknode=linknode,
        flags=flags,
        deltachunks=(diffheader, data),
    )

def deltagroup(repo, revs, store, ischangelog, lookup, deltaparentfn,
               units=None,
               ellipses=False, clrevtolocalrev=None, fullclnodes=None,

39016:39b8277e2115

class revisiondelta(object):
    """Describes a delta entry in a changegroup.

    Captured data is sufficient to serialize the delta into multiple
    formats.

    ``revision`` and ``delta`` are mutually exclusive.
    """
    # 20 byte node of this revision.
    node = attr.ib()
    # 20 byte nodes of parent revisions.
    p1node = attr.ib()

    basenode = attr.ib()
    # 20 byte node of changeset revision this delta is associated with.
    linknode = attr.ib()
    # 2 bytes of flags to apply to revision data.
    flags = attr.ib()
    # Size of base revision this delta is against. May be None if
    # basenode is nullid.
    baserevisionsize = attr.ib()
    # Raw fulltext revision data.
    revision = attr.ib()
    # Delta between the basenode and node.
    delta = attr.ib()

def _revisiondeltatochunks(delta, headerfn):
    """Serialize a revisiondelta to changegroup chunks."""

    # The captured revision delta may be encoded as a delta against
    # a base revision or as a full revision. The changegroup format
    # requires that everything on the wire be deltas. So for full
    # revisions, we need to invent a header that says to rewrite
    # data.

    if delta.delta is not None:
        prefix, data = b'', delta.delta
    elif delta.basenode == nullid:
        data = delta.revision
        prefix = mdiff.trivialdiffheader(len(data))
    else:
        data = delta.revision
        prefix = mdiff.replacediffheader(delta.baserevisionsize,
                                         len(data))

    meta = headerfn(delta)

    yield chunkheader(len(meta) + len(prefix) + len(data))
    yield meta
    if prefix:
        yield prefix
    yield data

def _sortnodesnormal(store, nodes, reorder):
    """Sort nodes for changegroup generation and turn into revnums."""
    # for generaldelta revlogs, we linearize the revs; this will both be
    # much quicker and generate a much smaller bundle

    """Construct a revision delta for non-ellipses changegroup generation."""
    node = store.node(rev)
    p1, p2 = store.parentrevs(rev)
    base = deltaparentfn(store, rev, p1, p2, prev)

    revision = None
    delta = None
    baserevisionsize = None

    if store.iscensored(base) or store.iscensored(rev):
        try:
            revision = store.revision(node, raw=True)
        except error.CensoredNodeError as e:
            revision = e.tombstone

        if base != nullrev:
            baserevisionsize = store.rawsize(base)

    elif base == nullrev:
        revision = store.revision(node, raw=True)
    else:
        delta = store.revdiff(base, rev)

    p1n, p2n = store.parents(node)

    return revisiondelta(
        node=node,
        p1node=p1n,
        p2node=p2n,
        basenode=store.node(base),
        linknode=linknode,
        flags=store.flags(rev),
        baserevisionsize=baserevisionsize,
        revision=revision,
        delta=delta,
    )

def _revisiondeltanarrow(cl, store, ischangelog, rev, linkrev,
                         linknode, clrevtolocalrev, fullclnodes,
                         precomputedellipsis):

    p1n, p2n = store.node(p1), store.node(p2)
    flags = store.flags(rev)
    flags |= revlog.REVIDX_ELLIPSIS

    # TODO: try and actually send deltas for ellipsis data blocks

    return revisiondelta(
        node=n,
        p1node=p1n,
        p2node=p2n,
        basenode=nullid,
        linknode=linknode,
        flags=flags,
        baserevisionsize=None,
        revision=store.revision(n),
        delta=None,
    )

def deltagroup(repo, revs, store, ischangelog, lookup, deltaparentfn,
               units=None,
               ellipses=False, clrevtolocalrev=None, fullclnodes=None,