use enum

Author: MaksymMalicki

core/trie2/triedb/database/db.go

@@ -9,6 +9,14 @@ import (
 	"github.com/NethermindEth/juno/db"
 )
 
+type TrieDBScheme uint8
+
+const (
+	RawScheme TrieDBScheme = iota + 1
+	PathScheme
+	HashScheme
+)
+
 // Represents a reader for trie nodes
 type NodeReader interface {
 	Node(owner *felt.Felt, path *trieutils.Path, hash *felt.Felt, isLeaf bool) ([]byte, error)
@@ -37,5 +45,5 @@ type TrieDB interface {
 		mergeClassNodes,
 		mergeContractNodes *trienode.MergeNodeSet,
 	) error
-	Scheme() string
+	Scheme() TrieDBScheme
 }

core/trie2/triedb/hashdb/database.go

@@ -13,8 +13,6 @@ import (
 	"github.com/NethermindEth/juno/utils"
 )
 
-const HashScheme string = "hash"
-
 var _ database.TrieDB = (*Database)(nil)
 
 type Config struct {
@@ -259,6 +257,6 @@ func (d *Database) GetTrieRootNodes(classRootHash, contractRootHash *felt.Felt)
 	return classRootNode, contractRootNode, nil
 }
 
-func (d *Database) Scheme() string {
-	return HashScheme
+func (d *Database) Scheme() database.TrieDBScheme {
+	return database.HashScheme
 }

core/trie2/triedb/pathdb/database.go

@@ -12,9 +12,8 @@ import (
 )
 
 const (
-	maxDiffLayers                  = 128 // TODO(weiihann): might want to make this configurable
-	contractClassTrieHeight        = 251
-	PathScheme              string = "path"
+	maxDiffLayers           = 128 // TODO(weiihann): might want to make this configurable
+	contractClassTrieHeight = 251
 )
 
 var _ database.TrieDB = (*Database)(nil)
@@ -107,6 +106,6 @@ func (d *Database) NewIterator(id trieutils.TrieID) (db.Iterator, error) {
 	return d.disk.NewIterator(prefix, true)
 }
 
-func (d *Database) Scheme() string {
-	return PathScheme
+func (d *Database) Scheme() database.TrieDBScheme {
+	return database.PathScheme
 }

core/trie2/triedb/rawdb/database.go

@@ -11,8 +11,6 @@ import (
 	"github.com/NethermindEth/juno/utils"
 )
 
-const RawScheme string = "raw"
-
 var _ database.TrieDB = (*Database)(nil)
 
 type Config struct{}
@@ -137,6 +135,6 @@ func (d *Database) Close() error {
 	return nil
 }
 
-func (d *Database) Scheme() string {
-	return RawScheme
+func (d *Database) Scheme() database.TrieDBScheme {
+	return database.RawScheme
 }