| hg-reviewers |
This is used to track which files are modified. It is an
implementation detail of current transactions and doesn't need
to be exposed to 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. |
| # 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 | ||||
| self._undoname = undoname | self._undoname = undoname | ||||
| self._queue = [] | self._queue = [] | ||||
| # A callback to validate transaction content before closing it. | # A callback to validate transaction content before closing it. | ||||
| # should raise exception is anything is wrong. | # should raise exception is anything is wrong. | ||||
| # target user is repository hooks. | # target user is repository hooks. | ||||
| if validator is None: | if validator is None: | ||||
| validator = lambda tr: None | validator = lambda tr: None | ||||
| sees either none or all of the strip actions to be done.""" | sees either none or all of the strip actions to be done.""" | ||||
| q = self._queue.pop() | q = self._queue.pop() | ||||
| for f, o, data in q: | for f, o, data in q: | ||||
| self._addentry(f, o, data) | self._addentry(f, o, data) | ||||
| @active | @active | ||||
| def add(self, file, offset, data=None): | def add(self, file, offset, data=None): | ||||
| """record the state of an append-only file before update""" | """record the state of an append-only file before update""" | ||||
| if file in self.map or file in self._backupmap: | if file in self._map or file in self._backupmap: | ||||
| return | return | ||||
| if self._queue: | if self._queue: | ||||
| self._queue[-1].append((file, offset, data)) | self._queue[-1].append((file, offset, data)) | ||||
| return | return | ||||
| self._addentry(file, offset, data) | self._addentry(file, offset, data) | ||||
| 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. | ||||
| * `file`: the file path, relative to .hg/store | * `file`: the file path, relative to .hg/store | ||||
| * `hardlink`: use a hardlink to quickly create the backup | * `hardlink`: use a hardlink to quickly create the backup | ||||
| """ | """ | ||||
| if self._queue: | if self._queue: | ||||
| msg = 'cannot use transaction.addbackup inside "group"' | msg = 'cannot use transaction.addbackup inside "group"' | ||||
| raise error.ProgrammingError(msg) | raise error.ProgrammingError(msg) | ||||
| if file in self.map or file in self._backupmap: | if file in self._map or file in self._backupmap: | ||||
| return | return | ||||
| vfs = self._vfsmap[location] | vfs = self._vfsmap[location] | ||||
| dirname, filename = vfs.split(file) | dirname, filename = vfs.split(file) | ||||
| backupfilename = "%s.backup.%s" % (self._journal, filename) | backupfilename = "%s.backup.%s" % (self._journal, filename) | ||||
| backupfile = vfs.reljoin(dirname, backupfilename) | backupfile = vfs.reljoin(dirname, backupfilename) | ||||
| if vfs.exists(file): | if vfs.exists(file): | ||||
| filepath = vfs.join(file) | filepath = vfs.join(file) | ||||
| backuppath = vfs.join(backupfile) | backuppath = vfs.join(backupfile) | ||||
| genfunc(*files) | genfunc(*files) | ||||
| finally: | finally: | ||||
| for f in files: | for f in files: | ||||
| f.close() | f.close() | ||||
| return any | return any | ||||
| @active | @active | ||||
| def find(self, file): | def find(self, file): | ||||
| if file in self.map: | if file in self._map: | ||||
| return self.entries[self.map[file]] | return self.entries[self._map[file]] | ||||
| if file in self._backupmap: | if file in self._backupmap: | ||||
| return self._backupentries[self._backupmap[file]] | return self._backupentries[self._backupmap[file]] | ||||
| return None | return None | ||||
| @active | @active | ||||
| def replace(self, file, offset, data=None): | def replace(self, file, offset, data=None): | ||||
| ''' | ''' | ||||
| 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 | ||||