transaction: track newly introduced revisions
authorPierre-Yves David <pierre-yves.david@ens-lyon.org>
Tue, 02 May 2017 18:45:51 +0200
changeset 32262 85ef5a073114
parent 32261 976681123416
child 32263 604d65e2c0b2
transaction: track newly introduced revisions Tracking revisions is not the data that will unlock the most new capability. However, they are the simplest thing to track and still unlock some nice improvements in regard with caching. We plug ourself at the changelog level to make sure we do not miss any revision additions. The 'revs' set is configured at the repository level because the transaction itself does not needs to know that much about the business logic.
mercurial/changelog.py
mercurial/localrepo.py
--- a/mercurial/changelog.py	Tue May 02 18:31:18 2017 +0200
+++ b/mercurial/changelog.py	Tue May 02 18:45:51 2017 +0200
@@ -535,3 +535,14 @@
         just to access this is costly."""
         extra = self.read(rev)[5]
         return encoding.tolocal(extra.get("branch")), 'close' in extra
+
+    def _addrevision(self, node, rawtext, transaction, *args, **kwargs):
+        # overlay over the standard revlog._addrevision to track the new
+        # revision on the transaction.
+        rev = len(self)
+        node = super(changelog, self)._addrevision(node, rawtext, transaction,
+                                                   *args, **kwargs)
+        revs = transaction.changes.get('revs')
+        if revs is not None:
+            revs.add(rev)
+        return node
--- a/mercurial/localrepo.py	Tue May 02 18:31:18 2017 +0200
+++ b/mercurial/localrepo.py	Tue May 02 18:45:51 2017 +0200
@@ -1072,6 +1072,7 @@
                                      self.store.createmode,
                                      validator=validate,
                                      releasefn=releasefn)
+        tr.changes['revs'] = set()
 
         tr.hookargs['txnid'] = txnid
         # note: writing the fncache only during finalize mean that the file is