Mercurial > hg
annotate mercurial/transaction.py @ 8262:3d7b1d97230d
localrepo: use set for requirements
author | Matt Mackall <mpm@selenic.com> |
---|---|
date | Wed, 29 Apr 2009 20:47:28 -0500 |
parents | 46293a0c7e9f |
children | fe8a3e56039f |
rev | line source |
---|---|
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
1 # transaction.py - simple journalling scheme for mercurial |
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 # |
2859 | 9 # Copyright 2005, 2006 Matt Mackall <mpm@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 |
46293a0c7e9f
updated license to be explicit about GPL version 2
Martin Geisler <mg@lazybytes.net>
parents:
8071
diff
changeset
|
12 # GNU General Public License version 2, incorporated herein by reference. |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
13 |
3891 | 14 from i18n import _ |
7335
866d2715aff5
add missing import from 618140c75d8d
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
7334
diff
changeset
|
15 import os, errno |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
16 |
1559
59b3639df0a9
Convert all classes to new-style classes by deriving them from object.
Eric Hopper <hopper@omnifarious.org>
parents:
1541
diff
changeset
|
17 class transaction(object): |
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
|
18 def __init__(self, report, opener, journal, after=None, createmode=None): |
162 | 19 self.journal = None |
20 | |
1806
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
21 self.count = 1 |
582 | 22 self.report = report |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
23 self.opener = opener |
95 | 24 self.after = after |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
25 self.entries = [] |
42
91f1fa847158
Fix multiple changes to file per transaction
mpm@selenic.com
parents:
13
diff
changeset
|
26 self.map = {} |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
27 self.journal = journal |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
28 |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
29 self.file = open(self.journal, "w") |
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
|
30 if createmode is not None: |
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
|
31 os.chmod(self.journal, createmode & 0666) |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
32 |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
33 def __del__(self): |
558
0ceea19182a9
transaction: __del__ should do nothing if the journal already exists
mpm@selenic.com
parents:
515
diff
changeset
|
34 if self.journal: |
0ceea19182a9
transaction: __del__ should do nothing if the journal already exists
mpm@selenic.com
parents:
515
diff
changeset
|
35 if self.entries: self.abort() |
0ceea19182a9
transaction: __del__ should do nothing if the journal already exists
mpm@selenic.com
parents:
515
diff
changeset
|
36 self.file.close() |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
37 |
2084 | 38 def add(self, file, offset, data=None): |
42
91f1fa847158
Fix multiple changes to file per transaction
mpm@selenic.com
parents:
13
diff
changeset
|
39 if file in self.map: return |
2084 | 40 self.entries.append((file, offset, data)) |
41 self.map[file] = len(self.entries) - 1 | |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
42 # add enough data to the journal to do the truncate |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
43 self.file.write("%s\0%d\n" % (file, offset)) |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
44 self.file.flush() |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
45 |
2084 | 46 def find(self, file): |
47 if file in self.map: | |
48 return self.entries[self.map[file]] | |
49 return None | |
50 | |
51 def replace(self, file, offset, data=None): | |
52 if file not in self.map: | |
53 raise KeyError(file) | |
54 index = self.map[file] | |
55 self.entries[index] = (file, offset, data) | |
56 self.file.write("%s\0%d\n" % (file, offset)) | |
57 self.file.flush() | |
58 | |
1806
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
59 def nest(self): |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
60 self.count += 1 |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
61 return self |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
62 |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
63 def running(self): |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
64 return self.count > 0 |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
65 |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
66 def close(self): |
1806
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
67 self.count -= 1 |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
68 if self.count != 0: |
a2c69737e65e
Automatic nesting into running transactions in the same repository.
mason@suse.com
parents:
1559
diff
changeset
|
69 return |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
70 self.file.close() |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
71 self.entries = [] |
95 | 72 if self.after: |
785 | 73 self.after() |
95 | 74 else: |
75 os.unlink(self.journal) | |
573 | 76 self.journal = None |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
77 |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
78 def abort(self): |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
79 if not self.entries: return |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
80 |
1402
9d2c2e6b32b5
i18n part2: use '_' for all strings who are part of the user interface
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
1400
diff
changeset
|
81 self.report(_("transaction abort!\n")) |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
82 |
8071
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
83 failed = False |
2084 | 84 for f, o, ignore in self.entries: |
108 | 85 try: |
86 self.opener(f, "a").truncate(o) | |
87 except: | |
8071
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
88 failed = True |
1402
9d2c2e6b32b5
i18n part2: use '_' for all strings who are part of the user interface
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
1400
diff
changeset
|
89 self.report(_("failed to truncate %s\n") % f) |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
90 |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
91 self.entries = [] |
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
92 |
8071
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
93 if not failed: |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
94 self.file.close() |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
95 os.unlink(self.journal) |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
96 self.journal = None |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
97 self.report(_("rollback completed\n")) |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
98 else: |
9f14b66830a8
transaction: only delete journal on successful abort/commit
Henrik Stuart <henrik.stuart@edlund.dk>
parents:
7335
diff
changeset
|
99 self.report(_("rollback failed - please run hg recover\n")) |
515 | 100 |
162 | 101 def rollback(opener, file): |
2084 | 102 files = {} |
162 | 103 for l in open(file).readlines(): |
104 f, o = l.split('\0') | |
6441
c9b8f2820bc0
rollback: unlink files truncated to length 0
Brendan Cully <brendan@kublai.com>
parents:
6065
diff
changeset
|
105 files[f] = int(o) |
2084 | 106 for f in files: |
107 o = files[f] | |
6441
c9b8f2820bc0
rollback: unlink files truncated to length 0
Brendan Cully <brendan@kublai.com>
parents:
6065
diff
changeset
|
108 if o: |
c9b8f2820bc0
rollback: unlink files truncated to length 0
Brendan Cully <brendan@kublai.com>
parents:
6065
diff
changeset
|
109 opener(f, "a").truncate(int(o)) |
c9b8f2820bc0
rollback: unlink files truncated to length 0
Brendan Cully <brendan@kublai.com>
parents:
6065
diff
changeset
|
110 else: |
7334
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
111 try: |
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
112 fn = opener(f).name |
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
113 os.unlink(fn) |
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
114 except OSError, inst: |
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
115 if inst.errno != errno.ENOENT: |
618140c75d8d
fix restart of interrupted recover
Benoit Boissinot <benoit.boissinot@ens-lyon.org>
parents:
6441
diff
changeset
|
116 raise |
162 | 117 os.unlink(file) |
0
9117c6561b0b
Add back links from file revisions to changeset revisions
mpm@selenic.com
parents:
diff
changeset
|
118 |