erigon-pulse/ethdb/storage_mode.go

package ethdb

import (
	"errors"
	"fmt"

	"github.com/ledgerwatch/turbo-geth/common/dbutils"
)

type StorageMode struct {
	History   bool
	Receipts  bool
	TxIndex   bool
	Preimages bool
}

var DefaultStorageMode = StorageMode{History: true, Receipts: false, TxIndex: true, Preimages: true}

func (m StorageMode) ToString() string {
	modeString := ""
	if m.History {
		modeString += "h"
	}
	if m.Preimages {
		modeString += "p"
	}
	if m.Receipts {
		modeString += "r"
	}
	if m.TxIndex {
		modeString += "t"
	}
	return modeString
}

func StorageModeFromString(flags string) (StorageMode, error) {
	mode := StorageMode{}
	for _, flag := range flags {
		switch flag {
		case 'h':
			mode.History = true
		case 'r':
			mode.Receipts = true
		case 't':
			mode.TxIndex = true
		case 'p':
			mode.Preimages = true
		default:
			return mode, fmt.Errorf("unexpected flag found: %c", flag)
		}
	}

	return mode, nil
}

func GetStorageModeFromDB(db Database) (StorageMode, error) {
	var (
		sm  StorageMode
		v   []byte
		err error
	)
	v, err = db.Get(dbutils.DatabaseInfoBucket, dbutils.StorageModeHistory)
	if err != nil && !errors.Is(err, ErrKeyNotFound) {
		return StorageMode{}, err
	}
	sm.History = len(v) > 0

	v, err = db.Get(dbutils.DatabaseInfoBucket, dbutils.StorageModePreImages)
	if err != nil && !errors.Is(err, ErrKeyNotFound) {
		return StorageMode{}, err
	}
	sm.Preimages = len(v) > 0

	v, err = db.Get(dbutils.DatabaseInfoBucket, dbutils.StorageModeReceipts)
	if err != nil && !errors.Is(err, ErrKeyNotFound) {
		return StorageMode{}, err
	}
	sm.Receipts = len(v) > 0

	v, err = db.Get(dbutils.DatabaseInfoBucket, dbutils.StorageModeTxIndex)
	if err != nil && !errors.Is(err, ErrKeyNotFound) {
		return StorageMode{}, err
	}
	sm.TxIndex = len(v) > 0

	return sm, nil
}

func SetStorageModeIfNotExist(db Database, sm StorageMode) error {
	var (
		err error
	)
	err = setModeOnEmpty(db, dbutils.StorageModeHistory, sm.History)
	if err != nil {
		return err
	}

	err = setModeOnEmpty(db, dbutils.StorageModePreImages, sm.Preimages)
	if err != nil {
		return err
	}

	err = setModeOnEmpty(db, dbutils.StorageModeReceipts, sm.Receipts)
	if err != nil {
		return err
	}

	err = setModeOnEmpty(db, dbutils.StorageModeTxIndex, sm.TxIndex)
	if err != nil {
		return err
	}

	return nil
}

func setModeOnEmpty(db Database, key []byte, currentValue bool) error {
	_, err := db.Get(dbutils.DatabaseInfoBucket, key)
	if err != nil && !errors.Is(err, ErrKeyNotFound) {
		return err
	}
	if errors.Is(err, ErrKeyNotFound) {
		val := []byte{}
		if currentValue {
			val = []byte{1}
		}
		if err = db.Put(dbutils.DatabaseInfoBucket, key, val); err != nil {
			return err
		}
	}

	return nil
}