[#598] Fix read-after-close bug in badger writecache #604

Merged
fyrchik merged 1 commit from ale64bit/frostfs-node:wc-annoying-bug-ci into master 2023-08-15 07:04:08 +00:00
2 changed files with 15 additions and 0 deletions

View file

@ -80,6 +80,13 @@ func (c *cache) flushSmallObjects() {
continue
}
// Using the db after Close will panic and badger won't wait for outstanding txs,
// so we need to check manually.
if c.db.IsClosed() {
c.modeMtx.RUnlock()
return
}
_ = c.db.View(func(tx *badger.Txn) error {
it := tx.NewIterator(badger.DefaultIteratorOptions)
defer it.Close()

View file

@ -20,6 +20,14 @@ func (c *cache) runGCLoop() {
case <-c.closeCh:
return
case <-t.C:
// This serves to synchronize the c.db field when changing mode as well.
c.modeMtx.RLock()
ro := c.readOnly()
c.modeMtx.RUnlock()
if ro {
continue
}
// 0.5 is the recommended value so that write amplification of the value log is 2.
// See https://pkg.go.dev/github.com/dgraph-io/badger/v4#DB.RunValueLogGC for more info.
for c.db.RunValueLogGC(0.5) == nil {