| hg-reviewers |
The VFS instance is an implementation detail of the transaction
and doesn't belong as part of the public interface. So mark it as
private.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| `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.map = {} | self.map = {} | ||||
| self._journal = journalname | self._journal = journalname | ||||
| # 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: | ||||
| transaction is not explicitly committed before going out of | transaction is not explicitly committed before going out of | ||||
| scope)''' | scope)''' | ||||
| self._abort() | self._abort() | ||||
| def _writeundo(self): | def _writeundo(self): | ||||
| """write transaction data for possible future undo call""" | """write transaction data for possible future undo call""" | ||||
| if self._undoname is None: | if self._undoname is None: | ||||
| return | return | ||||
| undobackupfile = self.opener.open("%s.backupfiles" % self._undoname, | undobackupfile = self._opener.open("%s.backupfiles" % self._undoname, | ||||
| 'w') | 'w') | ||||
| 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._count = 0 | self._count = 0 | ||||
| self._usages = 0 | self._usages = 0 | ||||
| self._file.close() | self._file.close() | ||||
| self._backupsfile.close() | self._backupsfile.close() | ||||
| 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, self._vfsmap, | _playback(self._journal, self.report, self._opener, | ||||
| self.entries, self._backupentries, False, | self._vfsmap, self.entries, self._backupentries, | ||||
| 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. | ||||