| hg-reviewers |
This is a callable used for logging. It isn't used outside the
transaction code. It doesn't need to be part of the public interface.
Let's mark it as private.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| * `releasefn`: called after releasing (with transaction and result) | * `releasefn`: called after releasing (with transaction and result) | ||||
| `checkambigfiles` is a set of (path, vfs-location) tuples, | `checkambigfiles` is a set of (path, vfs-location) tuples, | ||||
| which determine whether file stat ambiguity should be avoided | which determine whether file stat ambiguity should be avoided | ||||
| for corresponded files. | for corresponded files. | ||||
| """ | """ | ||||
| self._count = 1 | self._count = 1 | ||||
| self._usages = 1 | self._usages = 1 | ||||
| self.report = report | self._report = report | ||||
| # a vfs to the store content | # a vfs to the store content | ||||
| self._opener = opener | self._opener = opener | ||||
| # a map to access file in various {location -> vfs} | # a map to access file in various {location -> vfs} | ||||
| vfsmap = vfsmap.copy() | vfsmap = vfsmap.copy() | ||||
| vfsmap[''] = opener # set default value | vfsmap[''] = opener # set default value | ||||
| self._vfsmap = vfsmap | self._vfsmap = vfsmap | ||||
| self._after = after | self._after = after | ||||
| self.entries = [] | self.entries = [] | ||||
| self._count -= 1 | self._count -= 1 | ||||
| if self._count != 0: | if self._count != 0: | ||||
| return | return | ||||
| self._file.close() | self._file.close() | ||||
| self._backupsfile.close() | self._backupsfile.close() | ||||
| # cleanup temporary files | # cleanup temporary files | ||||
| for l, f, b, c in self._backupentries: | for l, f, b, c in self._backupentries: | ||||
| if l not in self._vfsmap and c: | if l not in self._vfsmap and c: | ||||
| self.report("couldn't remove %s: unknown cache location %s\n" | self._report("couldn't remove %s: unknown cache location %s\n" | ||||
| % (b, l)) | % (b, l)) | ||||
| continue | continue | ||||
| vfs = self._vfsmap[l] | vfs = self._vfsmap[l] | ||||
| if not f and b and vfs.exists(b): | if not f and b and vfs.exists(b): | ||||
| try: | try: | ||||
| vfs.unlink(b) | vfs.unlink(b) | ||||
| except (IOError, OSError, error.Abort) as inst: | except (IOError, OSError, error.Abort) as inst: | ||||
| if not c: | if not c: | ||||
| raise | raise | ||||
| # Abort may be raise by read only opener | # Abort may be raise by read only opener | ||||
| self.report("couldn't remove %s: %s\n" | self._report("couldn't remove %s: %s\n" | ||||
| % (vfs.join(b), inst)) | % (vfs.join(b), inst)) | ||||
| self.entries = [] | self.entries = [] | ||||
| self._writeundo() | self._writeundo() | ||||
| if self._after: | if self._after: | ||||
| self._after() | self._after() | ||||
| self._after = None # Help prevent cycles. | self._after = None # Help prevent cycles. | ||||
| if self._opener.isfile(self._backupjournal): | if self._opener.isfile(self._backupjournal): | ||||
| self._opener.unlink(self._backupjournal) | self._opener.unlink(self._backupjournal) | ||||
| if self._opener.isfile(self._journal): | if self._opener.isfile(self._journal): | ||||
| self._opener.unlink(self._journal) | self._opener.unlink(self._journal) | ||||
| for l, _f, b, c in self._backupentries: | for l, _f, b, c in self._backupentries: | ||||
| if l not in self._vfsmap and c: | if l not in self._vfsmap and c: | ||||
| self.report("couldn't remove %s: unknown cache location" | self._report("couldn't remove %s: unknown cache location" | ||||
| "%s\n" % (b, l)) | "%s\n" % (b, l)) | ||||
| continue | continue | ||||
| vfs = self._vfsmap[l] | vfs = self._vfsmap[l] | ||||
| if b and vfs.exists(b): | if b and vfs.exists(b): | ||||
| try: | try: | ||||
| vfs.unlink(b) | vfs.unlink(b) | ||||
| except (IOError, OSError, error.Abort) as inst: | except (IOError, OSError, error.Abort) as inst: | ||||
| if not c: | if not c: | ||||
| raise | raise | ||||
| # Abort may be raise by read only opener | # Abort may be raise by read only opener | ||||
| self.report("couldn't remove %s: %s\n" | self._report("couldn't remove %s: %s\n" | ||||
| % (vfs.join(b), inst)) | % (vfs.join(b), inst)) | ||||
| self._backupentries = [] | self._backupentries = [] | ||||
| self._journal = None | self._journal = None | ||||
| self._releasefn(self, True) # notify success of closing transaction | self._releasefn(self, True) # notify success of closing transaction | ||||
| self._releasefn = None # Help prevent cycles. | self._releasefn = None # Help prevent cycles. | ||||
| # run post close action | # run post close action | ||||
| categories = sorted(self._postclosecallback) | categories = sorted(self._postclosecallback) | ||||
| undobackupfile.write('%d\n' % version) | undobackupfile.write('%d\n' % version) | ||||
| for l, f, b, c in self._backupentries: | for l, f, b, c in self._backupentries: | ||||
| if not f: # temporary file | if not f: # temporary file | ||||
| continue | continue | ||||
| if not b: | if not b: | ||||
| u = '' | u = '' | ||||
| else: | else: | ||||
| if l not in self._vfsmap and c: | if l not in self._vfsmap and c: | ||||
| self.report("couldn't remove %s: unknown cache location" | self._report("couldn't remove %s: unknown cache location" | ||||
| "%s\n" % (b, l)) | "%s\n" % (b, l)) | ||||
| continue | continue | ||||
| vfs = self._vfsmap[l] | vfs = self._vfsmap[l] | ||||
| base, name = vfs.split(b) | base, name = vfs.split(b) | ||||
| assert name.startswith(self._journal), name | assert name.startswith(self._journal), name | ||||
| uname = name.replace(self._journal, self._undoname, 1) | uname = name.replace(self._journal, self._undoname, 1) | ||||
| u = vfs.reljoin(base, uname) | u = vfs.reljoin(base, uname) | ||||
| util.copyfile(vfs.join(b), vfs.join(u), hardlink=True) | util.copyfile(vfs.join(b), vfs.join(u), hardlink=True) | ||||
| undobackupfile.write("%s\0%s\0%s\0%d\n" % (l, f, u, c)) | undobackupfile.write("%s\0%s\0%s\0%d\n" % (l, f, u, c)) | ||||
| try: | try: | ||||
| if not self.entries and not self._backupentries: | if not self.entries and not self._backupentries: | ||||
| if self._backupjournal: | if self._backupjournal: | ||||
| self._opener.unlink(self._backupjournal) | self._opener.unlink(self._backupjournal) | ||||
| if self._journal: | if self._journal: | ||||
| self._opener.unlink(self._journal) | self._opener.unlink(self._journal) | ||||
| return | return | ||||
| self.report(_("transaction abort!\n")) | self._report(_("transaction abort!\n")) | ||||
| try: | try: | ||||
| for cat in sorted(self._abortcallback): | for cat in sorted(self._abortcallback): | ||||
| self._abortcallback[cat](self) | self._abortcallback[cat](self) | ||||
| # Prevent double usage and help clear cycles. | # Prevent double usage and help clear cycles. | ||||
| self._abortcallback = None | self._abortcallback = None | ||||
| _playback(self._journal, self.report, self._opener, | _playback(self._journal, self._report, self._opener, | ||||
| self._vfsmap, self.entries, self._backupentries, | self._vfsmap, self.entries, self._backupentries, | ||||
| False, checkambigfiles=self._checkambigfiles) | False, checkambigfiles=self._checkambigfiles) | ||||
| self.report(_("rollback completed\n")) | self._report(_("rollback completed\n")) | ||||
| except BaseException: | except BaseException: | ||||
| self.report(_("rollback failed - please run hg recover\n")) | self._report(_("rollback failed - please run hg recover\n")) | ||||
| finally: | finally: | ||||
| self._journal = None | self._journal = None | ||||
| self._releasefn(self, False) # notify failure of transaction | self._releasefn(self, False) # notify failure of transaction | ||||
| self._releasefn = None # Help prevent cycles. | self._releasefn = None # Help prevent cycles. | ||||
| def rollback(opener, vfsmap, file, report, checkambigfiles=None): | def rollback(opener, vfsmap, file, report, checkambigfiles=None): | ||||
| """Rolls back the transaction contained in the given file | """Rolls back the transaction contained in the given file | ||||
| if rev >= repolen: | if rev >= repolen: | ||||
| result += b"'" | result += b"'" | ||||
| return result | return result | ||||
| for i in xrange(1, 2 ** len(srevs)): | for i in xrange(1, 2 ** len(srevs)): | ||||
| subset = [rev for j, rev in enumerate(srevs) if i & (1 << j) != 0] | subset = [rev for j, rev in enumerate(srevs) if i & (1 << j) != 0] | ||||
| spec = revsetlang.formatspec(b'%ld', subset) | spec = revsetlang.formatspec(b'%ld', subset) | ||||
| tr = repo.transaction(b'rebase') | tr = repo.transaction(b'rebase') | ||||
| tr.report = lambda x: 0 # hide "transaction abort" | tr._report = lambda x: 0 # hide "transaction abort" | ||||
| ui.pushbuffer() | ui.pushbuffer() | ||||
| try: | try: | ||||
| rebase.rebase(ui, repo, dest=dest, rev=[spec]) | rebase.rebase(ui, repo, dest=dest, rev=[spec]) | ||||
| except error.Abort as ex: | except error.Abort as ex: | ||||
| summary = b'ABORT: %s' % ex | summary = b'ABORT: %s' % ex | ||||
| except Exception as ex: | except Exception as ex: | ||||
| summary = b'CRASH: %s' % ex | summary = b'CRASH: %s' % ex | ||||