| hg-reviewers |
This holds a file handle for the journal file. This file handle
should not be touched outside the journal class and doesn't
belong on the public interface.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| self.names = [name] | self.names = [name] | ||||
| # A dict dedicated to precisely tracking the changes introduced in the | # A dict dedicated to precisely tracking the changes introduced in the | ||||
| # transaction. | # transaction. | ||||
| self.changes = {} | self.changes = {} | ||||
| # a dict of arguments to be passed to hooks | # a dict of arguments to be passed to hooks | ||||
| self.hookargs = {} | self.hookargs = {} | ||||
| self.file = opener.open(self._journal, "w") | self._file = opener.open(self._journal, "w") | ||||
| # a list of ('location', 'path', 'backuppath', cache) entries. | # a list of ('location', 'path', 'backuppath', cache) entries. | ||||
| # - if 'backuppath' is empty, no file existed at backup time | # - if 'backuppath' is empty, no file existed at backup time | ||||
| # - if 'path' is empty, this is a temporary transaction file | # - if 'path' is empty, this is a temporary transaction file | ||||
| # - if 'location' is not empty, the path is outside main opener reach. | # - if 'location' is not empty, the path is outside main opener reach. | ||||
| # use 'location' value as a key in a vfsmap to find the right 'vfs' | # use 'location' value as a key in a vfsmap to find the right 'vfs' | ||||
| # (cache is currently unused) | # (cache is currently unused) | ||||
| self._backupentries = [] | self._backupentries = [] | ||||
| def _addentry(self, file, offset, data): | def _addentry(self, file, offset, data): | ||||
| """add a append-only entry to memory and on-disk state""" | """add a append-only entry to memory and on-disk state""" | ||||
| if file in self.map or file in self._backupmap: | if file in self.map or file in self._backupmap: | ||||
| return | return | ||||
| self.entries.append((file, offset, data)) | self.entries.append((file, offset, data)) | ||||
| self.map[file] = len(self.entries) - 1 | self.map[file] = len(self.entries) - 1 | ||||
| # add enough data to the journal to do the truncate | # add enough data to the journal to do the truncate | ||||
| self.file.write("%s\0%d\n" % (file, offset)) | self._file.write("%s\0%d\n" % (file, offset)) | ||||
| self.file.flush() | self._file.flush() | ||||
| @active | @active | ||||
| def addbackup(self, file, hardlink=True, location=''): | def addbackup(self, file, hardlink=True, location=''): | ||||
| """Adds a backup of the file to the transaction | """Adds a backup of the file to the transaction | ||||
| Calling addbackup() creates a hardlink backup of the specified file | Calling addbackup() creates a hardlink backup of the specified file | ||||
| that is used to recover the file in the event of the transaction | that is used to recover the file in the event of the transaction | ||||
| aborting. | aborting. | ||||
| replace can only replace already committed entries | replace can only replace already committed entries | ||||
| that are not pending in the queue | that are not pending in the queue | ||||
| ''' | ''' | ||||
| if file not in self.map: | if file not in self.map: | ||||
| raise KeyError(file) | raise KeyError(file) | ||||
| index = self.map[file] | index = self.map[file] | ||||
| self.entries[index] = (file, offset, data) | self.entries[index] = (file, offset, data) | ||||
| self.file.write("%s\0%d\n" % (file, offset)) | self._file.write("%s\0%d\n" % (file, offset)) | ||||
| self.file.flush() | self._file.flush() | ||||
| @active | @active | ||||
| def nest(self, name=r'<unnamed>'): | def nest(self, name=r'<unnamed>'): | ||||
| self._count += 1 | self._count += 1 | ||||
| self._usages += 1 | self._usages += 1 | ||||
| self.names.append(name) | self.names.append(name) | ||||
| return self | return self | ||||
| self._finalizecallback[cat](self) | self._finalizecallback[cat](self) | ||||
| # Prevent double usage and help clear cycles. | # Prevent double usage and help clear cycles. | ||||
| self._finalizecallback = None | self._finalizecallback = None | ||||
| self._generatefiles(group=gengrouppostfinalize) | self._generatefiles(group=gengrouppostfinalize) | ||||
| 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] | ||||
| 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)) | ||||
| undobackupfile.close() | undobackupfile.close() | ||||
| def _abort(self): | def _abort(self): | ||||
| 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) | ||||