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

annotate mercurial/transaction.py @ 49995:27fd12eca557

transaction: quietly rollback if no other changes than temporary files If no actual change have been made, we don't really need to roll them back. We only have to cleanup some temporary files and it seems reasonable to do that quietly. This will help us to use the transaction in wider context? without impacting the user experience. [1] as in Python context managers that lives longer.

author	Pierre-Yves David <pierre-yves.david@octobus.net>
date	Tue, 14 Feb 2023 20:09:39 +0100
parents	3128018e878b
children	c8f32aa80dca

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: 46063 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
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	15 from .i18n import _
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	16 from . import (
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	17 error,
36735 ef345f9e4295 transaction: fix an error string with bytestr() on a repr()d value Augie Fackler <augie@google.com> parents: 35832 diff changeset	18 pycompat,
25986 89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	19 util,
89049011f304 transaction: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25660 diff changeset	20 )
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	21 from .utils import stringutil
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	22
23313 991098579940 transaction: set backupentries version to proper value Pierre-Yves David <pierre-yves.david@fb.com> parents: 23312 diff changeset	23 version = 2
23064 5dc888b79e70 transactions: add version number to journal.backupfiles Durham Goode <durham@fb.com> parents: 23063 diff changeset	24
44406 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	25 GEN_GROUP_ALL = b'all'
baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	26 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	27 GEN_GROUP_POST_FINALIZE = b'postfinalize'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	28
28830 a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	29
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	30 def active(func):
fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	31 def _active(self, args, *kwds):
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	32 if self._count == 0:
45738 5df1655edf42 transaction: use ProgrammingError for when an committed transaction is used Martin von Zweigbergk <martinvonz@google.com> parents: 44543 diff changeset	33 raise error.ProgrammingError(
5df1655edf42 transaction: use ProgrammingError for when an committed transaction is used Martin von Zweigbergk <martinvonz@google.com> parents: 44543 diff changeset	34 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	35 )
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	36 return func(self, args, *kwds)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	37
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	38 return _active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	39
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	40
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	41 def _playback(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	42 journal,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	43 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	44 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	45 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	46 entries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	47 backupentries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	48 unlink=True,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	49 checkambigfiles=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	50 ):
47286 18415fc918a1 recover: only apply last journal record per file (issue6423) Joerg Sonnenberger <joerg@bec.de> parents: 46819 diff changeset	51 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	52 if o or not unlink:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	53 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	54 try:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	55 fp = opener(f, b'a', checkambig=checkambig)
42963 8502f76dbfd7 transaction: detect an attempt to truncate-to-extend on playback, raise error Kyle Lippincott <spectral@google.com> parents: 41365 diff changeset	56 if fp.tell() < o:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	57 raise error.Abort(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	58 _(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	59 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	60 b"already %d bytes\n"
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	61 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	62 % (f, o, fp.tell())
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	63 )
13400 14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	64 fp.truncate(o)
14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	65 fp.close()
9686 ddf2adf88b89 transaction: more specific exceptions, os.unlink can raise OSError Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 9220 diff changeset	66 except IOError:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	67 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	68 raise
48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	69 else:
48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	70 try:
20084 a3378a1b0a05 transaction: unlink target file via vfs FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 17424 diff changeset	71 opener.unlink(f)
49306 2e726c934fcd py3: catch FileNotFoundError instead of checking errno == ENOENT Manuel Jacob <me@manueljacob.de> parents: 48913 diff changeset	72 except FileNotFoundError:
2e726c934fcd py3: catch FileNotFoundError instead of checking errno == ENOENT Manuel Jacob <me@manueljacob.de> parents: 48913 diff changeset	73 pass
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	74
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	75 backupfiles = []
23309 7eb520f5efe4 transaction: change the on disk format for backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23291 diff changeset	76 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	77 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	78 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	79 vfs = vfsmap[l]
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	80 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	81 if f and b:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	82 filepath = vfs.join(f)
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	83 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	84 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	85 try:
33279 7912404b70f2 transaction: apply checkambig=True only on limited files for similarity FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33278 diff changeset	86 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	87 backupfiles.append(b)
47418 22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47307 diff changeset	88 except IOError as exc:
22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47307 diff changeset	89 e_msg = stringutil.forcebytestr(exc)
22e21deed474 transaction: explain why some recovery failed Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47307 diff changeset	90 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	91 else:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	92 target = f or b
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	93 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	94 vfs.unlink(target)
49306 2e726c934fcd py3: catch FileNotFoundError instead of checking errno == ENOENT Manuel Jacob <me@manueljacob.de> parents: 48913 diff changeset	95 except FileNotFoundError:
2e726c934fcd py3: catch FileNotFoundError instead of checking errno == ENOENT Manuel Jacob <me@manueljacob.de> parents: 48913 diff changeset	96 pass
41365 876494fd967d cleanup: delete lots of unused local variables Martin von Zweigbergk <martinvonz@google.com> parents: 41107 diff changeset	97 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	98 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	99 raise
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	100
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	101 backuppath = b"%s.backupfiles" % journal
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	102 if opener.exists(backuppath):
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	103 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	104 opener.unlink(journal)
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	105 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	106 for f in backupfiles:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	107 if opener.exists(f):
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	108 opener.unlink(f)
41365 876494fd967d cleanup: delete lots of unused local variables Martin von Zweigbergk <martinvonz@google.com> parents: 41107 diff changeset	109 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	110 # 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	111 pass
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	112
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	113
33793 bbbbd3c30bfc util: add base class for transactional context managers Martin von Zweigbergk <martinvonz@google.com> parents: 33279 diff changeset	114 class transaction(util.transactional):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	115 def __init__(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	116 self,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	117 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	118 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	119 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	120 journalname,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	121 undoname=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	122 after=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	123 createmode=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	124 validator=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	125 releasefn=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	126 checkambigfiles=None,
43506 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	127 name='<unnamed>',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	128 ):
20881 3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	129 """Begin a new transaction
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	130
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	131 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	132 an exception.
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	133
3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	134 * `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	135 * `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	136 * `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: 33087 diff changeset	137
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	138 `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: 33087 diff changeset	139 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: 33087 diff changeset	140 for corresponded files.
20881 3c47677a8d04 transaction: add onclose/onabort hook for pre-close logic Durham Goode <durham@fb.com> parents: 20524 diff changeset	141 """
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	142 self._count = 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	143 self._usages = 1
39683 337d6e0fd9c9 transaction: make report a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39682 diff changeset	144 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	145 # a vfs to the store content
39682 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39681 diff changeset	146 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	147 # 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	148 vfsmap = vfsmap.copy()
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	149 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	150 self._vfsmap = vfsmap
39681 0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39680 diff changeset	151 self._after = after
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	152 self._offsetmap = {}
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	153 self._newfiles = set()
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	154 self._journal = journalname
39675 da9ce63bfa9b transaction: make undoname a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39674 diff changeset	155 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	156 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	157 # 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	158 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	159 releasefn = lambda tr, success: None
39678 040007cd3d81 transaction: make releasefn a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39677 diff changeset	160 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	161
39680 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39679 diff changeset	162 self._checkambigfiles = set()
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	163 if checkambigfiles:
39680 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39679 diff changeset	164 self._checkambigfiles.update(checkambigfiles)
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	165
39685 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39684 diff changeset	166 self._names = [name]
36819 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36735 diff changeset	167
32261 976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31648 diff changeset	168 # 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: 31648 diff changeset	169 # transaction.
976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31648 diff changeset	170 self.changes = {}
976681123416 transaction: introduce "changes" dictionary to precisely track updates Pierre-Yves David <pierre-yves.david@ens-lyon.org> parents: 31648 diff changeset	171
23279 e245775f8fd3 transaction: gather backupjournal logic together in the __init__ Pierre-Yves David <pierre-yves.david@fb.com> parents: 23278 diff changeset	172 # 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	173 self.hookargs = {}
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	174 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	175
23309 7eb520f5efe4 transaction: change the on disk format for backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23291 diff changeset	176 # 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	177 # - 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	178 # - 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	179 # - 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	180 # 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	181 # (cache is currently unused)
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	182 self._backupentries = []
84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	183 self._backupmap = {}
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	184 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	185 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	186 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	187
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	188 if createmode is not None:
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	189 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	190 opener.chmod(self._backupjournal, createmode & 0o666)
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	191
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	192 # 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	193 self._filegenerators = {}
23543 4dd8a6a1240d spelling: fixes from proofreading of spell checker issues Mads Kiilerich <madski@unity3d.com> parents: 23513 diff changeset	194 # 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	195 self._pendingcallback = {}
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	196 # 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	197 self._anypending = False
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	198 # 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	199 self._finalizecallback = {}
44543 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	200 # holds callback to call when validating the transaction
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	201 # should raise exception if anything is wrong
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	202 self._validatecallback = {}
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	203 if validator is not None:
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	204 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	205 # 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	206 self._postclosecallback = {}
23764 d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	207 # holds callbacks to call during abort
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	208 self._abortcallback = {}
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	209
36819 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36735 diff changeset	210 def __repr__(self):
43506 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	211 name = '/'.join(self._names)
9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	212 return '<transaction name=%s, count=%d, usages=%d>' % (
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	213 name,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	214 self._count,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	215 self._usages,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	216 )
36819 aff5996f3043 transaction: add a name and a __repr__ implementation (API) Martin von Zweigbergk <martinvonz@google.com> parents: 36735 diff changeset	217
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	218 def __del__(self):
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	219 if self._journal:
9693 c40a1ee20aa5 transaction: always remove empty journal on abort Sune Foldager <cryo@cyanite.org> parents: 9686 diff changeset	220 self._abort()
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	221
48676 3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47422 diff changeset	222 @property
3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47422 diff changeset	223 def finalized(self):
3f618484eeb6 transaction: add a way to know a transaction has been finalized Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47422 diff changeset	224 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: 47422 diff changeset	225
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	226 @active
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	227 def startgroup(self):
23250 8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	228 """delay registration of file entry
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	229
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	230 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	231 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	232 self._queue.append([])
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	233
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	234 @active
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	235 def endgroup(self):
23250 8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	236 """apply delayed registration of file entry.
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	237
8919dc7f2dbb transaction: document startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23249 diff changeset	238 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	239 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	240 q = self._queue.pop()
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	241 for f, o in q:
63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	242 self._addentry(f, o)
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	243
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	244 @active
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	245 def add(self, file, offset):
23252 70809438c644 transaction: document `tr.add` Pierre-Yves David <pierre-yves.david@fb.com> parents: 23251 diff changeset	246 """record the state of an append-only file before update"""
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	247 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	248 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	249 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	250 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	251 ):
10282 08a0f04b56bd many, many trivial check-code fixups Matt Mackall <mpm@selenic.com> parents: 10263 diff changeset	252 return
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	253 if self._queue:
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	254 self._queue[-1].append((file, offset))
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	255 return
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	256
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	257 self._addentry(file, offset)
23253 8d84b7a2dd91 transaction: factorise append-only file registration Pierre-Yves David <pierre-yves.david@fb.com> parents: 23252 diff changeset	258
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	259 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	260 """add a append-only entry to memory and on-disk state"""
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	261 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	262 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	263 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	264 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	265 ):
23253 8d84b7a2dd91 transaction: factorise append-only file registration Pierre-Yves David <pierre-yves.david@fb.com> parents: 23252 diff changeset	266 return
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	267 if offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	268 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	269 else:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	270 self._newfiles.add(file)
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	271 # 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	272 self._file.write(b"%s\0%d\n" % (file, offset))
39677 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39676 diff changeset	273 self._file.flush()
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	274
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	275 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	276 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	277 """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	278
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	279 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	280 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	281 aborting.
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	282
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	283 * `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	284 * `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	285 """
23251 85c634ff395a transaction: drop backupentries logic from startgroup and endgroup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23250 diff changeset	286 if self._queue:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	287 msg = b'cannot use transaction.addbackup inside "group"'
31648 8defc7d9adae transaction: use ProgrammingError Jun Wu <quark@fb.com> parents: 30002 diff changeset	288 raise error.ProgrammingError(msg)
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	289
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	290 if (
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	291 file in self._newfiles
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	292 or file in self._offsetmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	293 or file in self._backupmap
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	294 ):
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	295 return
23582 7559dc8c4238 vfs: add a 'split' method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23581 diff changeset	296 vfs = self._vfsmap[location]
7559dc8c4238 vfs: add a 'split' method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23581 diff changeset	297 dirname, filename = vfs.split(file)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	298 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	299 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	300 if vfs.exists(file):
4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	301 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	302 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	303 util.copyfile(filepath, backuppath, hardlink=hardlink)
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	304 else:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	305 backupfile = b''
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	306
23316 fc3670f41d3e transaction: use 'location' instead of 'vfs' in the addbackup method Pierre-Yves David <pierre-yves.david@fb.com> parents: 23315 diff changeset	307 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	308
b04263c38a92 transaction: extract backupentry registration in a dedicated function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23282 diff changeset	309 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	310 """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	311 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	312 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	313 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	314 self._backupsfile.flush()
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	315
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	316 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	317 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	318 """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	319
23355 7faa55c20b0e transaction: fix some docstring grammar Matt Mackall <mpm@selenic.com> parents: 23354 diff changeset	320 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	321 failure and success).
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	322 """
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	323 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	324
03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	325 @active
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	326 def addfilegenerator(
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	327 self,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	328 genid,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	329 filenames,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	330 genfunc,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	331 order=0,
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	332 location=b'',
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	333 post_finalize=False,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	334 ):
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	335 """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	336
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	337 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	338 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	339
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	340 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	341 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	342
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	343 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	344 final write of such file.
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	345
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	346 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	347 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	348 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	349
feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	350 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	351 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	352
197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	353 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	354 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	355 one of the key of the `transaction.vfsmap` dictionary.
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	356
21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	357 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: 48684 diff changeset	358 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	359 """
22663 4c6198737ad8 transaction: allow generating file outside of store Pierre-Yves David <pierre-yves.david@fb.com> parents: 22662 diff changeset	360 # 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	361 # but for bookmarks that are handled outside this mechanism.
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	362 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: 48684 diff changeset	363 self._filegenerators[genid] = entry
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	364
33056 2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	365 @active
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	366 def removefilegenerator(self, genid):
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	367 """reverse of addfilegenerator, remove a file generator function"""
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	368 if genid in self._filegenerators:
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	369 del self._filegenerators[genid]
2312e70cf78b rebase: clean up rebasestate from active transaction Jun Wu <quark@fb.com> parents: 32558 diff changeset	370
44406 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	371 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	372 # 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	373 any = False
44407 f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	374
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	375 if group == GEN_GROUP_ALL:
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	376 skip_post = skip_pre = False
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	377 else:
f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	378 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: 44406 diff changeset	379 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: 44406 diff changeset	380
48913 f254fc73d956 global: bulk replace simple pycompat.iteritems(x) with x.items() Gregory Szorc <gregory.szorc@gmail.com> parents: 48875 diff changeset	381 for id, entry in sorted(self._filegenerators.items()):
23357 ba033f461f00 transaction: have _generatefile return a boolean Pierre-Yves David <pierre-yves.david@fb.com> parents: 23356 diff changeset	382 any = True
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	383 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	384
a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	385 # for generation at closing, check if it's before or after finalize
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	386 if skip_post and post_finalize:
44407 f6798c1a80fa transaction: clarify the logic around pre-finalize/post-finalize Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44406 diff changeset	387 continue
48685 21ac6aedd5e5 transaction: do not rely on a global variable to post_finalize file Pierre-Yves David <pierre-yves.david@octobus.net> parents: 48684 diff changeset	388 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	389 continue
a5009789960c transaction: allow running file generators after finalizers Durham Goode <durham@fb.com> parents: 27924 diff changeset	390
23317 197e17be5407 transaction: use 'location' instead of 'vfs' objects for file generation Pierre-Yves David <pierre-yves.david@fb.com> parents: 23316 diff changeset	391 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	392 files = []
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	393 try:
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	394 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	395 name += suffix
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	396 if suffix:
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	397 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	398 checkambig = False
23356 140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	399 else:
140c21fbf4eb transaction: allow generating files with a suffix Pierre-Yves David <pierre-yves.david@fb.com> parents: 23355 diff changeset	400 self.addbackup(name, location=location)
39680 264d56954dda transaction: make checkambigfiles a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39679 diff changeset	401 checkambig = (name, location) in self._checkambigfiles
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	402 files.append(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	403 vfs(name, b'w', atomictemp=True, checkambig=checkambig)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	404 )
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	405 genfunc(*files)
41107 3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40579 diff changeset	406 for f in files:
3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40579 diff changeset	407 f.close()
3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40579 diff changeset	408 # 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: 40579 diff changeset	409 del files[:]
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	410 finally:
16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	411 for f in files:
41107 3e2c02836420 transaction: do not overwrite atomic-temp files on error Yuya Nishihara <yuya@tcha.org> parents: 40579 diff changeset	412 f.discard()
23357 ba033f461f00 transaction: have _generatefile return a boolean Pierre-Yves David <pierre-yves.david@fb.com> parents: 23356 diff changeset	413 return any
23102 16da812ad970 transaction: extract file generation into its own function Pierre-Yves David <pierre-yves.david@fb.com> parents: 23065 diff changeset	414
22078 feb4797c676e transaction: add a file generation mechanism Pierre-Yves David <pierre-yves.david@fb.com> parents: 22077 diff changeset	415 @active
45870 a6f08085edfe transaction: rename find to findoffset and drop backup file support Joerg Sonnenberger <joerg@bec.de> parents: 45869 diff changeset	416 def findoffset(self, file):
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	417 if file in self._newfiles:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	418 return 0
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	419 return self._offsetmap.get(file)
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	420
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	421 @active
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	422 def readjournal(self):
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	423 self._file.seek(0)
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	424 entries = []
46063 88de2639901b transaction: windows workaround for missing line iteration support Joerg Sonnenberger <joerg@bec.de> parents: 45942 diff changeset	425 for l in self._file.readlines():
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	426 file, troffset = l.split(b'\0')
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	427 entries.append((file, int(troffset)))
a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	428 return entries
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	429
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	430 @active
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	431 def replace(self, file, offset):
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	432 """
8363 c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	433 replace can only replace already committed entries
c8e81f557da7 transaction: add atomic groups to transaction logic Henrik Stuart <hg@hstuart.dk> parents: 8294 diff changeset	434 that are not pending in the queue
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	435 """
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	436 if file in self._newfiles:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	437 if not offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	438 return
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	439 self._newfiles.remove(file)
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	440 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	441 elif file in self._offsetmap:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	442 if not offset:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	443 del self._offsetmap[file]
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	444 self._newfiles.add(file)
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	445 else:
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	446 self._offsetmap[file] = offset
ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	447 else:
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	448 raise KeyError(file)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	449 self._file.write(b"%s\0%d\n" % (file, offset))
39677 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39676 diff changeset	450 self._file.flush()
2084 d66278012853 Chris Mason <mason@suse.com> parents: 1806 diff changeset	451
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	452 @active
43506 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43106 diff changeset	453 def nest(self, name='<unnamed>'):
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	454 self._count += 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	455 self._usages += 1
39685 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39684 diff changeset	456 self._names.append(name)
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	457 return self
a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	458
11230 5116a077c3da make transactions work on non-refcounted python implementations Ronny Pfannschmidt <Ronny.Pfannschmidt@gmx.de> parents: 10282 diff changeset	459 def release(self):
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	460 if self._count > 0:
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	461 self._usages -= 1
39685 4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39684 diff changeset	462 if self._names:
4024c363cd33 transaction: make names a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39684 diff changeset	463 self._names.pop()
11685 aade8f133d11 cleanup: typos Patrick Mezard <pmezard@gmail.com> parents: 11230 diff changeset	464 # if the transaction scopes are left without being closed, fail
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	465 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	466 self._abort()
5116a077c3da make transactions work on non-refcounted python implementations Ronny Pfannschmidt <Ronny.Pfannschmidt@gmx.de> parents: 10282 diff changeset	467
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	468 def running(self):
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	469 return self._count > 0
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	470
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	471 def addpending(self, category, callback):
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	472 """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	473
23280 b01c491af0cf transaction: pass the transaction to 'pending' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23279 diff changeset	474 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	475
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	476 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	477 with a newer callback.
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	478 """
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	479 self._pendingcallback[category] = callback
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	480
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	481 @active
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	482 def writepending(self):
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	483 """write pending file to temporary version
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	484
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	485 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	486 categories = sorted(self._pendingcallback)
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	487 for cat in categories:
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	488 # 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	489 any = self._pendingcallback.pop(cat)(self)
23202 ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	490 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	491 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	492 return self._anypending
ea5af863fbff transaction: add 'writepending' logic Pierre-Yves David <pierre-yves.david@fb.com> parents: 23103 diff changeset	493
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	494 @active
44056 8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43506 diff changeset	495 def hasfinalize(self, category):
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	496 """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: 43506 diff changeset	497 return category in self._finalizecallback
8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43506 diff changeset	498
8e09551206f5 transaction: add a `hasfinalize` method Pierre-Yves David <pierre-yves.david@octobus.net> parents: 43506 diff changeset	499 @active
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	500 def addfinalize(self, category, callback):
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	501 """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	502
23281 f60ed8cf4afc transaction: pass the transaction to 'finalize' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23280 diff changeset	503 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	504
23204 10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	505 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	506 newer callbacks.
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	507 """
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	508 self._finalizecallback[category] = callback
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	509
10beda5bd2b7 transaction: allow registering a finalization callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23202 diff changeset	510 @active
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	511 def addpostclose(self, category, callback):
33087 fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33056 diff changeset	512 """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	513
23282 6c1351352b6c transaction: pass the transaction to 'postclose' callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23281 diff changeset	514 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	515
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	516 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	517 with a newer callback.
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	518 """
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	519 self._postclosecallback[category] = callback
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	520
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	521 @active
33087 fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33056 diff changeset	522 def getpostclose(self, category):
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33056 diff changeset	523 """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: 33056 diff changeset	524 return self._postclosecallback.get(category, None)
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33056 diff changeset	525
fcd1c483f5ea strip: add a delayedstrip method that works in a transaction Jun Wu <quark@fb.com> parents: 33056 diff changeset	526 @active
23764 d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	527 def addabort(self, category, callback):
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	528 """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	529
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	530 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	531
d486e52352e8 transaction: support for callbacks during abort Gregory Szorc <gregory.szorc@gmail.com> parents: 23727 diff changeset	532 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	533 with a newer callback.
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 self._abortcallback[category] = 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 @active
44543 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	538 def addvalidator(self, category, callback):
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	539 """adds a callback to be called when validating the transaction.
44543 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	540
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	541 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: 44407 diff changeset	542
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	543 callback should raise exception if to abort transaction"""
44543 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	544 self._validatecallback[category] = callback
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	545
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	546 @active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	547 def close(self):
9220 8a4da1388553 transaction: document close(), abort() methods Greg Ward <greg-hg@gerg.ca> parents: 9094 diff changeset	548 '''commit the transaction'''
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	549 if self._count == 1:
44543 36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	550 for category in sorted(self._validatecallback):
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	551 self._validatecallback[category](self)
36f08ae87ef6 transaction: add functionality to have multiple validators Pulkit Goyal <7895pulkit@gmail.com> parents: 44407 diff changeset	552 self._validatecallback = None # Help prevent cycles.
44406 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	553 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	554 while self._finalizecallback:
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	555 callbacks = self._finalizecallback
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	556 self._finalizecallback = {}
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	557 categories = sorted(callbacks)
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	558 for cat in categories:
2f1d6180737f transaction: allow finalizer to add finalizer Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44056 diff changeset	559 callbacks[cat](self)
28960 14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	560 # Prevent double usage and help clear cycles.
14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	561 self._finalizecallback = None
44406 baf8c3f944eb transaction: move constant to upper case Pierre-Yves David <pierre-yves.david@octobus.net> parents: 44097 diff changeset	562 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	563
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	564 self._count -= 1
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	565 if self._count != 0:
1806 a2c69737e65e Automatic nesting into running transactions in the same repository. mason@suse.com parents: 1559 diff changeset	566 return
39677 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39676 diff changeset	567 self._file.close()
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	568 self._backupsfile.close()
23291 03d2d6931836 transaction: allow registering a temporary transaction file Pierre-Yves David <pierre-yves.david@fb.com> parents: 23290 diff changeset	569 # cleanup temporary files
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	570 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	571 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	572 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	573 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	574 )
23312 006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	575 continue
23311 64ab33ffba14 transaction: use the location value when doing backup Pierre-Yves David <pierre-yves.david@fb.com> parents: 23310 diff changeset	576 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	577 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	578 try:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	579 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	580 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	581 if not c:
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	582 raise
006e9ef05c31 transaction: support cache file in backupentries Pierre-Yves David <pierre-yves.david@fb.com> parents: 23311 diff changeset	583 # Abort may be raise by read only opener
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	584 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	585 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	586 )
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	587 self._offsetmap = {}
45872 ec73a6a75985 transaction: split new files into a separate set Joerg Sonnenberger <joerg@bec.de> parents: 45871 diff changeset	588 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	589 self._writeundo()
39681 0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39680 diff changeset	590 if self._after:
0d7b9db85675 transaction: make after a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39680 diff changeset	591 self._after()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	592 self._after = None # Help prevent cycles.
39682 b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39681 diff changeset	593 if self._opener.isfile(self._backupjournal):
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39681 diff changeset	594 self._opener.unlink(self._backupjournal)
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39681 diff changeset	595 if self._opener.isfile(self._journal):
b590f4763aba transaction: make opener a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39681 diff changeset	596 self._opener.unlink(self._journal)
27662 f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	597 for l, _f, b, c in self._backupentries:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	598 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	599 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	600 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	601 b"%s\n" % (b, l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	602 )
27662 f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	603 continue
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	604 vfs = self._vfsmap[l]
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	605 if b and vfs.exists(b):
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	606 try:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	607 vfs.unlink(b)
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	608 except (IOError, OSError, error.Abort) as inst:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	609 if not c:
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	610 raise
f7ab50c721ac transaction: remove 'if True:' Martin von Zweigbergk <martinvonz@google.com> parents: 26754 diff changeset	611 # Abort may be raise by read only opener
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	612 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	613 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	614 )
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	615 self._backupentries = []
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	616 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	617
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	618 self._releasefn(self, True) # notify success of closing transaction
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	619 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	620
23220 3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	621 # run post close action
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	622 categories = sorted(self._postclosecallback)
3f543f6be500 transaction: allow registering a post-close callback Pierre-Yves David <pierre-yves.david@fb.com> parents: 23204 diff changeset	623 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	624 self._postclosecallback[cat](self)
28960 14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	625 # Prevent double usage and help clear cycles.
14e683d6b273 transaction: clear callback instances after usage Gregory Szorc <gregory.szorc@gmail.com> parents: 28830 diff changeset	626 self._postclosecallback = None
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	627
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	628 @active
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	629 def abort(self):
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	630 """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	631 transaction is not explicitly committed before going out of
45942 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 45872 diff changeset	632 scope)"""
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	633 self._abort()
fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	634
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	635 def _writeundo(self):
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	636 """write transaction data for possible future undo call"""
39675 da9ce63bfa9b transaction: make undoname a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39674 diff changeset	637 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	638 return
47422 88439c6fbafc transaction: simplify `undo.backupfiles` file creation with a variable Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47420 diff changeset	639
88439c6fbafc transaction: simplify `undo.backupfiles` file creation with a variable Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47420 diff changeset	640 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: 47420 diff changeset	641 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	642 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	643 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	644 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	645 continue
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	646 if not b:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	647 u = b''
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	648 else:
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	649 if l not in self._vfsmap and c:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	650 self._report(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	651 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	652 b"%s\n" % (b, l)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	653 )
23904 d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	654 continue
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	655 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	656 base, name = vfs.split(b)
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	657 assert name.startswith(self._journal), name
77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	658 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	659 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	660 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	661 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	662 undobackupfile.close()
d251da5e0e84 transaction: include backup file in the "undo" transaction Pierre-Yves David <pierre-yves.david@fb.com> parents: 23903 diff changeset	663
8289 fe8a3e56039f transaction: ensure finished transactions are not reused Henrik Stuart <hg@hstuart.dk> parents: 8225 diff changeset	664 def _abort(self):
45871 a985c4fb23ca transaction: change list of journal entries into a dictionary Joerg Sonnenberger <joerg@bec.de> parents: 45870 diff changeset	665 entries = self.readjournal()
39674 3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	666 self._count = 0
3e8952c0cb45 transaction: make count and usages private attributes Gregory Szorc <gregory.szorc@gmail.com> parents: 36819 diff changeset	667 self._usages = 0
39677 d27fde3e023e transaction: make file a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39676 diff changeset	668 self._file.close()
23249 84720eab4fbd transaction: mark backup-related attributes private Pierre-Yves David <pierre-yves.david@fb.com> parents: 23248 diff changeset	669 self._backupsfile.close()
8290 560af1bbfd6e transaction: reset transaction on abort Henrik Stuart <hg@hstuart.dk> parents: 8289 diff changeset	670
49993 2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	671 quick = self._can_quick_abort(entries)
10228 056c366fea8c transaction: initialize self.journal to None after deletion Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 9693 diff changeset	672 try:
49994 3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	673 if not quick:
3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	674 self._report(_(b"transaction abort!\n"))
3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	675 for cat in sorted(self._abortcallback):
3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	676 self._abortcallback[cat](self)
3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	677 # Prevent double usage and help clear cycles.
3128018e878b transaction: run abort callback in all cases Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49993 diff changeset	678 self._abortcallback = None
49993 2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	679 if quick:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	680 self._do_quick_abort(entries)
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	681 else:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	682 self._do_full_abort(entries)
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	683 finally:
39676 77c4e2ae9f07 transaction: make journal a private attribute Gregory Szorc <gregory.szorc@gmail.com> parents: 39675 diff changeset	684 self._journal = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	685 self._releasefn(self, False) # notify failure of transaction
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	686 self._releasefn = None # Help prevent cycles.
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	687
49993 2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	688 def _can_quick_abort(self, entries):
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	689 """False if any semantic content have been written on disk
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	690
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	691 True if nothing, except temporary files has been writen on disk."""
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	692 if entries:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	693 return False
49995 27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	694 for e in self._backupentries:
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	695 if e[1]:
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	696 return False
49993 2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	697 return True
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	698
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	699 def _do_quick_abort(self, entries):
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	700 """(Silently) do a quick cleanup (see _can_quick_abort)"""
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	701 assert self._can_quick_abort(entries)
49995 27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	702 tmp_files = [e for e in self._backupentries if not e[1]]
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	703 for vfs_id, old_path, tmp_path, xxx in tmp_files:
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	704 assert not old_path
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	705 vfs = self._vfsmap[vfs_id]
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	706 try:
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	707 vfs.unlink(tmp_path)
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	708 except FileNotFoundError:
27fd12eca557 transaction: quietly rollback if no other changes than temporary files Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49994 diff changeset	709 pass
49993 2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	710 if self._backupjournal:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	711 self._opener.unlink(self._backupjournal)
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	712 if self._journal:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	713 self._opener.unlink(self._journal)
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	714
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	715 def _do_full_abort(self, entries):
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	716 """(Noisily) rollback all the change introduced by the transaction"""
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	717 try:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	718 _playback(
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	719 self._journal,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	720 self._report,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	721 self._opener,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	722 self._vfsmap,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	723 entries,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	724 self._backupentries,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	725 False,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	726 checkambigfiles=self._checkambigfiles,
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	727 )
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	728 self._report(_(b"rollback completed\n"))
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	729 except BaseException as exc:
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	730 self._report(_(b"rollback failed - please run hg recover\n"))
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	731 self._report(
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	732 _(b"(failure reason: %s)\n") % stringutil.forcebytestr(exc)
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	733 )
2f348babe30d transaction: clarify the "quick abort" scenario Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49306 diff changeset	734
8290 560af1bbfd6e transaction: reset transaction on abort Henrik Stuart <hg@hstuart.dk> parents: 8289 diff changeset	735
47419 0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47418 diff changeset	736 BAD_VERSION_MSG = _(
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47418 diff changeset	737 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: 47418 diff changeset	738 )
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47418 diff changeset	739
0e4e9c1b4cc8 transaction: extract message about different version in a constants Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47418 diff changeset	740
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	741 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	742 """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	743
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	744 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	745 '*.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	746
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	747 * `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	748 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	749 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	750 '*.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	751 pairs, delimited by \0.
33278 87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	752
87bca10a06ed transaction: avoid file stat ambiguity only for files in blacklist FUJIWARA Katsunori <foozy@lares.dti.ne.jp> parents: 33087 diff changeset	753 `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: 33087 diff changeset	754 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: 33087 diff changeset	755 restoring corresponded files.
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	756 """
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	757 entries = []
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	758 backupentries = []
8294 48a382c23226 transaction: refactor transaction.abort and rollback to use the same code Henrik Stuart <hg@hstuart.dk> parents: 8290 diff changeset	759
47307 13dd5bb5492a transaction: trivial refactoring Valentin Gatien-Baron <valentin.gatienbaron@gmail.com> parents: 47286 diff changeset	760 with opener.open(file) as fp:
13dd5bb5492a transaction: trivial refactoring Valentin Gatien-Baron <valentin.gatienbaron@gmail.com> parents: 47286 diff changeset	761 lines = fp.readlines()
13400 14f3795a5ed7 explicitly close files Dan Villiom Podlaski Christiansen <danchr@gmail.com> parents: 11685 diff changeset	762 for l in lines:
20524 28b8ff84db3f journal: report parsing errors on recover/rollback (issue4172) Matt Mackall <mpm@selenic.com> parents: 20087 diff changeset	763 try:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	764 f, o = l.split(b'\0')
45869 63edc384d3b7 transaction: drop per-file extra data support Joerg Sonnenberger <joerg@bec.de> parents: 45738 diff changeset	765 entries.append((f, int(o)))
20524 28b8ff84db3f journal: report parsing errors on recover/rollback (issue4172) Matt Mackall <mpm@selenic.com> parents: 20087 diff changeset	766 except ValueError:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	767 report(
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	768 _(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	769 )
0 9117c6561b0b Add back links from file revisions to changeset revisions mpm@selenic.com parents: diff changeset	770
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	771 backupjournal = b"%s.backupfiles" % file
20882 5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	772 if opener.exists(backupjournal):
5dffd06f1e50 transaction: add support for non-append files Durham Goode <durham@fb.com> parents: 20881 diff changeset	773 fp = opener.open(backupjournal)
23065 963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	774 lines = fp.readlines()
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	775 if lines:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	776 ver = lines[0][:-1]
47420 e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47419 diff changeset	777 if ver != (b'%d' % version):
e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47419 diff changeset	778 report(BAD_VERSION_MSG)
e7ad2490d623 transaction: clarify a conditionnal about version check Pierre-Yves David <pierre-yves.david@octobus.net> parents: 47419 diff changeset	779 else:
23065 963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	780 for line in lines[1:]:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	781 if line:
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	782 # Shave off the trailing newline
963f311e3a81 transactions: change backupfiles format to use newlines Durham Goode <durham@fb.com> parents: 23064 diff changeset	783 line = line[:-1]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	784 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	785 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	786
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	787 _playback(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	788 file,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	789 report,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	790 opener,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	791 vfsmap,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	792 entries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	793 backupentries,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	794 checkambigfiles=checkambigfiles,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43050 diff changeset	795 )

Mercurial > public > mercurial-scm > hg

annotate mercurial/transaction.py @ 49995:27fd12eca557