mercurial-scm/hg-stable: mercurial/transaction.py annotate

annotate mercurial/transaction.py @ 48813:2bb75c65fa6c stable

ci: use the `v1.0` flavor of the docker images in the CI This new versioning will help us to maintain backward compatibility in the docker image. This will be useful to deal with mismatch between default/stable in version and the re-run CI on older changesets in the future. Once this changeset land on stable, we will have to merge it in default. Then we can start make backward incompatible changes in a new image version. Differential Revision: https://phab.mercurial-scm.org/D12388

author	Pierre-Yves David <pierre-yves.david@octobus.net>
date	Fri, 18 Mar 2022 18:09:46 +0100
parents	21ac6aedd5e5
children	6000f5b25c9b

rev	line source
17424 e7cfe3587ea4 fix trivial spelling errors Mads Kiilerich <mads@kiilerich.com> parents: 16689 diff changeset	1 # transaction.py - simple journaling scheme for mercurial
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	2 #
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	3 # This transaction scheme is intended to gracefully handle program
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	4 # errors and interruptions. More serious failures like system crashes
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	5 # can be recovered with an fsck-like tool. As the whole repository is
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	6 # effectively log-structured, this should amount to simply truncating
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	7 # anything that isn't referenced in the changelog.
9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	8 #
46819 d4ba4d51f85f contributor: change mentions of mpm to olivia Rapha?l Gom?s <rgomes@octobus.net> parents: 46067 diff changeset	9 # Copyright 2005, 2006 Olivia Mackall <olivia@selenic.com>
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	10 #
8225 46293a0c7e9f updated license to be explicit about GPL version 2 Martin Geisler <mg@lazybytes.net> parents: 8071 diff changeset	11 # This software may be used and distributed according to the terms of the
10263 25e572394f5c Update license to GPLv2+ Matt Mackall <mpm@selenic.com> parents: 10228 diff changeset	12 # GNU General Public License version 2 or any later version.
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	13
25986 89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	14 from __future__ import absolute_import
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	15
20886 203908968644 transaction: drop extra import caught by pyflakes Matt Mackall <mpm@selenic.com> parents: 20882 diff changeset	16 import errno
25986 89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	17
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	18 from .i18n import _
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	19 from . import (
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	20 error,
36758 ef345f9e4295 transaction: fix an error string with bytestr() on a repr()d value Augie Fackler <augie@google.com> parents: 35872 diff changeset	21 pycompat,
25986 89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	22 util,
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	23 )
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	24 from .utils import stringutil
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	25
23313 991098579940 transaction: set backupentries version to proper value Pierre-Yves David <pierre-yves.david@fb.com> parents: 23312 diff changeset	26 version = 2
23064 5dc888b79e70 transactions: add version number to journal.backupfiles Durham Goode <durham@fb.com> parents: 23063 diff changeset	27
44433 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	28 GEN_GROUP_ALL = b'all'
baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	29 GEN_GROUP_PRE_FINALIZE = b'prefinalize'
baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	30 GEN_GROUP_POST_FINALIZE = b'postfinalize'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	31
28830 a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	32
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	33 def active(func):
fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	34 def _active(self, args, *kwds):
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	35 if self._count == 0:
45738 5df1655edf42 transaction: use ProgrammingError for when an committed transaction is used Martin von Zweigbergk <martinvonz@google.com> parents: 44553 diff changeset	36 raise error.ProgrammingError(
5df1655edf42 transaction: use ProgrammingError for when an committed transaction is used Martin von Zweigbergk <martinvonz@google.com> parents: 44553 diff changeset	37 b'cannot use transaction when it is already committed/aborted'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	38 )
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	39 return func(self, args, *kwds)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	40
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	41 return _active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	42
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	43
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	44 def _playback(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	45 journal,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	46 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	47 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	48 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	49 entries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	50 backupentries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	51 unlink=True,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	52 checkambigfiles=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	53 ):
47297 18415fc918a1 recover: only apply last journal record per file (issue6423) Joerg Sonnenberger <joerg@bec.de> parents: 46819 diff changeset	54 for f, o in sorted(dict(entries).items()):
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	55 if o or not unlink:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	56 checkambig = checkambigfiles and (f, b'') in checkambigfiles
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	57 try:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	58 fp = opener(f, b'a', checkambig=checkambig)
42965 8502f76dbfd7 transaction: detect an attempt to truncate-to-extend on playback, raise error Kyle Lippincott <spectral@google.com> parents: 41387 diff changeset	59 if fp.tell() < o:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	60 raise error.Abort(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	61 _(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	62 b"attempted to truncate %s to %d bytes, but it was "
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	63 b"already %d bytes\n"
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	64 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	65 % (f, o, fp.tell())
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	66 )
13400 14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	67 fp.truncate(o)
14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	68 fp.close()
9686 ddf2adf88b89 transaction: more specific exceptions, os.unlink can raise OSError Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 9220 diff changeset	69 except IOError:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	70 report(_(b"failed to truncate %s\n") % f)
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	71 raise
48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	72 else:
48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	73 try:
20084 a3378a1b0a05 transaction: unlink target file via vfs FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 17424 diff changeset	74 opener.unlink(f)
25660 328739ea70c3 global: mass rewrite to use modern exception syntax Gregory Szorc <gregory.szorc@gmail.com> parents: 25658 diff changeset	75 except (IOError, OSError) as inst:
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	76 if inst.errno != errno.ENOENT:
48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	77 raise
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	78
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	79 backupfiles = []
23309 7eb520f5efe4 transaction: change the on disk format for backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23291 diff changeset	80 for l, f, b, c in backupentries:
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	81 if l not in vfsmap and c:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	82 report(b"couldn't handle %s: unknown cache location %s\n" % (b, l))
23311 64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	83 vfs = vfsmap[l]
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	84 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	85 if f and b:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	86 filepath = vfs.join(f)
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	87 backuppath = vfs.join(b)
33279 7912404b70f2 transaction: apply checkambig=True only on limited files for similarity FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33278 diff changeset	88 checkambig = checkambigfiles and (f, l) in checkambigfiles
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	89 try:
33279 7912404b70f2 transaction: apply checkambig=True only on limited files for similarity FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33278 diff changeset	90 util.copyfile(backuppath, filepath, checkambig=checkambig)
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	91 backupfiles.append(b)
47424 22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47318 diff changeset	92 except IOError as exc:
22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47318 diff changeset	93 e_msg = stringutil.forcebytestr(exc)
22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47318 diff changeset	94 report(_(b"failed to recover %s (%s)\n") % (f, e_msg))
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	95 else:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	96 target = f or b
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	97 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	98 vfs.unlink(target)
25660 328739ea70c3 global: mass rewrite to use modern exception syntax Gregory Szorc <gregory.szorc@gmail.com> parents: 25658 diff changeset	99 except (IOError, OSError) as inst:
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	100 if inst.errno != errno.ENOENT:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	101 raise
41387 876494fd967d cleanup: delete lots of unused local variables Martin von Zweigbergk <martinvonz@google.com> parents: 41110 diff changeset	102 except (IOError, OSError, error.Abort):
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	103 if not c:
23278 aa19432764d6 transaction: handle missing file in backupentries (instead of using entries) Pierre-Yves David <pierre-yves.david@fb.com> parents: 23253 diff changeset	104 raise
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	105
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	106 backuppath = b"%s.backupfiles" % journal
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	107 if opener.exists(backuppath):
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	108 opener.unlink(backuppath)
26753 96dd93de548c transaction: reorder unlinking .hg/journal and .hg/journal.backupfiles FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 26587 diff changeset	109 opener.unlink(journal)
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	110 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	111 for f in backupfiles:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	112 if opener.exists(f):
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	113 opener.unlink(f)
41387 876494fd967d cleanup: delete lots of unused local variables Martin von Zweigbergk <martinvonz@google.com> parents: 41110 diff changeset	114 except (IOError, OSError, error.Abort):
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	115 # only pure backup file remains, it is sage to ignore any error
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	116 pass
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	117
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	118
33814 bbbbd3c30bfc util: add base class for transactional context managers Martin von Zweigbergk <martinvonz@google.com> parents: 33279 diff changeset	119 class transaction(util.transactional):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	120 def __init__(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	121 self,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	122 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	123 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	124 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	125 journalname,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	126 undoname=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	127 after=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	128 createmode=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	129 validator=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	130 releasefn=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	131 checkambigfiles=None,
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	132 name='<unnamed>',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	133 ):
20881 3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	134 """Begin a new transaction
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	135
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	136 Begins a new transaction that allows rolling back writes in the event of
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	137 an exception.
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	138
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	139 * `after`: called after the transaction has been committed
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	140 * `createmode`: the mode of the journal file that will be created
26576 9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	141 * `releasefn`: called after releasing (with transaction and result)
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	142
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	143 `checkambigfiles` is a set of (path, vfs-location) tuples,
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	144 which determine whether file stat ambiguity should be avoided
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	145 for corresponded files.
20881 3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	146 """
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	147 self._count = 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	148 self._usages = 1
39699 337d6e0fd9c9 transaction: make report a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39698 diff changeset	149 self._report = report
23310 5bd1f6572db0 transaction: pass a vfs map to the transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23309 diff changeset	150 # a vfs to the store content
39698 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	151 self._opener = opener
23310 5bd1f6572db0 transaction: pass a vfs map to the transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23309 diff changeset	152 # a map to access file in various {location -> vfs}
5bd1f6572db0 transaction: pass a vfs map to the transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23309 diff changeset	153 vfsmap = vfsmap.copy()
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	154 vfsmap[b''] = opener # set default value
23310 5bd1f6572db0 transaction: pass a vfs map to the transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23309 diff changeset	155 self._vfsmap = vfsmap
39697 0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39696 diff changeset	156 self._after = after
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	157 self._offsetmap = {}
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	158 self._newfiles = set()
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	159 self._journal = journalname
39691 da9ce63bfa9b transaction: make undoname a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39690 diff changeset	160 self._undoname = undoname
23279 e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	161 self._queue = []
26576 9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	162 # A callback to do something just after releasing transaction.
9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	163 if releasefn is None:
9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	164 releasefn = lambda tr, success: None
39694 040007cd3d81 transaction: make releasefn a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39693 diff changeset	165 self._releasefn = releasefn
26576 9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	166
39696 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39695 diff changeset	167 self._checkambigfiles = set()
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	168 if checkambigfiles:
39696 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39695 diff changeset	169 self._checkambigfiles.update(checkambigfiles)
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	170
39701 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39700 diff changeset	171 self._names = [name]
36827 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36758 diff changeset	172
32301 976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31653 diff changeset	173 # A dict dedicated to precisely tracking the changes introduced in the
976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31653 diff changeset	174 # transaction.
976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31653 diff changeset	175 self.changes = {}
976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31653 diff changeset	176
23279 e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	177 # a dict of arguments to be passed to hooks
e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	178 self.hookargs = {}
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	179 self._file = opener.open(self._journal, b"w+")
23279 e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	180
23309 7eb520f5efe4 transaction: change the on disk format for backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23291 diff changeset	181 # a list of ('location', 'path', 'backuppath', cache) entries.
23311 64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	182 # - if 'backuppath' is empty, no file existed at backup time
64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	183 # - if 'path' is empty, this is a temporary transaction file
64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	184 # - if 'location' is not empty, the path is outside main opener reach.
64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	185 # use 'location' value as a key in a vfsmap to find the right 'vfs'
64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	186 # (cache is currently unused)
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	187 self._backupentries = []
84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	188 self._backupmap = {}
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	189 self._backupjournal = b"%s.backupfiles" % self._journal
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	190 self._backupsfile = opener.open(self._backupjournal, b'w')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	191 self._backupsfile.write(b'%d\n' % version)
23279 e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	192
6065 53ed9b40cfc4 make the journal/undo files from transactions inherit the mode from .hg/store Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 5865 diff changeset	193 if createmode is not None:
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	194 opener.chmod(self._journal, createmode & 0o666)
25658 e93036747902 global: mass rewrite to use modern octal syntax Gregory Szorc <gregory.szorc@gmail.com> parents: 25302 diff changeset	195 opener.chmod(self._backupjournal, createmode & 0o666)
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	196
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	197 # hold file generations to be performed on commit
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	198 self._filegenerators = {}
23543 4dd8a6a1240d spelling: fixes from proofreading of spell checker issues Mads Kiilerich <madski@unity3d.com> parents: 23513 diff changeset	199 # hold callback to write pending data for hooks
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	200 self._pendingcallback = {}
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	201 # True is any pending data have been written ever
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	202 self._anypending = False
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	203 # holds callback to call when writing the transaction
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	204 self._finalizecallback = {}
44553 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	205 # holds callback to call when validating the transaction
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	206 # should raise exception if anything is wrong
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	207 self._validatecallback = {}
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	208 if validator is not None:
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	209 self._validatecallback[b'001-userhooks'] = validator
23543 4dd8a6a1240d spelling: fixes from proofreading of spell checker issues Mads Kiilerich <madski@unity3d.com> parents: 23513 diff changeset	210 # hold callback for post transaction close
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	211 self._postclosecallback = {}
23764 d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	212 # holds callbacks to call during abort
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	213 self._abortcallback = {}
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	214
36827 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36758 diff changeset	215 def __repr__(self):
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	216 name = '/'.join(self._names)
9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	217 return '<transaction name=%s, count=%d, usages=%d>' % (
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	218 name,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	219 self._count,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	220 self._usages,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	221 )
36827 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36758 diff changeset	222
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	223 def __del__(self):
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	224 if self._journal:
9693 c40a1ee20aa5 transaction: always remove empty journal on abort Sune Foldager <cryo@cyanite.org> parents: 9686 diff changeset	225 self._abort()
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	226
48362 3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47428 diff changeset	227 @property
3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47428 diff changeset	228 def finalized(self):
3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47428 diff changeset	229 return self._finalizecallback is None
3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47428 diff changeset	230
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	231 @active
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	232 def startgroup(self):
23250 8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	233 """delay registration of file entry
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	234
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	235 This is used by strip to delay vision of strip offset. The transaction
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	236 sees either none or all of the strip actions to be done."""
23251 85c634ff395a transaction: drop backupentries logic from startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23250 diff changeset	237 self._queue.append([])
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	238
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	239 @active
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	240 def endgroup(self):
23250 8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	241 """apply delayed registration of file entry.
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	242
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	243 This is used by strip to delay vision of strip offset. The transaction
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	244 sees either none or all of the strip actions to be done."""
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	245 q = self._queue.pop()
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	246 for f, o in q:
63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	247 self._addentry(f, o)
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	248
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	249 @active
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	250 def add(self, file, offset):
23252 70809438c644 transaction: document `tr.add` Pierre-Yves David <pierre-yves.david@fb.com> parents: 23251 diff changeset	251 """record the state of an append-only file before update"""
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	252 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	253 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	254 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	255 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	256 ):
10282 08a0f04b56bd many, many trivial check-code fixups Matt Mackall <mpm@selenic.com> parents: 10263 diff changeset	257 return
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	258 if self._queue:
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	259 self._queue[-1].append((file, offset))
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	260 return
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	261
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	262 self._addentry(file, offset)
23253 8d84b7a2dd91 transaction: factorise append-only file registration Pierre-Yves David <pierre-yves.david@fb.com> parents: 23252 diff changeset	263
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	264 def _addentry(self, file, offset):
23253 8d84b7a2dd91 transaction: factorise append-only file registration Pierre-Yves David <pierre-yves.david@fb.com> parents: 23252 diff changeset	265 """add a append-only entry to memory and on-disk state"""
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	266 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	267 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	268 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	269 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	270 ):
23253 8d84b7a2dd91 transaction: factorise append-only file registration Pierre-Yves David <pierre-yves.david@fb.com> parents: 23252 diff changeset	271 return
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	272 if offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	273 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	274 else:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	275 self._newfiles.add(file)
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	276 # add enough data to the journal to do the truncate
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	277 self._file.write(b"%s\0%d\n" % (file, offset))
39693 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39692 diff changeset	278 self._file.flush()
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	279
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	280 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	281 def addbackup(self, file, hardlink=True, location=b''):
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	282 """Adds a backup of the file to the transaction
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	283
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	284 Calling addbackup() creates a hardlink backup of the specified file
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	285 that is used to recover the file in the event of the transaction
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	286 aborting.
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	287
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	288 * `file`: the file path, relative to .hg/store
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	289 * `hardlink`: use a hardlink to quickly create the backup
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	290 """
23251 85c634ff395a transaction: drop backupentries logic from startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23250 diff changeset	291 if self._queue:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	292 msg = b'cannot use transaction.addbackup inside "group"'
31653 8defc7d9adae transaction: use ProgrammingError Jun Wu <quark@fb.com> parents: 30006 diff changeset	293 raise error.ProgrammingError(msg)
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	294
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	295 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	296 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	297 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	298 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	299 ):
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	300 return
23582 7559dc8c4238 vfs: add a 'split' method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23581 diff changeset	301 vfs = self._vfsmap[location]
7559dc8c4238 vfs: add a 'split' method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23581 diff changeset	302 dirname, filename = vfs.split(file)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	303 backupfilename = b"%s.backup.%s" % (self._journal, filename)
23581 aed981c7bebf vfs: add a 'reljoin' function for joining relative paths Pierre-Yves David <pierre-yves.david@fb.com> parents: 23543 diff changeset	304 backupfile = vfs.reljoin(dirname, backupfilename)
22663 4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	305 if vfs.exists(file):
4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	306 filepath = vfs.join(file)
23314 43f66ae57a66 addbackup: use the vfs for the backup destination too Pierre-Yves David <pierre-yves.david@fb.com> parents: 23313 diff changeset	307 backuppath = vfs.join(backupfile)
23900 5eb3541f907e transaction: use 'util.copyfile' for creating backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23764 diff changeset	308 util.copyfile(filepath, backuppath, hardlink=hardlink)
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	309 else:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	310 backupfile = b''
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	311
23316 fc3670f41d3e transaction: use 'location' instead of 'vfs' in the addbackup method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23315 diff changeset	312 self._addbackupentry((location, file, backupfile, False))
23283 b04263c38a92 transaction: extract backupentry registration in a dedicated function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23282 diff changeset	313
b04263c38a92 transaction: extract backupentry registration in a dedicated function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23282 diff changeset	314 def _addbackupentry(self, entry):
b04263c38a92 transaction: extract backupentry registration in a dedicated function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23282 diff changeset	315 """register a new backup entry and write it to disk"""
b04263c38a92 transaction: extract backupentry registration in a dedicated function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23282 diff changeset	316 self._backupentries.append(entry)
25294 b1b89a0a606d transaction: really fix _addbackupentry key usage (issue4684) Pierre-Yves David <pierre-yves.david@fb.com> parents: 25263 diff changeset	317 self._backupmap[entry[1]] = len(self._backupentries) - 1
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	318 self._backupsfile.write(b"%s\0%s\0%s\0%d\n" % entry)
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	319 self._backupsfile.flush()
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	320
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	321 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	322 def registertmp(self, tmpfile, location=b''):
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	323 """register a temporary transaction file
03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	324
23355 7faa55c20b0e transaction: fix some docstring grammar Matt Mackall <mpm@selenic.com> parents: 23354 diff changeset	325 Such files will be deleted when the transaction exits (on both
7faa55c20b0e transaction: fix some docstring grammar Matt Mackall <mpm@selenic.com> parents: 23354 diff changeset	326 failure and success).
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	327 """
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	328 self._addbackupentry((location, b'', tmpfile, False))
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	329
03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	330 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	331 def addfilegenerator(
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	332 self,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	333 genid,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	334 filenames,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	335 genfunc,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	336 order=0,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	337 location=b'',
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	338 post_finalize=False,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	339 ):
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	340 """add a function to generates some files at transaction commit
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	341
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	342 The `genfunc` argument is a function capable of generating proper
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	343 content of each entry in the `filename` tuple.
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	344
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	345 At transaction close time, `genfunc` will be called with one file
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	346 object argument per entries in `filenames`.
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	347
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	348 The transaction itself is responsible for the backup, creation and
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	349 final write of such file.
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	350
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	351 The `genid` argument is used to ensure the same set of file is only
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	352 generated once. Call to `addfilegenerator` for a `genid` already
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	353 present will overwrite the old entry.
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	354
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	355 The `order` argument may be used to control the order in which multiple
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	356 generator will be executed.
23317 197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	357
197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	358 The `location` arguments may be used to indicate the files are located
197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	359 outside of the the standard directory for transaction. It should match
23543 4dd8a6a1240d spelling: fixes from proofreading of spell checker issues Mads Kiilerich <madski@unity3d.com> parents: 23513 diff changeset	360 one of the key of the `transaction.vfsmap` dictionary.
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	361
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	362 The `post_finalize` argument can be set to `True` for file generation
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	363 that must be run after the transaction has been finalized.
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	364 """
22663 4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	365 # For now, we are unable to do proper backup and restore of custom vfs
4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	366 # but for bookmarks that are handled outside this mechanism.
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	367 entry = (order, filenames, genfunc, location, post_finalize)
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	368 self._filegenerators[genid] = entry
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	369
33068 2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	370 @active
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	371 def removefilegenerator(self, genid):
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	372 """reverse of addfilegenerator, remove a file generator function"""
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	373 if genid in self._filegenerators:
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	374 del self._filegenerators[genid]
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32591 diff changeset	375
44433 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	376 def _generatefiles(self, suffix=b'', group=GEN_GROUP_ALL):
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	377 # write files registered for generation
23357 ba033f461f00 transaction: have _generatefile return a boolean Pierre-Yves David <pierre-yves.david@fb.com> parents: 23356 diff changeset	378 any = False
44434 f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	379
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	380 if group == GEN_GROUP_ALL:
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	381 skip_post = skip_pre = False
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	382 else:
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	383 skip_pre = group == GEN_GROUP_POST_FINALIZE
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	384 skip_post = group == GEN_GROUP_PRE_FINALIZE
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	385
43106 d783f945a701 py3: finish porting iteritems() to pycompat and remove source transformer Gregory Szorc <gregory.szorc@gmail.com> parents: 43077 diff changeset	386 for id, entry in sorted(pycompat.iteritems(self._filegenerators)):
23357 ba033f461f00 transaction: have _generatefile return a boolean Pierre-Yves David <pierre-yves.david@fb.com> parents: 23356 diff changeset	387 any = True
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	388 order, filenames, genfunc, location, post_finalize = entry
28830 a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	389
a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	390 # for generation at closing, check if it's before or after finalize
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	391 if skip_post and post_finalize:
44434 f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44433 diff changeset	392 continue
48699 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48698 diff changeset	393 elif skip_pre and not post_finalize:
28830 a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	394 continue
a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	395
23317 197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	396 vfs = self._vfsmap[location]
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	397 files = []
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	398 try:
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	399 for name in filenames:
23356 140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	400 name += suffix
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	401 if suffix:
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	402 self.registertmp(name, location=location)
33279 7912404b70f2 transaction: apply checkambig=True only on limited files for similarity FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33278 diff changeset	403 checkambig = False
23356 140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	404 else:
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	405 self.addbackup(name, location=location)
39696 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39695 diff changeset	406 checkambig = (name, location) in self._checkambigfiles
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	407 files.append(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	408 vfs(name, b'w', atomictemp=True, checkambig=checkambig)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	409 )
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	410 genfunc(*files)
41110 3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40626 diff changeset	411 for f in files:
3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40626 diff changeset	412 f.close()
3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40626 diff changeset	413 # skip discard() loop since we're sure no open file remains
3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40626 diff changeset	414 del files[:]
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	415 finally:
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	416 for f in files:
41110 3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40626 diff changeset	417 f.discard()
23357 ba033f461f00 transaction: have _generatefile return a boolean Pierre-Yves David <pierre-yves.david@fb.com> parents: 23356 diff changeset	418 return any
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	419
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	420 @active
45885 a6f08085edfe transaction: rename find to findoffset and drop backup file support Joerg Sonnenberger <joerg@bec.de> parents: 45884 diff changeset	421 def findoffset(self, file):
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	422 if file in self._newfiles:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	423 return 0
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	424 return self._offsetmap.get(file)
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	425
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	426 @active
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	427 def readjournal(self):
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	428 self._file.seek(0)
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	429 entries = []
46067 88de2639901b transaction: windows workaround for missing line iteration support Joerg Sonnenberger <joerg@bec.de> parents: 45957 diff changeset	430 for l in self._file.readlines():
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	431 file, troffset = l.split(b'\0')
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	432 entries.append((file, int(troffset)))
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	433 return entries
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	434
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	435 @active
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	436 def replace(self, file, offset):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	437 """
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	438 replace can only replace already committed entries
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	439 that are not pending in the queue
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	440 """
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	441 if file in self._newfiles:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	442 if not offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	443 return
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	444 self._newfiles.remove(file)
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	445 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	446 elif file in self._offsetmap:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	447 if not offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	448 del self._offsetmap[file]
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	449 self._newfiles.add(file)
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	450 else:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	451 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	452 else:
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	453 raise KeyError(file)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	454 self._file.write(b"%s\0%d\n" % (file, offset))
39693 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39692 diff changeset	455 self._file.flush()
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	456
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	457 @active
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	458 def nest(self, name='<unnamed>'):
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	459 self._count += 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	460 self._usages += 1
39701 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39700 diff changeset	461 self._names.append(name)
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	462 return self
a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	463
11230 5116a077c3da make transactions work on non-refcounted python implementations Ronny Pfannschmidt <Ronny.Pfannschmidt@gmx.de> parents: 10282 diff changeset	464 def release(self):
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	465 if self._count > 0:
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	466 self._usages -= 1
39701 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39700 diff changeset	467 if self._names:
4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39700 diff changeset	468 self._names.pop()
11685 aade8f133d11 cleanup: typos Patrick Mezard <pmezard@gmail.com> parents: 11230 diff changeset	469 # if the transaction scopes are left without being closed, fail
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	470 if self._count > 0 and self._usages == 0:
11230 5116a077c3da make transactions work on non-refcounted python implementations Ronny Pfannschmidt <Ronny.Pfannschmidt@gmx.de> parents: 10282 diff changeset	471 self._abort()
5116a077c3da make transactions work on non-refcounted python implementations Ronny Pfannschmidt <Ronny.Pfannschmidt@gmx.de> parents: 10282 diff changeset	472
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	473 def running(self):
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	474 return self._count > 0
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	475
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	476 def addpending(self, category, callback):
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	477 """add a callback to be called when the transaction is pending
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	478
23280 b01c491af0cf transaction: pass the transaction to 'pending' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23279 diff changeset	479 The transaction will be given as callback's first argument.
b01c491af0cf transaction: pass the transaction to 'pending' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23279 diff changeset	480
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	481 Category is a unique identifier to allow overwriting an old callback
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	482 with a newer callback.
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	483 """
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	484 self._pendingcallback[category] = callback
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	485
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	486 @active
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	487 def writepending(self):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	488 """write pending file to temporary version
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	489
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	490 This is used to allow hooks to view a transaction before commit"""
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	491 categories = sorted(self._pendingcallback)
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	492 for cat in categories:
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	493 # remove callback since the data will have been flushed
23280 b01c491af0cf transaction: pass the transaction to 'pending' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23279 diff changeset	494 any = self._pendingcallback.pop(cat)(self)
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	495 self._anypending = self._anypending or any
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	496 self._anypending \|= self._generatefiles(suffix=b'.pending')
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	497 return self._anypending
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	498
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	499 @active
44056 8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43554 diff changeset	500 def hasfinalize(self, category):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	501 """check is a callback already exist for a category"""
44056 8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43554 diff changeset	502 return category in self._finalizecallback
8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43554 diff changeset	503
8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43554 diff changeset	504 @active
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	505 def addfinalize(self, category, callback):
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	506 """add a callback to be called when the transaction is closed
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	507
23281 f60ed8cf4afc transaction: pass the transaction to 'finalize' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23280 diff changeset	508 The transaction will be given as callback's first argument.
f60ed8cf4afc transaction: pass the transaction to 'finalize' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23280 diff changeset	509
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	510 Category is a unique identifier to allow overwriting old callbacks with
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	511 newer callbacks.
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	512 """
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	513 self._finalizecallback[category] = callback
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	514
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	515 @active
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	516 def addpostclose(self, category, callback):
33099 fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	517 """add or replace a callback to be called after the transaction closed
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	518
23282 6c1351352b6c transaction: pass the transaction to 'postclose' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23281 diff changeset	519 The transaction will be given as callback's first argument.
6c1351352b6c transaction: pass the transaction to 'postclose' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23281 diff changeset	520
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	521 Category is a unique identifier to allow overwriting an old callback
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	522 with a newer callback.
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	523 """
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	524 self._postclosecallback[category] = callback
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	525
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	526 @active
33099 fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	527 def getpostclose(self, category):
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	528 """return a postclose callback added before, or None"""
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	529 return self._postclosecallback.get(category, None)
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	530
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33068 diff changeset	531 @active
23764 d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	532 def addabort(self, category, callback):
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	533 """add a callback to be called when the transaction is aborted.
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	534
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	535 The transaction will be given as the first argument to the callback.
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	536
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	537 Category is a unique identifier to allow overwriting an old callback
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	538 with a newer callback.
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	539 """
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	540 self._abortcallback[category] = callback
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	541
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	542 @active
44553 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	543 def addvalidator(self, category, callback):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	544 """adds a callback to be called when validating the transaction.
44553 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	545
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	546 The transaction will be given as the first argument to the callback.
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	547
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	548 callback should raise exception if to abort transaction"""
44553 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	549 self._validatecallback[category] = callback
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	550
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	551 @active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	552 def close(self):
9220 8a4da1388553 transaction: document close(), abort() methods Greg Ward <greg-hg@gerg.ca> parents: 9094 diff changeset	553 '''commit the transaction'''
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	554 if self._count == 1:
44553 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	555 for category in sorted(self._validatecallback):
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	556 self._validatecallback[category](self)
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44434 diff changeset	557 self._validatecallback = None # Help prevent cycles.
44433 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	558 self._generatefiles(group=GEN_GROUP_PRE_FINALIZE)
44097 2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	559 while self._finalizecallback:
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	560 callbacks = self._finalizecallback
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	561 self._finalizecallback = {}
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	562 categories = sorted(callbacks)
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	563 for cat in categories:
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	564 callbacks[cat](self)
28960 14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	565 # Prevent double usage and help clear cycles.
14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	566 self._finalizecallback = None
44433 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	567 self._generatefiles(group=GEN_GROUP_POST_FINALIZE)
20881 3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	568
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	569 self._count -= 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	570 if self._count != 0:
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	571 return
39693 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39692 diff changeset	572 self._file.close()
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	573 self._backupsfile.close()
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	574 # cleanup temporary files
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	575 for l, f, b, c in self._backupentries:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	576 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	577 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	578 b"couldn't remove %s: unknown cache location %s\n" % (b, l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	579 )
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	580 continue
23311 64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	581 vfs = self._vfsmap[l]
64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	582 if not f and b and vfs.exists(b):
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	583 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	584 vfs.unlink(b)
26587 56b2bcea2529 error: get Abort from 'error' instead of 'util' Pierre-Yves David <pierre-yves.david@fb.com> parents: 26576 diff changeset	585 except (IOError, OSError, error.Abort) as inst:
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	586 if not c:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	587 raise
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	588 # Abort may be raise by read only opener
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	589 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	590 b"couldn't remove %s: %s\n" % (vfs.join(b), inst)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	591 )
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	592 self._offsetmap = {}
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	593 self._newfiles = set()
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	594 self._writeundo()
39697 0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39696 diff changeset	595 if self._after:
0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39696 diff changeset	596 self._after()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	597 self._after = None # Help prevent cycles.
39698 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	598 if self._opener.isfile(self._backupjournal):
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	599 self._opener.unlink(self._backupjournal)
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	600 if self._opener.isfile(self._journal):
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	601 self._opener.unlink(self._journal)
27662 f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	602 for l, _f, b, c in self._backupentries:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	603 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	604 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	605 b"couldn't remove %s: unknown cache location"
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	606 b"%s\n" % (b, l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	607 )
27662 f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	608 continue
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	609 vfs = self._vfsmap[l]
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	610 if b and vfs.exists(b):
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	611 try:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	612 vfs.unlink(b)
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	613 except (IOError, OSError, error.Abort) as inst:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	614 if not c:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	615 raise
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	616 # Abort may be raise by read only opener
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	617 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	618 b"couldn't remove %s: %s\n" % (vfs.join(b), inst)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	619 )
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	620 self._backupentries = []
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	621 self._journal = None
26576 9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	622
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	623 self._releasefn(self, True) # notify success of closing transaction
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	624 self._releasefn = None # Help prevent cycles.
26576 9e0aaac0d9eb transaction: add releasefn to notify the end of a transaction scope FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 25986 diff changeset	625
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	626 # run post close action
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	627 categories = sorted(self._postclosecallback)
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	628 for cat in categories:
23282 6c1351352b6c transaction: pass the transaction to 'postclose' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23281 diff changeset	629 self._postclosecallback[cat](self)
28960 14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	630 # Prevent double usage and help clear cycles.
14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	631 self._postclosecallback = None
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	632
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	633 @active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	634 def abort(self):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	635 """abort the transaction (generally called on error, or when the
9220 8a4da1388553 transaction: document close(), abort() methods Greg Ward <greg-hg@gerg.ca> parents: 9094 diff changeset	636 transaction is not explicitly committed before going out of
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45887 diff changeset	637 scope)"""
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	638 self._abort()
fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	639
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	640 def _writeundo(self):
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	641 """write transaction data for possible future undo call"""
39691 da9ce63bfa9b transaction: make undoname a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39690 diff changeset	642 if self._undoname is None:
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	643 return
47428 88439c6fbafc transaction: simplify `undo.backupfiles` file creation with a variable Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47426 diff changeset	644
88439c6fbafc transaction: simplify `undo.backupfiles` file creation with a variable Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47426 diff changeset	645 undo_backup_path = b"%s.backupfiles" % self._undoname
88439c6fbafc transaction: simplify `undo.backupfiles` file creation with a variable Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47426 diff changeset	646 undobackupfile = self._opener.open(undo_backup_path, b'w')
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	647 undobackupfile.write(b'%d\n' % version)
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	648 for l, f, b, c in self._backupentries:
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	649 if not f: # temporary file
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	650 continue
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	651 if not b:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	652 u = b''
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	653 else:
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	654 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	655 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	656 b"couldn't remove %s: unknown cache location"
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	657 b"%s\n" % (b, l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	658 )
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	659 continue
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	660 vfs = self._vfsmap[l]
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	661 base, name = vfs.split(b)
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	662 assert name.startswith(self._journal), name
77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	663 uname = name.replace(self._journal, self._undoname, 1)
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	664 u = vfs.reljoin(base, uname)
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	665 util.copyfile(vfs.join(b), vfs.join(u), hardlink=True)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	666 undobackupfile.write(b"%s\0%s\0%s\0%d\n" % (l, f, u, c))
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	667 undobackupfile.close()
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	668
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	669 def _abort(self):
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	670 entries = self.readjournal()
39690 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	671 self._count = 0
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36827 diff changeset	672 self._usages = 0
39693 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39692 diff changeset	673 self._file.close()
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	674 self._backupsfile.close()
8290 560af1bbfd6e transaction: reset transaction on abort Henrik Stuart <hg@hstuart.dk> parents: 8289 diff changeset	675
10228 056c366fea8c transaction: initialize self.journal to None after deletion Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 9693 diff changeset	676 try:
45887 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45886 diff changeset	677 if not entries and not self._backupentries:
26753 96dd93de548c transaction: reorder unlinking .hg/journal and .hg/journal.backupfiles FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 26587 diff changeset	678 if self._backupjournal:
39698 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	679 self._opener.unlink(self._backupjournal)
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	680 if self._journal:
39698 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39697 diff changeset	681 self._opener.unlink(self._journal)
10228 056c366fea8c transaction: initialize self.journal to None after deletion Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 9693 diff changeset	682 return
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	683
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	684 self._report(_(b"transaction abort!\n"))
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	685
108 8d55c2d72c7c Warn if we fail to truncate something mpm@selenic.com parents: 95 diff changeset	686 try:
23764 d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	687 for cat in sorted(self._abortcallback):
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	688 self._abortcallback[cat](self)
28960 14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	689 # Prevent double usage and help clear cycles.
14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	690 self._abortcallback = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	691 _playback(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	692 self._journal,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	693 self._report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	694 self._opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	695 self._vfsmap,
45886 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45885 diff changeset	696 entries,
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	697 self._backupentries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	698 False,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	699 checkambigfiles=self._checkambigfiles,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	700 )
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	701 self._report(_(b"rollback completed\n"))
40626 aca09df32819 transaction: display data about why the transaction failed to rollback Boris Feld <boris.feld@octobus.net> parents: 39702 diff changeset	702 except BaseException as exc:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	703 self._report(_(b"rollback failed - please run hg recover\n"))
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	704 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	705 _(b"(failure reason: %s)\n") % stringutil.forcebytestr(exc)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	706 )
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	707 finally:
39692 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39691 diff changeset	708 self._journal = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	709 self._releasefn(self, False) # notify failure of transaction
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	710 self._releasefn = None # Help prevent cycles.
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	711
8290 560af1bbfd6e transaction: reset transaction on abort Henrik Stuart <hg@hstuart.dk> parents: 8289 diff changeset	712
47425 0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47424 diff changeset	713 BAD_VERSION_MSG = _(
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47424 diff changeset	714 b"journal was created by a different version of Mercurial\n"
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47424 diff changeset	715 )
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47424 diff changeset	716
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47424 diff changeset	717
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	718 def rollback(opener, vfsmap, file, report, checkambigfiles=None):
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	719 """Rolls back the transaction contained in the given file
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	720
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	721 Reads the entries in the specified file, and the corresponding
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	722 '*.backupfiles' file, to recover from an incomplete transaction.
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	723
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	724 * `file`: a file containing a list of entries, specifying where
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	725 to truncate each file. The file should contain a list of
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	726 file\0offset pairs, delimited by newlines. The corresponding
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	727 '*.backupfiles' file should contain a list of file\0backupfile
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	728 pairs, delimited by \0.
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	729
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	730 `checkambigfiles` is a set of (path, vfs-location) tuples,
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	731 which determine whether file stat ambiguity should be avoided at
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33099 diff changeset	732 restoring corresponded files.
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	733 """
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	734 entries = []
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	735 backupentries = []
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	736
47318 13dd5bb5492a transaction: trivial refactoring Valentin Gatien-Baron <valentin.gatienbaron@gmail.com> parents: 47297 diff changeset	737 with opener.open(file) as fp:
13dd5bb5492a transaction: trivial refactoring Valentin Gatien-Baron <valentin.gatienbaron@gmail.com> parents: 47297 diff changeset	738 lines = fp.readlines()
13400 14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	739 for l in lines:
20524 28b8ff84db3f journal: report parsing errors on recover/rollback (issue4172) Matt Mackall <mpm@selenic.com> parents: 20087 diff changeset	740 try:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	741 f, o = l.split(b'\0')
45884 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	742 entries.append((f, int(o)))
20524 28b8ff84db3f journal: report parsing errors on recover/rollback (issue4172) Matt Mackall <mpm@selenic.com> parents: 20087 diff changeset	743 except ValueError:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	744 report(
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	745 _(b"couldn't read journal entry %r!\n") % pycompat.bytestr(l)
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	746 )
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	747
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	748 backupjournal = b"%s.backupfiles" % file
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	749 if opener.exists(backupjournal):
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	750 fp = opener.open(backupjournal)
23065 963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	751 lines = fp.readlines()
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	752 if lines:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	753 ver = lines[0][:-1]
47426 e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47425 diff changeset	754 if ver != (b'%d' % version):
e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47425 diff changeset	755 report(BAD_VERSION_MSG)
e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47425 diff changeset	756 else:
23065 963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	757 for line in lines[1:]:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	758 if line:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	759 # Shave off the trailing newline
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	760 line = line[:-1]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	761 l, f, b, c = line.split(b'\0')
23309 7eb520f5efe4 transaction: change the on disk format for backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23291 diff changeset	762 backupentries.append((l, f, b, bool(c)))
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	763
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	764 _playback(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	765 file,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	766 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	767 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	768 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	769 entries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	770 backupentries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	771 checkambigfiles=checkambigfiles,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	772 )

Mercurial > public > mercurial-scm > hg-stable

annotate mercurial/transaction.py @ 48813:2bb75c65fa6c stable