triedb: refactor database for a bit cleaner

triedb/database.go

@@ -31,6 +31,8 @@ import (
 	"github.com/ethereum/go-ethereum/triedb/pathdb"
 )
 
+var errNotSupported = errors.New("not supported")
+
 // Config defines all necessary options for database.
 type Config struct {
 	Preimages bool           // Flag whether the preimage of node key is recorded
@@ -222,14 +224,13 @@ func (db *Database) InsertPreimage(preimages map[common.Hash][]byte) {
 //
 // It's only supported by hash-based database and will return an error for others.
 func (db *Database) Cap(limit common.StorageSize) error {
-	hdb, ok := db.backend.(*hashdb.Database)
+	if hdb, ok := db.backend.(*hashdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		if db.preimages != nil {
 			db.preimages.commit(false)
 		}
 		return hdb.Cap(limit)
+	}
+	return errNotSupported
 }
 
 // Reference adds a new reference from a parent node to a child node. This function
@@ -238,23 +239,21 @@ func (db *Database) Cap(limit common.StorageSize) error {
 //
 // It's only supported by hash-based database and will return an error for others.
 func (db *Database) Reference(root common.Hash, parent common.Hash) error {
-	hdb, ok := db.backend.(*hashdb.Database)
+	if hdb, ok := db.backend.(*hashdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		hdb.Reference(root, parent)
 		return nil
+	}
+	return errNotSupported
 }
 
 // Dereference removes an existing reference from a root node. It's only
 // supported by hash-based database and will return an error for others.
 func (db *Database) Dereference(root common.Hash) error {
-	hdb, ok := db.backend.(*hashdb.Database)
+	if hdb, ok := db.backend.(*hashdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		hdb.Dereference(root)
 		return nil
+	}
+	return errNotSupported
 }
 
 // Recover rollbacks the database to a specified historical point. The state is
@@ -262,10 +261,7 @@ func (db *Database) Dereference(root common.Hash) error {
 // corresponding trie histories are existent. It's only supported by path-based
 // database and will return an error for others.
 func (db *Database) Recover(target common.Hash) error {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		var loader triestate.TrieLoader
 		if db.config.IsVerkle {
 			// TODO define verkle loader
@@ -274,17 +270,18 @@ func (db *Database) Recover(target common.Hash) error {
 			loader = trie.NewMerkleLoader(db)
 		}
 		return pdb.Recover(target, loader)
+	}
+	return errNotSupported
 }
 
 // Recoverable returns the indicator if the specified state is enabled to be
 // recovered. It's only supported by path-based database and will return an
 // error for others.
 func (db *Database) Recoverable(root common.Hash) (bool, error) {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return false, errors.New("not supported")
-	}
 		return pdb.Recoverable(root), nil
+	}
+	return false, errNotSupported
 }
 
 // Disable deactivates the database and invalidates all available state layers
@@ -293,21 +290,19 @@ func (db *Database) Recoverable(root common.Hash) (bool, error) {
 //
 // It's only supported by path-based database and will return an error for others.
 func (db *Database) Disable() error {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		return pdb.Disable()
+	}
+	return errNotSupported
 }
 
 // Enable activates database and resets the state tree with the provided persistent
 // state root once the state sync is finished.
 func (db *Database) Enable(root common.Hash) error {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		return pdb.Enable(root)
+	}
+	return errNotSupported
 }
 
 // Journal commits an entire diff hierarchy to disk into a single journal entry.
@@ -315,22 +310,20 @@ func (db *Database) Enable(root common.Hash) error {
 // flattening everything down (bad for reorgs). It's only supported by path-based
 // database and will return an error for others.
 func (db *Database) Journal(root common.Hash) error {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		return pdb.Journal(root)
+	}
+	return errNotSupported
 }
 
 // SetBufferSize sets the node buffer size to the provided value(in bytes).
 // It's only supported by path-based database and will return an error for
 // others.
 func (db *Database) SetBufferSize(size int) error {
-	pdb, ok := db.backend.(*pathdb.Database)
+	if pdb, ok := db.backend.(*pathdb.Database); ok {
-	if !ok {
-		return errors.New("not supported")
-	}
 		return pdb.SetBufferSize(size)
+	}
+	return errNotSupported
 }
 
 // IsVerkle returns the indicator if the database is holding a verkle tree.