mercurial-scm/hg-stable: tests/simplestorerepo.py annotate

annotate tests/simplestorerepo.py @ 39878:2ac4f3e97813

filelog: stop proxying flags() (API) Per-revision storage flags are kinda a revlog-centric API. (Except for the fact that changegroup uses the same integer flags as revlog does and there's minimal verification that the server's flags map to the client's storage flags - but that's another problem.) The last user of flags() was in verify.py and that code was just moved into revlog.py and is accessed behind the verifyintegrity() file storage API. Since there are no more consumers, let's drop the proxy and remove the method from the file storage interface. This commit only drops the dedicated API for reading a single revision's storage flags: we still support reading and writing flags through the bulk data retrieval and add revision APIs. And since changegroups encode revlog integer flags over the wire, we'll always need to support flags at some level. The removal of individual storage flags may be too premature. But since flags() is now unused, I'd like to see how far we can get without that dedicated API - especially since it uses revision numbers instead of nodes. Differential Revision: https://phab.mercurial-scm.org/D4746

author	Gregory Szorc <gregory.szorc@gmail.com>
date	Mon, 24 Sep 2018 11:16:33 -0700
parents	4b816a83e17e
children	d909c44d29e1

rev	line source
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1 # simplestorerepo.py - Extension that swaps in alternate repository storage.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	2 #
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	3 # Copyright 2018 Gregory Szorc <gregory.szorc@gmail.com>
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	4 #
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	5 # This software may be used and distributed according to the terms of the
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	6 # GNU General Public License version 2 or any later version.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	7
37338 cbc4425e81b5 tests: conditionalize tests based on presence of revlogs for files Gregory Szorc <gregory.szorc@gmail.com> parents: 37337 diff changeset	8 # To use this with the test suite:
cbc4425e81b5 tests: conditionalize tests based on presence of revlogs for files Gregory Szorc <gregory.szorc@gmail.com> parents: 37337 diff changeset	9 #
cbc4425e81b5 tests: conditionalize tests based on presence of revlogs for files Gregory Szorc <gregory.szorc@gmail.com> parents: 37337 diff changeset	10 # $ HGREPOFEATURES="simplestore" ./run-tests.py \
cbc4425e81b5 tests: conditionalize tests based on presence of revlogs for files Gregory Szorc <gregory.szorc@gmail.com> parents: 37337 diff changeset	11 # --extra-config-opt extensions.simplestore=`pwd`/simplestorerepo.py
cbc4425e81b5 tests: conditionalize tests based on presence of revlogs for files Gregory Szorc <gregory.szorc@gmail.com> parents: 37337 diff changeset	12
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	13 from __future__ import absolute_import
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	14
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	15 import stat
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	16
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	17 from mercurial.i18n import _
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	18 from mercurial.node import (
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	19 bin,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	20 hex,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	21 nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	22 nullrev,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	23 )
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	24 from mercurial.thirdparty import (
39259 b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	25 attr,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	26 cbor,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	27 )
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	28 from mercurial import (
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	29 ancestor,
37346 45a4799174a1 tests: disallow using simple store repo with bundlerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37341 diff changeset	30 bundlerepo,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	31 error,
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	32 extensions,
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	33 localrepo,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	34 mdiff,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	35 pycompat,
37441 a3202fa83aff filelog: declare that filelog implements a storage interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37436 diff changeset	36 repository,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	37 revlog,
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	38 store,
37417 76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	39 verify,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	40 )
38548 404eab7ff33f tests: use interfaceutil in simplestorerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37442 diff changeset	41 from mercurial.utils import (
404eab7ff33f tests: use interfaceutil in simplestorerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37442 diff changeset	42 interfaceutil,
404eab7ff33f tests: use interfaceutil in simplestorerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37442 diff changeset	43 )
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	44
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	45 # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	46 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	47 # be specifying the version(s) of Mercurial they are tested with, or
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	48 # leave the attribute unspecified.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	49 testedwith = 'ships-with-hg-core'
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	50
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	51 REQUIREMENT = 'testonly-simplestore'
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	52
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	53 def validatenode(node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	54 if isinstance(node, int):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	55 raise ValueError('expected node; got int')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	56
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	57 if len(node) != 20:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	58 raise ValueError('expected 20 byte node')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	59
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	60 def validaterev(rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	61 if not isinstance(rev, int):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	62 raise ValueError('expected int')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	63
39793 b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	64 class simplestoreerror(error.StorageError):
b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	65 pass
b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	66
39259 b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	67 @interfaceutil.implementer(repository.irevisiondelta)
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	68 @attr.s(slots=True, frozen=True)
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	69 class simplestorerevisiondelta(object):
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	70 node = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	71 p1node = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	72 p2node = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	73 basenode = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	74 linknode = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	75 flags = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	76 baserevisionsize = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	77 revision = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	78 delta = attr.ib()
b41d023a412a repository: establish API for emitting revision deltas Gregory Szorc <gregory.szorc@gmail.com> parents: 38548 diff changeset	79
38548 404eab7ff33f tests: use interfaceutil in simplestorerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37442 diff changeset	80 @interfaceutil.implementer(repository.ifilestorage)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	81 class filestorage(object):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	82 """Implements storage for a tracked path.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	83
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	84 Data is stored in the VFS in a directory corresponding to the tracked
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	85 path.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	86
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	87 Index data is stored in an ``index`` file using CBOR.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	88
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	89 Fulltext data is stored in files having names of the node.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	90 """
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	91
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	92 def __init__(self, svfs, path):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	93 self._svfs = svfs
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	94 self._path = path
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	95
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	96 self._storepath = b'/'.join([b'data', path])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	97 self._indexpath = b'/'.join([self._storepath, b'index'])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	98
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	99 indexdata = self._svfs.tryread(self._indexpath)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	100 if indexdata:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	101 indexdata = cbor.loads(indexdata)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	102
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	103 self._indexdata = indexdata or []
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	104 self._indexbynode = {}
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	105 self._indexbyrev = {}
39865 d9b3cc3d5d07 filelog: drop index attribute (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39849 diff changeset	106 self._index = []
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	107 self._refreshindex()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	108
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	109 def _refreshindex(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	110 self._indexbynode.clear()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	111 self._indexbyrev.clear()
39865 d9b3cc3d5d07 filelog: drop index attribute (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39849 diff changeset	112 self._index = []
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	113
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	114 for i, entry in enumerate(self._indexdata):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	115 self._indexbynode[entry[b'node']] = entry
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	116 self._indexbyrev[i] = entry
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	117
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	118 self._indexbynode[nullid] = {
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	119 b'node': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	120 b'p1': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	121 b'p2': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	122 b'linkrev': nullrev,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	123 b'flags': 0,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	124 }
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	125
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	126 self._indexbyrev[nullrev] = {
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	127 b'node': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	128 b'p1': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	129 b'p2': nullid,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	130 b'linkrev': nullrev,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	131 b'flags': 0,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	132 }
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	133
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	134 for i, entry in enumerate(self._indexdata):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	135 p1rev, p2rev = self.parentrevs(self.rev(entry[b'node']))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	136
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	137 # start, length, rawsize, chainbase, linkrev, p1, p2, node
39865 d9b3cc3d5d07 filelog: drop index attribute (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39849 diff changeset	138 self._index.append((0, 0, 0, -1, entry[b'linkrev'], p1rev, p2rev,
d9b3cc3d5d07 filelog: drop index attribute (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39849 diff changeset	139 entry[b'node']))
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	140
39865 d9b3cc3d5d07 filelog: drop index attribute (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39849 diff changeset	141 self._index.append((0, 0, 0, -1, -1, -1, -1, nullid))
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	142
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	143 def __len__(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	144 return len(self._indexdata)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	145
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	146 def __iter__(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	147 return iter(range(len(self)))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	148
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	149 def revs(self, start=0, stop=None):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	150 step = 1
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	151 if stop is not None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	152 if start > stop:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	153 step = -1
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	154
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	155 stop += step
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	156 else:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	157 stop = len(self)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	158
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	159 return range(start, stop, step)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	160
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	161 def parents(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	162 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	163
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	164 if node not in self._indexbynode:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	165 raise KeyError('unknown node')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	166
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	167 entry = self._indexbynode[node]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	168
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	169 return entry[b'p1'], entry[b'p2']
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	170
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	171 def parentrevs(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	172 p1, p2 = self.parents(self._indexbyrev[rev][b'node'])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	173 return self.rev(p1), self.rev(p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	174
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	175 def rev(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	176 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	177
37408 dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	178 try:
dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	179 self._indexbynode[node]
dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	180 except KeyError:
dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	181 raise error.LookupError(node, self._indexpath, _('no node'))
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	182
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	183 for rev, entry in self._indexbyrev.items():
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	184 if entry[b'node'] == node:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	185 return rev
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	186
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	187 raise error.ProgrammingError('this should not occur')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	188
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	189 def node(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	190 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	191
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	192 return self._indexbyrev[rev][b'node']
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	193
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	194 def lookup(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	195 if isinstance(node, int):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	196 return self.node(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	197
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	198 if len(node) == 20:
37408 dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	199 self.rev(node)
dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	200 return node
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	201
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	202 try:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	203 rev = int(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	204 if '%d' % rev != node:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	205 raise ValueError
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	206
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	207 if rev < 0:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	208 rev = len(self) + rev
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	209 if rev < 0 or rev >= len(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	210 raise ValueError
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	211
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	212 return self.node(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	213 except (ValueError, OverflowError):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	214 pass
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	215
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	216 if len(node) == 40:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	217 try:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	218 rawnode = bin(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	219 self.rev(rawnode)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	220 return rawnode
37408 dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	221 except TypeError:
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	222 pass
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	223
37408 dd2753729853 simplestore: shore up lookup errors Gregory Szorc <gregory.szorc@gmail.com> parents: 37346 diff changeset	224 raise error.LookupError(node, self._path, _('invalid lookup input'))
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	225
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	226 def linkrev(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	227 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	228
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	229 return self._indexbyrev[rev][b'linkrev']
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	230
39878 2ac4f3e97813 filelog: stop proxying flags() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39872 diff changeset	231 def _flags(self, rev):
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	232 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	233
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	234 return self._indexbyrev[rev][b'flags']
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	235
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	236 def deltaparent(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	237 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	238
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	239 p1node = self.parents(self.node(rev))[0]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	240 return self.rev(p1node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	241
39262 3682b49e0213 repository: remove candelta() from ifileindex Gregory Szorc <gregory.szorc@gmail.com> parents: 39260 diff changeset	242 def _candelta(self, baserev, rev):
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	243 validaterev(baserev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	244 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	245
39878 2ac4f3e97813 filelog: stop proxying flags() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39872 diff changeset	246 if ((self._flags(baserev) & revlog.REVIDX_RAWTEXT_CHANGING_FLAGS)
2ac4f3e97813 filelog: stop proxying flags() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39872 diff changeset	247 or (self._flags(rev) & revlog.REVIDX_RAWTEXT_CHANGING_FLAGS)):
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	248 return False
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	249
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	250 return True
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	251
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	252 def rawsize(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	253 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	254 node = self.node(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	255 return len(self.revision(node, raw=True))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	256
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	257 def _processflags(self, text, flags, operation, raw=False):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	258 if flags == 0:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	259 return text, True
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	260
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	261 if flags & ~revlog.REVIDX_KNOWN_FLAGS:
39793 b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	262 raise simplestoreerror(_("incompatible revision flag '%#x'") %
b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	263 (flags & ~revlog.REVIDX_KNOWN_FLAGS))
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	264
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	265 validatehash = True
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	266 # Depending on the operation (read or write), the order might be
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	267 # reversed due to non-commutative transforms.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	268 orderedflags = revlog.REVIDX_FLAGS_ORDER
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	269 if operation == 'write':
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	270 orderedflags = reversed(orderedflags)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	271
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	272 for flag in orderedflags:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	273 # If a flagprocessor has been registered for a known flag, apply the
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	274 # related operation transform and update result tuple.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	275 if flag & flags:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	276 vhash = True
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	277
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	278 if flag not in revlog._flagprocessors:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	279 message = _("missing processor for flag '%#x'") % (flag)
39793 b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	280 raise simplestoreerror(message)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	281
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	282 processor = revlog._flagprocessors[flag]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	283 if processor is not None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	284 readtransform, writetransform, rawtransform = processor
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	285
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	286 if raw:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	287 vhash = rawtransform(self, text)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	288 elif operation == 'read':
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	289 text, vhash = readtransform(self, text)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	290 else: # write operation
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	291 text, vhash = writetransform(self, text)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	292 validatehash = validatehash and vhash
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	293
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	294 return text, validatehash
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	295
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	296 def checkhash(self, text, node, p1=None, p2=None, rev=None):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	297 if p1 is None and p2 is None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	298 p1, p2 = self.parents(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	299 if node != revlog.hash(text, p1, p2):
39793 b63dee7bd0d9 global: replace most uses of RevlogError with StorageError (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39789 diff changeset	300 raise simplestoreerror(_("integrity check failed on %s") %
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	301 self._path)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	302
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	303 def revision(self, node, raw=False):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	304 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	305
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	306 if node == nullid:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	307 return b''
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	308
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	309 rev = self.rev(node)
39878 2ac4f3e97813 filelog: stop proxying flags() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39872 diff changeset	310 flags = self._flags(rev)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	311
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	312 path = b'/'.join([self._storepath, hex(node)])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	313 rawtext = self._svfs.read(path)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	314
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	315 text, validatehash = self._processflags(rawtext, flags, 'read', raw=raw)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	316 if validatehash:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	317 self.checkhash(text, node, rev=rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	318
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	319 return text
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	320
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	321 def read(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	322 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	323
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	324 revision = self.revision(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	325
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	326 if not revision.startswith(b'\1\n'):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	327 return revision
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	328
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	329 start = revision.index(b'\1\n', 2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	330 return revision[start + 2:]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	331
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	332 def renamed(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	333 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	334
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	335 if self.parents(node)[0] != nullid:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	336 return False
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	337
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	338 fulltext = self.revision(node)
37442 0596d27457c6 revlog: move parsemeta() and packmeta() from filelog (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37441 diff changeset	339 m = revlog.parsemeta(fulltext)[0]
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	340
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	341 if m and 'copy' in m:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	342 return m['copy'], bin(m['copyrev'])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	343
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	344 return False
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	345
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	346 def cmp(self, node, text):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	347 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	348
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	349 t = text
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	350
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	351 if text.startswith(b'\1\n'):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	352 t = b'\1\n\1\n' + text
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	353
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	354 p1, p2 = self.parents(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	355
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	356 if revlog.hash(t, p1, p2) == node:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	357 return False
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	358
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	359 if self.iscensored(self.rev(node)):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	360 return text != b''
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	361
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	362 if self.renamed(node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	363 t2 = self.read(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	364 return t2 != text
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	365
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	366 return True
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	367
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	368 def size(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	369 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	370
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	371 node = self._indexbyrev[rev][b'node']
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	372
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	373 if self.renamed(node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	374 return len(self.read(node))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	375
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	376 if self.iscensored(rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	377 return 0
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	378
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	379 return len(self.revision(node))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	380
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	381 def iscensored(self, rev):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	382 validaterev(rev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	383
39878 2ac4f3e97813 filelog: stop proxying flags() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39872 diff changeset	384 return self._flags(rev) & revlog.REVIDX_ISCENSORED
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	385
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	386 def commonancestorsheads(self, a, b):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	387 validatenode(a)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	388 validatenode(b)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	389
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	390 a = self.rev(a)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	391 b = self.rev(b)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	392
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	393 ancestors = ancestor.commonancestorsheads(self.parentrevs, a, b)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	394 return pycompat.maplist(self.node, ancestors)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	395
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	396 def descendants(self, revs):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	397 # This is a copy of revlog.descendants()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	398 first = min(revs)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	399 if first == nullrev:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	400 for i in self:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	401 yield i
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	402 return
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	403
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	404 seen = set(revs)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	405 for i in self.revs(start=first + 1):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	406 for x in self.parentrevs(i):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	407 if x != nullrev and x in seen:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	408 seen.add(i)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	409 yield i
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	410 break
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	411
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	412 # Required by verify.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	413 def files(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	414 entries = self._svfs.listdir(self._storepath)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	415
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	416 # Strip out undo.backup.* files created as part of transaction
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	417 # recording.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	418 entries = [f for f in entries if not f.startswith('undo.backup.')]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	419
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	420 return [b'/'.join((self._storepath, f)) for f in entries]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	421
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	422 def add(self, text, meta, transaction, linkrev, p1, p2):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	423 if meta or text.startswith(b'\1\n'):
37442 0596d27457c6 revlog: move parsemeta() and packmeta() from filelog (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37441 diff changeset	424 text = revlog.packmeta(meta, text)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	425
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	426 return self.addrevision(text, transaction, linkrev, p1, p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	427
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	428 def addrevision(self, text, transaction, linkrev, p1, p2, node=None,
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	429 flags=revlog.REVIDX_DEFAULT_FLAGS, cachedelta=None):
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	430 validatenode(p1)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	431 validatenode(p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	432
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	433 if flags:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	434 node = node or revlog.hash(text, p1, p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	435
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	436 rawtext, validatehash = self._processflags(text, flags, 'write')
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	437
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	438 node = node or revlog.hash(text, p1, p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	439
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	440 if node in self._indexbynode:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	441 return node
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	442
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	443 if validatehash:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	444 self.checkhash(rawtext, node, p1=p1, p2=p2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	445
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	446 return self._addrawrevision(node, rawtext, transaction, linkrev, p1, p2,
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	447 flags)
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	448
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	449 def _addrawrevision(self, node, rawtext, transaction, link, p1, p2, flags):
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	450 transaction.addbackup(self._indexpath)
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	451
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	452 path = b'/'.join([self._storepath, hex(node)])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	453
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	454 self._svfs.write(path, rawtext)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	455
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	456 self._indexdata.append({
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	457 b'node': node,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	458 b'p1': p1,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	459 b'p2': p2,
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	460 b'linkrev': link,
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	461 b'flags': flags,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	462 })
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	463
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	464 self._reflectindexupdate()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	465
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	466 return node
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	467
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	468 def _reflectindexupdate(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	469 self._refreshindex()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	470 self._svfs.write(self._indexpath, cbor.dumps(self._indexdata))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	471
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	472 def addgroup(self, deltas, linkmapper, transaction, addrevisioncb=None):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	473 nodes = []
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	474
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	475 transaction.addbackup(self._indexpath)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	476
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	477 for node, p1, p2, linknode, deltabase, delta, flags in deltas:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	478 linkrev = linkmapper(linknode)
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	479 flags = flags or revlog.REVIDX_DEFAULT_FLAGS
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	480
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	481 nodes.append(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	482
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	483 if node in self._indexbynode:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	484 continue
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	485
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	486 # Need to resolve the fulltext from the delta base.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	487 if deltabase == nullid:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	488 text = mdiff.patch(b'', delta)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	489 else:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	490 text = mdiff.patch(self.revision(deltabase), delta)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	491
37436 9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	492 self._addrawrevision(node, text, transaction, linkrev, p1, p2,
9d4f09bfe3ec simplestore: correctly implement flag processors Gregory Szorc <gregory.szorc@gmail.com> parents: 37424 diff changeset	493 flags)
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	494
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	495 if addrevisioncb:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	496 addrevisioncb(self, node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	497
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	498 return nodes
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	499
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	500 def revdiff(self, rev1, rev2):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	501 validaterev(rev1)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	502 validaterev(rev2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	503
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	504 node1 = self.node(rev1)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	505 node2 = self.node(rev2)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	506
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	507 return mdiff.textdiff(self.revision(node1, raw=True),
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	508 self.revision(node2, raw=True))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	509
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	510 def heads(self, start=None, stop=None):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	511 # This is copied from revlog.py.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	512 if start is None and stop is None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	513 if not len(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	514 return [nullid]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	515 return [self.node(r) for r in self.headrevs()]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	516
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	517 if start is None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	518 start = nullid
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	519 if stop is None:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	520 stop = []
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	521 stoprevs = set([self.rev(n) for n in stop])
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	522 startrev = self.rev(start)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	523 reachable = {startrev}
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	524 heads = {startrev}
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	525
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	526 parentrevs = self.parentrevs
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	527 for r in self.revs(start=startrev + 1):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	528 for p in parentrevs(r):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	529 if p in reachable:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	530 if r not in stoprevs:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	531 reachable.add(r)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	532 heads.add(r)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	533 if p in heads and p not in stoprevs:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	534 heads.remove(p)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	535
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	536 return [self.node(r) for r in heads]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	537
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	538 def children(self, node):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	539 validatenode(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	540
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	541 # This is a copy of revlog.children().
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	542 c = []
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	543 p = self.rev(node)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	544 for r in self.revs(start=p + 1):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	545 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	546 if prevs:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	547 for pr in prevs:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	548 if pr == p:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	549 c.append(self.node(r))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	550 elif p == nullrev:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	551 c.append(self.node(r))
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	552 return c
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	553
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	554 def getstrippoint(self, minlink):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	555
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	556 # This is largely a copy of revlog.getstrippoint().
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	557 brokenrevs = set()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	558 strippoint = len(self)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	559
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	560 heads = {}
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	561 futurelargelinkrevs = set()
39801 979e9f124caa filelog: stop proxying headrevs() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39793 diff changeset	562 for head in self.heads():
979e9f124caa filelog: stop proxying headrevs() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39793 diff changeset	563 headlinkrev = self.linkrev(self.rev(head))
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	564 heads[head] = headlinkrev
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	565 if headlinkrev >= minlink:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	566 futurelargelinkrevs.add(headlinkrev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	567
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	568 # This algorithm involves walking down the rev graph, starting at the
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	569 # heads. Since the revs are topologically sorted according to linkrev,
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	570 # once all head linkrevs are below the minlink, we know there are
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	571 # no more revs that could have a linkrev greater than minlink.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	572 # So we can stop walking.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	573 while futurelargelinkrevs:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	574 strippoint -= 1
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	575 linkrev = heads.pop(strippoint)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	576
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	577 if linkrev < minlink:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	578 brokenrevs.add(strippoint)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	579 else:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	580 futurelargelinkrevs.remove(linkrev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	581
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	582 for p in self.parentrevs(strippoint):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	583 if p != nullrev:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	584 plinkrev = self.linkrev(p)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	585 heads[p] = plinkrev
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	586 if plinkrev >= minlink:
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	587 futurelargelinkrevs.add(plinkrev)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	588
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	589 return strippoint, brokenrevs
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	590
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	591 def strip(self, minlink, transaction):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	592 if not len(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	593 return
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	594
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	595 rev, _ignored = self.getstrippoint(minlink)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	596 if rev == len(self):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	597 return
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	598
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	599 # Purge index data starting at the requested revision.
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	600 self._indexdata[rev:] = []
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	601 self._reflectindexupdate()
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	602
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	603 def issimplestorefile(f, kind, st):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	604 if kind != stat.S_IFREG:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	605 return False
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	606
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	607 if store.isrevlog(f, kind, st):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	608 return False
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	609
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	610 # Ignore transaction undo files.
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	611 if f.startswith('undo.'):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	612 return False
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	613
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	614 # Otherwise assume it belongs to the simple store.
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	615 return True
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	616
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	617 class simplestore(store.encodedstore):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	618 def datafiles(self):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	619 for x in super(simplestore, self).datafiles():
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	620 yield x
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	621
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	622 # Supplement with non-revlog files.
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	623 extrafiles = self._walk('data', True, filefilter=issimplestorefile)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	624
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	625 for unencoded, encoded, size in extrafiles:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	626 try:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	627 unencoded = store.decodefilename(unencoded)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	628 except KeyError:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	629 unencoded = None
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	630
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	631 yield unencoded, encoded, size
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	632
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	633 def reposetup(ui, repo):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	634 if not repo.local():
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	635 return
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	636
37346 45a4799174a1 tests: disallow using simple store repo with bundlerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37341 diff changeset	637 if isinstance(repo, bundlerepo.bundlerepository):
45a4799174a1 tests: disallow using simple store repo with bundlerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37341 diff changeset	638 raise error.Abort(_('cannot use simple store with bundlerepo'))
45a4799174a1 tests: disallow using simple store repo with bundlerepo Gregory Szorc <gregory.szorc@gmail.com> parents: 37341 diff changeset	639
37337 d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	640 class simplestorerepo(repo.__class__):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	641 def file(self, f):
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	642 return filestorage(self.svfs, f)
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	643
d257c5f2a940 tests: add test extension implementing custom filelog storage Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	644 repo.__class__ = simplestorerepo
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	645
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	646 def featuresetup(ui, supported):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	647 supported.add(REQUIREMENT)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	648
39563 41aa5dced975 localrepo: pass ui to newreporequirements() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39262 diff changeset	649 def newreporequirements(orig, ui):
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	650 """Modifies default requirements for new repos to use the simple store."""
39563 41aa5dced975 localrepo: pass ui to newreporequirements() (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 39262 diff changeset	651 requirements = orig(ui)
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	652
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	653 # These requirements are only used to affect creation of the store
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	654 # object. We have our own store. So we can remove them.
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	655 # TODO do this once we feel like taking the test hit.
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	656 #if 'fncache' in requirements:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	657 # requirements.remove('fncache')
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	658 #if 'dotencode' in requirements:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	659 # requirements.remove('dotencode')
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	660
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	661 requirements.add(REQUIREMENT)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	662
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	663 return requirements
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	664
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	665 def makestore(orig, requirements, path, vfstype):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	666 if REQUIREMENT not in requirements:
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	667 return orig(requirements, path, vfstype)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	668
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	669 return simplestore(path, vfstype)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	670
37417 76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	671 def verifierinit(orig, self, args, *kwargs):
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	672 orig(self, args, *kwargs)
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	673
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	674 # We don't care that files in the store don't align with what is
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	675 # advertised. So suppress these warnings.
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	676 self.warnorphanstorefiles = False
76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	677
37415 c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	678 def extsetup(ui):
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	679 localrepo.featuresetupfuncs.add(featuresetup)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	680
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	681 extensions.wrapfunction(localrepo, 'newreporequirements',
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	682 newreporequirements)
c2c8962a9465 simplestore: use a custom store for the simple store repo Gregory Szorc <gregory.szorc@gmail.com> parents: 37408 diff changeset	683 extensions.wrapfunction(store, 'store', makestore)
37417 76d2115cb817 verify: allow suppressing warnings about extra files Gregory Szorc <gregory.szorc@gmail.com> parents: 37415 diff changeset	684 extensions.wrapfunction(verify.verifier, '__init__', verifierinit)

Mercurial > public > mercurial-scm > hg-stable

annotate tests/simplestorerepo.py @ 39878:2ac4f3e97813