| durham |
| Restricted Project |
Avoid writing if nothing changes, and write inside a lock.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| node = content[offset:offset + 20] | node = content[offset:offset + 20] | ||||
| if not node: | if not node: | ||||
| break | break | ||||
| if node in nodemap: | if node in nodemap: | ||||
| yield node | yield node | ||||
| offset += 20 | offset += 20 | ||||
| def savepinnednodes(repo, nodes): | def savepinnednodes(repo, nodes): | ||||
| with repo.svfs.open('obsinhibit', 'wb', atomictemp=True) as f: | with repo.lock(), repo.svfs.open('obsinhibit', 'wb', atomictemp=True) as f: | ||||
| f.write(''.join(nodes)) | f.write(''.join(nodes)) | ||||
| def runcommand(orig, lui, repo, cmd, fullargs, *args): | def runcommand(orig, lui, repo, cmd, fullargs, *args): | ||||
| result = orig(lui, repo, cmd, fullargs, *args) | result = orig(lui, repo, cmd, fullargs, *args) | ||||
| # after a command completes, make sure working copy parent and all | # after a command completes, make sure working copy parent and all | ||||
| # bookmarks get "pinned". | # bookmarks get "pinned". | ||||
| if repo and repo.local(): | if repo and repo.local(): | ||||
| unfi = repo.unfiltered() | unfi = repo.unfiltered() | ||||
| wnode = None | wnode = None | ||||
| try: | try: | ||||
| # read dirstate directly to avoid dirstate object overhead | # read dirstate directly to avoid dirstate object overhead | ||||
| wnode = repo.vfs('dirstate').read(20) | wnode = repo.vfs('dirstate').read(20) | ||||
| except Exception: | except Exception: | ||||
| pass | pass | ||||
| tounpin = getattr(unfi, '_tounpinnodes', set()) | tounpin = getattr(unfi, '_tounpinnodes', set()) | ||||
| pinned = set(loadpinnednodes(repo)) - tounpin | origpinned = set(loadpinnednodes(repo)) | ||||
| pinned = origpinned - tounpin | |||||
| if wnode: | if wnode: | ||||
| pinned.add(wnode) | pinned.add(wnode) | ||||
| pinned.update(unfi._bookmarks.values()) | pinned.update(unfi._bookmarks.values()) | ||||
| if pinned != origpinned: | |||||
| savepinnednodes(repo, pinned) | savepinnednodes(repo, pinned) | ||||
| return result | return result | ||||
| def createmarkers(orig, repo, rels, *args, **kwargs): | def createmarkers(orig, repo, rels, *args, **kwargs): | ||||
| # this is a way to unpin revs - precursors are unpinned | # this is a way to unpin revs - precursors are unpinned | ||||
| unfi = repo.unfiltered() | unfi = repo.unfiltered() | ||||
| tounpin = getattr(unfi, '_tounpinnodes', set()) | tounpin = getattr(unfi, '_tounpinnodes', set()) | ||||
| for r in rels: | for r in rels: | ||||
| try: | try: | ||||
| tounpin.add(r[0].node()) | tounpin.add(r[0].node()) | ||||
| except error.RepoLookupError: | except error.RepoLookupError: | ||||
| pass | pass | ||||
| unfi._tounpinnodes = tounpin | unfi._tounpinnodes = tounpin | ||||
| return orig(repo, rels, *args, **kwargs) | return orig(repo, rels, *args, **kwargs) | ||||