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

annotate mercurial/transaction.py @ 48698:568f63b5a30f

dirstate: introduce a "tracked-key" feature A new format variant is introduced. When used, a `tracked-key` file will be generated. That file will be update when the set of tracked file might have changed. This will be useful for external automation (e.g. build tool) to be notified when the set of relevant files changes. One of the motivation for this changes is to mitigate effect dirstate-v2 has on such automation. Since the dirstate file is updated much more frequently on dirstate-v2, monitoring update to that file is no longer a viable strategy. See the associated documentation for details about the feature To prevent older client to update the repository without updating that file, a new requirements is introduced. The `postfinalizegenerators` business is a bit weird, so I'll likely clean that up soon. Differential Revision: https://phab.mercurial-scm.org/D12124

author	Pierre-Yves David <pierre-yves.david@octobus.net>
date	Mon, 31 Jan 2022 08:44:48 +0100
parents	111f5a0cbcaa
children	21ac6aedd5e5

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

Mercurial > public > mercurial-scm > hg-stable

annotate mercurial/transaction.py @ 48698:568f63b5a30f