cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

cleanup pre-migration code a bit

Browse Source
This commit is contained in:
Steven Allen 2021-01-26 18:50:25 -08:00
parent 4072f24bf2
commit 35d6a40071
1 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
chain/stmgr/forks.go
View File

@ -250,6 +250,22 @@ func (sm *StateManager) hasExpensiveFork(ctx context.Context, height abi.ChainEp
return ok return ok
} }
func runPreMigration(ctx context.Context, sm *StateManager, fn PreMigrationFunc, cache *nv10.MemMigrationCache, ts *types.TipSet) {
// Clone the cache so we don't actually _update_ it
// till we're done. Otherwise, if we fail, the next
// migration to use the cache may assume that
// certain blocks exist, even if they don't.
tmpCache := cache.Clone()
err := fn(ctx, sm, tmpCache, ts.ParentState(), ts.Height(), ts)
if err != nil {
log.Errorw("failed to run pre-migration",
"error", err)
return
}
// Finally, if everything worked, update the cache.
cache.Update(tmpCache)
}
func (sm *StateManager) preMigrationWorker(ctx context.Context) { func (sm *StateManager) preMigrationWorker(ctx context.Context) {
defer close(sm.shutdown) defer close(sm.shutdown)
@ -286,20 +302,7 @@ func (sm *StateManager) preMigrationWorker(ctx context.Context) {
wg.Add(1) wg.Add(1)
go func() { go func() {
defer wg.Done() defer wg.Done()
runPreMigration(preCtx, sm, migrationFunc, cache, ts)
// Clone the cache so we don't actually _update_ it
// till we're done. Otherwise, if we fail, the next
// migration to use the cache may assume that
// certain blocks exist, even if they don't.
tmpCache := cache.Clone()
err := migrationFunc(preCtx, sm, tmpCache, ts.ParentState(), ts.Height(), ts)
if err != nil {
log.Errorw("failed to run pre-migration",
"error", err)
return
}
// Finally, if everything worked, update the cache.
cache.Update(tmpCache)
}() }()
}, },
}) })