diff --git a/hgext/lfs/__init__.py b/hgext/lfs/__init__.py --- a/hgext/lfs/__init__.py +++ b/hgext/lfs/__init__.py @@ -140,6 +140,9 @@ def extsetup(ui): wrapfunction(scmutil, 'wrapconvertsink', wrapper.convertsink) + wrapfunction(upgrade, '_finishdatamigration', + wrapper.upgradefinishdatamigration) + wrapfunction(upgrade, 'preservedrequirements', wrapper.upgraderequirements) diff --git a/hgext/lfs/wrapper.py b/hgext/lfs/wrapper.py --- a/hgext/lfs/wrapper.py +++ b/hgext/lfs/wrapper.py @@ -19,6 +19,8 @@ from mercurial import ( util, ) +from ..largefiles import lfutil + from . import ( blobstore, pointer, @@ -303,6 +305,17 @@ def uploadblobs(repo, pointers): remoteblob = repo.svfs.lfsremoteblobstore remoteblob.writebatch(pointers, repo.svfs.lfslocalblobstore) +def upgradefinishdatamigration(orig, ui, srcrepo, dstrepo, requirements): + orig(ui, srcrepo, dstrepo, requirements) + + srclfsvfs = srcrepo.svfs.lfslocalblobstore.vfs + dstlfsvfs = dstrepo.svfs.lfslocalblobstore.vfs + + for dirpath, dirs, files in srclfsvfs.walk(): + for oid in files: + srcrepo.ui.write(_('copying lfs blob %s\n') % oid) + lfutil.link(srclfsvfs.join(oid), dstlfsvfs.join(oid)) + def upgraderequirements(orig, repo): reqs = orig(repo) if 'lfs' in repo.requirements: diff --git a/tests/test-upgrade-repo.t b/tests/test-upgrade-repo.t --- a/tests/test-upgrade-repo.t +++ b/tests/test-upgrade-repo.t @@ -548,6 +548,7 @@ Check upgrading a large file repository finished migrating 2 changelog revisions; change in size: 0 bytes finished migrating 6 total revisions; total change in store size: 0 bytes copying phaseroots + copying lfs blob d0beab232adff5ba365880366ad30b1edb85c4c5372442b5d2fe27adc96d653f data fully migrated to temporary repository marking source repository as being upgraded; clients will be unable to read from repository starting in-place swap of repository data @@ -561,11 +562,8 @@ Check upgrading a large file repository $ grep lfs .hg/requires lfs -TODO: restore the local lfs store. For now, objects are copied from the user -cache as needed. $ find .hg/store/lfs -type f - find: `.hg/store/lfs': $ENOENT$ - [1] + .hg/store/lfs/objects/d0/beab232adff5ba365880366ad30b1edb85c4c5372442b5d2fe27adc96d653f $ hg verify checking changesets checking manifests