changeset 48181:1e98f9b5bc71

largefiles: add tr backup for largefilesdirstate This will help us in automatically restoring the largefilesdirstate if a transaction is rolled back. Differential Revision: https://phab.mercurial-scm.org/D11611
author Pulkit Goyal <7895pulkit@gmail.com>
date Mon, 19 Jul 2021 04:12:14 +0530
parents df3021c1f093
children 941fbaab5aff
files hgext/largefiles/lfutil.py tests/test-largefiles-cache.t
diffstat 2 files changed, 4 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/hgext/largefiles/lfutil.py	Mon Jul 19 04:11:08 2021 +0530
+++ b/hgext/largefiles/lfutil.py	Mon Jul 19 04:12:14 2021 +0530
@@ -195,6 +195,8 @@
         # (1) disable PENDING mode always
         #     (lfdirstate isn't yet managed as a part of the transaction)
         # (2) avoid develwarn 'use dirstate.write with ....'
+        if tr:
+            tr.addbackup(b'largefiles/dirstate', location=b'plain')
         super(largefilesdirstate, self).write(None)
 
 
--- a/tests/test-largefiles-cache.t	Mon Jul 19 04:11:08 2021 +0530
+++ b/tests/test-largefiles-cache.t	Mon Jul 19 04:12:14 2021 +0530
@@ -185,10 +185,12 @@
 
   $ find share_dst/.hg/largefiles/* | sort
   share_dst/.hg/largefiles/dirstate
+  share_dst/.hg/largefiles/undo.backup.dirstate
 
   $ find src/.hg/largefiles/* | egrep "(dirstate|$hash)" | sort
   src/.hg/largefiles/dirstate
   src/.hg/largefiles/e2fb5f2139d086ded2cb600d5a91a196e76bf020
+  src/.hg/largefiles/undo.backup.dirstate
 
 Verify that backwards compatibility is maintained for old storage layout
   $ mv src/.hg/largefiles/$hash share_dst/.hg/largefiles