mirror of https://github.com/status-im/op-geth.git
swarm/storage/localstore: fix synchronization in TestDB_gcSize (#19235)
This commit is contained in:
parent
4687391213
commit
a6e5c6a2cc
|
@ -289,12 +289,9 @@ func TestDB_gcSize(t *testing.T) {
|
|||
}
|
||||
|
||||
// DB.Close writes gc size to disk, so
|
||||
// Instead calling Close, simulate database shutdown
|
||||
// Instead calling Close, close the database
|
||||
// without it.
|
||||
close(db.close)
|
||||
db.updateGCWG.Wait()
|
||||
err = db.shed.Close()
|
||||
if err != nil {
|
||||
if err := db.closeWithOptions(false); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
|
|
|
@ -367,6 +367,12 @@ func New(path string, baseKey []byte, o *Options) (db *DB, err error) {
|
|||
|
||||
// Close closes the underlying database.
|
||||
func (db *DB) Close() (err error) {
|
||||
return db.closeWithOptions(true)
|
||||
}
|
||||
|
||||
// closeWithOptions provides a more control which part of closing
|
||||
// is done for tests.
|
||||
func (db *DB) closeWithOptions(writeGCSize bool) (err error) {
|
||||
close(db.close)
|
||||
db.updateGCWG.Wait()
|
||||
|
||||
|
@ -384,8 +390,10 @@ func (db *DB) Close() (err error) {
|
|||
log.Error("localstore: write gc size worker did not return after db close")
|
||||
}
|
||||
|
||||
if err := db.writeGCSize(db.getGCSize()); err != nil {
|
||||
log.Error("localstore: write gc size", "err", err)
|
||||
if writeGCSize {
|
||||
if err := db.writeGCSize(db.getGCSize()); err != nil {
|
||||
log.Error("localstore: write gc size", "err", err)
|
||||
}
|
||||
}
|
||||
return db.shed.Close()
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue