fix: splitstore: Don't deadlock in mpool protector

This commit is contained in:
Łukasz Magiera 2022-12-17 13:12:33 +01:00
parent 1dee884358
commit 156ba420c3
2 changed files with 7 additions and 3 deletions

View File

@ -443,8 +443,12 @@ func New(ctx context.Context, api Provider, ds dtypes.MetadataDS, us stmgr.Upgra
return mp, nil return mp, nil
} }
func (mp *MessagePool) ForEachPendingMessage(f func(cid.Cid) error) error { func (mp *MessagePool) TryForEachPendingMessage(f func(cid.Cid) error) error {
mp.lk.Lock() // avoid deadlocks in splitstore compaction when something else needs to access the blockstore
// while holding the mpool lock
if !mp.lk.TryLock() {
return xerrors.Errorf("mpool TryForEachPendingMessage: could not acquire lock")
}
defer mp.lk.Unlock() defer mp.lk.Unlock()
for _, mset := range mp.pending { for _, mset := range mp.pending {

View File

@ -69,7 +69,7 @@ func MessagePool(lc fx.Lifecycle, mctx helpers.MetricsCtx, us stmgr.UpgradeSched
return mp.Close() return mp.Close()
}, },
}) })
protector.AddProtector(mp.ForEachPendingMessage) protector.AddProtector(mp.TryForEachPendingMessage)
return mp, nil return mp, nil
} }