anyproto/any-sync
1
0
Fork 0
mirror of https://github.com/anyproto/any-sync.git synced 2025-06-09 09:35:03 +09:00
Code Activity

Add optional removeFunc for migration

Browse source
This commit is contained in:
mcrakhman 2025-01-24 21:49:51 +01:00
parent 5457b40ea7
commit 39f17121d5
No known key found for this signature in database
GPG key ID: DED12CFEF5B8396B
1 changed files with 19 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

26
commonspace/spacestorage/migration/spacemigrator.go
View file

@ -37,19 +37,27 @@ type Progress interface {
AddDone(done int64)
}
type RemoveFunc func(newStorage spacestorage.SpaceStorage, rootPath string) error
type spaceMigrator struct {
oldProvider oldstorage.SpaceStorageProvider
newProvider spacestorage.SpaceStorageProvider
numParallel int
rootPath string
removeFunc RemoveFunc
}
func NewSpaceMigrator(oldProvider oldstorage.SpaceStorageProvider, newProvider spacestorage.SpaceStorageProvider, numParallel int, rootPath string) SpaceMigrator {
return NewSpaceMigratorWithRemoveFunc(oldProvider, newProvider, numParallel, rootPath, nil)
}
func NewSpaceMigratorWithRemoveFunc(oldProvider oldstorage.SpaceStorageProvider, newProvider spacestorage.SpaceStorageProvider, numParallel int, rootPath string, removeFunc RemoveFunc) SpaceMigrator {
return &spaceMigrator{
oldProvider: oldProvider,
newProvider: newProvider,
numParallel: numParallel,
rootPath: rootPath,
removeFunc: removeFunc,
}
}
@ -68,14 +76,18 @@ func (s *spaceMigrator) MigrateId(ctx context.Context, id string, progress Progr
return ErrAlreadyMigrated
}
if storage != nil {
anyStore := storage.AnyStore()
err := storage.Close(ctx)
if err != nil {
return fmt.Errorf("migration: failed to close old storage: %w", err)
if s.removeFunc != nil {
err := s.removeFunc(storage, s.rootPath)
if err != nil {
return fmt.Errorf("migration: failed to remove new storage: %w", err)
}
} else {
err := storage.Close(ctx)
if err != nil {
return fmt.Errorf("migration: failed to close old storage: %w", err)
}
os.RemoveAll(filepath.Join(s.rootPath, id))
}
// closing this just in case, because the storage doesn't always close the underlying store
anyStore.Close()
os.RemoveAll(filepath.Join(s.rootPath, id))
}
oldStorage, err := s.oldProvider.WaitSpaceStorage(ctx, id)
if err != nil {