tree.go

package iavl

import (
	"fmt"

	"github.com/okex/exchain/libs/iavl"
)

var (
	_ Tree = (*immutableTree)(nil)
	_ Tree = (*iavl.MutableTree)(nil)
)

type (
	// Tree defines an interface that both mutable and immutable IAVL trees
	// must implement. For mutable IAVL trees, the interface is directly
	// implemented by an iavl.MutableTree. For an immutable IAVL tree, a wrapper
	// must be made.
	Tree interface {
		Has(key []byte) bool
		Get(key []byte) (value []byte)
		Set(key, value []byte) bool
		Remove(key []byte) ([]byte, bool)
		PreChanges(keys []string, setOrDel []byte)
		SaveVersion(bool) ([]byte, int64, iavl.TreeDelta, error)
		GetModuleName() string
		GetDBWriteCount() int
		GetDBReadCount() int
		GetDBReadTime() int
		GetNodeReadCount() int
		ResetCount()
		DeleteVersion(version int64) error
		DeleteVersions(versions ...int64) error
		Version() int64
		Hash() []byte
		VersionExists(version int64) bool
		GetVersioned(key []byte, version int64) (int64, []byte)
		GetVersionedWithProof(key []byte, version int64) ([]byte, *iavl.RangeProof, error)
		GetImmutable(version int64) (*iavl.ImmutableTree, error)
		SetInitialVersion(version uint64)
		SetDelta(delta *iavl.TreeDelta)
		GetPersistedRoots() map[int64][]byte
		SetUpgradeVersion(int64)
	}

	// immutableTree is a simple wrapper around a reference to an iavl.ImmutableTree
	// that implements the Tree interface. It should only be used for querying
	// and iteration, specifically at previous heights.
	immutableTree struct {
		*iavl.ImmutableTree
	}
)

func (it *immutableTree) Set(_, _ []byte) bool {
	panic("cannot call 'Set' on an immutable IAVL tree")
}

func (it *immutableTree) Remove(_ []byte) ([]byte, bool) {
	panic("cannot call 'Remove' on an immutable IAVL tree")
}

func (it *immutableTree) PreChanges(keys []string, setOrDel []byte) {}

func (it *immutableTree) SaveVersion(bool) ([]byte, int64, iavl.TreeDelta, error) {
	panic("cannot call 'SaveVersion' on an immutable IAVL tree")
}

func (it *immutableTree) DeleteVersion(_ int64) error {
	panic("cannot call 'DeleteVersion' on an immutable IAVL tree")
}

func (it *immutableTree) DeleteVersions(_ ...int64) error {
	panic("cannot call 'DeleteVersions' on an immutable IAVL tree")
}

func (it *immutableTree) VersionExists(version int64) bool {
	return it.Version() == version
}

func (it *immutableTree) GetVersioned(key []byte, version int64) (int64, []byte) {
	if it.Version() != version {
		return -1, nil
	}

	return it.GetWithIndex(key)
}

func (it *immutableTree) GetVersionedWithProof(key []byte, version int64) ([]byte, *iavl.RangeProof, error) {
	if it.Version() != version {
		return nil, nil, fmt.Errorf("version mismatch on immutable IAVL tree; got: %d, expected: %d", version, it.Version())
	}

	return it.GetWithProof(key)
}

func (it *immutableTree) GetImmutable(version int64) (*iavl.ImmutableTree, error) {
	if it.Version() != version {
		return nil, fmt.Errorf("version mismatch on immutable IAVL tree; got: %d, expected: %d", version, it.Version())
	}

	return it.ImmutableTree, nil
}

func (it *immutableTree) SetInitialVersion(_ uint64) {
	panic("cannot call 'SetInitialVersion' on an immutable IAVL tree")
}

func (it *immutableTree) SetDelta(delta *iavl.TreeDelta) {
	panic("cannot call 'SetDelta' on an immutable IAVL tree")
}

func (it *immutableTree) GetModuleName() string {
	return ""
}

func (it *immutableTree) GetDBWriteCount() int {
	return 0
}

func (it *immutableTree) GetDBReadCount() int {
	return 0
}

func (it *immutableTree) GetDBReadTime() int {
	return 0
}

func (it *immutableTree) GetNodeReadCount() int { return 0 }

func (it *immutableTree) ResetCount() {

}