statestore: try to recover leveldb on corrupted db (#1253)

pkg/intervalstore/store_test.go

@@ -39,7 +39,7 @@ func TestDBStore(t *testing.T) {
 	}
 	defer os.RemoveAll(dir)
 
-	store, err := leveldb.NewStateStore(dir)
+	store, err := leveldb.NewStateStore(dir, nil)
 	if err != nil {
 		t.Fatal(err)
 	}

pkg/node/node.go

@@ -136,7 +136,7 @@ func NewBee(addr string, swarmAddress swarm.Address, publicKey ecdsa.PublicKey,
 		stateStore = mockinmem.NewStateStore()
 		logger.Warning("using in-mem state store. no node state will be persisted")
 	} else {
-		stateStore, err = leveldb.NewStateStore(filepath.Join(o.DataDir, "statestore"))
+		stateStore, err = leveldb.NewStateStore(filepath.Join(o.DataDir, "statestore"), logger)
 		if err != nil {
 			return nil, fmt.Errorf("statestore: %w", err)
 		}

pkg/statestore/leveldb/leveldb.go

@@ -8,9 +8,12 @@ import (
 	"encoding"
 	"encoding/json"
 	"errors"
+	"fmt"
 
+	"github.com/ethersphere/bee/pkg/logging"
 	"github.com/ethersphere/bee/pkg/storage"
 	"github.com/syndtr/goleveldb/leveldb"
+	ldberr "github.com/syndtr/goleveldb/leveldb/errors"
 	"github.com/syndtr/goleveldb/leveldb/util"
 )
 
@@ -19,16 +22,27 @@ var _ storage.StateStorer = (*store)(nil)
 // Store uses LevelDB to store values.
 type store struct {
 	db     *leveldb.DB
+	logger logging.Logger
 }
 
 // New creates a new persistent state storage.
-func NewStateStore(path string) (storage.StateStorer, error) {
+func NewStateStore(path string, l logging.Logger) (storage.StateStorer, error) {
 	db, err := leveldb.OpenFile(path, nil)
 	if err != nil {
+		if !ldberr.IsCorrupted(err) {
 			return nil, err
 		}
+
+		l.Warningf("statestore open failed: %v. attempting recovery", err)
+		db, err = leveldb.RecoverFile(path, nil)
+		if err != nil {
+			return nil, fmt.Errorf("statestore recovery: %w", err)
+		}
+		l.Warning("statestore recovery ok! you are kindly request to inform us about the steps that preceded the last Bee shutdown.")
+	}
 	return &store{
 		db:     db,
+		logger: l,
 	}, nil
 }
 

pkg/statestore/leveldb/leveldb_test.go

@@ -26,7 +26,7 @@ func TestPersistentStateStore(t *testing.T) {
 			}
 		})
 
-		store, err := leveldb.NewStateStore(dir)
+		store, err := leveldb.NewStateStore(dir, nil)
 		if err != nil {
 			t.Fatal(err)
 		}
@@ -40,7 +40,7 @@ func TestPersistentStateStore(t *testing.T) {
 	})
 
 	test.RunPersist(t, func(t *testing.T, dir string) storage.StateStorer {
-		store, err := leveldb.NewStateStore(dir)
+		store, err := leveldb.NewStateStore(dir, nil)
 		if err != nil {
 			t.Fatal(err)
 		}