neo-go/pkg/core/storage/store.go

package storage

import (
	"encoding/binary"
	"errors"
)

// KeyPrefix constants.
const (
	DataBlock        KeyPrefix = 0x01
	DataTransaction  KeyPrefix = 0x02
	DataMPT          KeyPrefix = 0x03
	STAccount        KeyPrefix = 0x40
	STNotification   KeyPrefix = 0x4d
	STContract       KeyPrefix = 0x50
	STContractID     KeyPrefix = 0x51
	STStorage        KeyPrefix = 0x70
	STNEP5Transfers  KeyPrefix = 0x72
	STNEP5Balances   KeyPrefix = 0x73
	IXHeaderHashList KeyPrefix = 0x80
	SYSCurrentBlock  KeyPrefix = 0xc0
	SYSCurrentHeader KeyPrefix = 0xc1
	SYSContractID    KeyPrefix = 0xc2
	SYSVersion       KeyPrefix = 0xf0
)

// ErrKeyNotFound is an error returned by Store implementations
// when a certain key is not found.
var ErrKeyNotFound = errors.New("key not found")

type (
	// Store is anything that can persist and retrieve the blockchain.
	// information.
	Store interface {
		Batch() Batch
		Delete(k []byte) error
		Get([]byte) ([]byte, error)
		Put(k, v []byte) error
		PutBatch(Batch) error
		Seek(k []byte, f func(k, v []byte))
		Close() error
	}

	// Batch represents an abstraction on top of batch operations.
	// Each Store implementation is responsible of casting a Batch
	// to its appropriate type.
	Batch interface {
		Delete(k []byte)
		Put(k, v []byte)
	}

	// KeyPrefix is a constant byte added as a prefix for each key
	// stored.
	KeyPrefix uint8
)

// Bytes returns the bytes representation of KeyPrefix.
func (k KeyPrefix) Bytes() []byte {
	return []byte{byte(k)}
}

// AppendPrefix appends byteslice b to the given KeyPrefix.
// AppendKeyPrefix(SYSVersion, []byte{0x00, 0x01})
func AppendPrefix(k KeyPrefix, b []byte) []byte {
	dest := make([]byte, len(b)+1)
	dest[0] = byte(k)
	copy(dest[1:], b)
	return dest
}

// AppendPrefixInt append int n to the given KeyPrefix.
//AppendPrefixInt(SYSCurrentHeader, 10001)
func AppendPrefixInt(k KeyPrefix, n int) []byte {
	b := make([]byte, 4)
	binary.LittleEndian.PutUint32(b, uint32(n))
	return AppendPrefix(k, b)
}

// NewStore creates storage with preselected in configuration database type.
func NewStore(cfg DBConfiguration) (Store, error) {
	var store Store
	var err error
	switch cfg.Type {
	case "leveldb":
		store, err = NewLevelDBStore(cfg.LevelDBOptions)
	case "inmemory":
		store = NewMemoryStore()
	case "redis":
		store, err = NewRedisStore(cfg.RedisDBOptions)
	case "boltdb":
		store, err = NewBoltDBStore(cfg.BoltDBOptions)
	case "badgerdb":
		store, err = NewBadgerDBStore(cfg.BadgerDBOptions)
	}
	return store, err
}