multimvview.go

package block_stm

import storetypes "cosmossdk.io/store/types"

// MultiMVMemoryView don't need to be thread-safe, there's a dedicated instance for each tx execution.
type MultiMVMemoryView struct {
	stores   map[storetypes.StoreKey]int
	views    map[storetypes.StoreKey]MVView
	initView func(storetypes.StoreKey) MVView
}

var _ MultiStore = (*MultiMVMemoryView)(nil)

func NewMultiMVMemoryView(
	stores map[storetypes.StoreKey]int,
	storage MultiStore,
	mvMemory *MVMemory,
	schedule *Scheduler,
	txn TxnIndex,
) *MultiMVMemoryView {
	initView := func(name storetypes.StoreKey) MVView {
		i := stores[name]
		return NewMVView(i, storage.GetStore(name), mvMemory.GetMVStore(i), schedule, txn)
	}
	return &MultiMVMemoryView{
		stores:   stores,
		views:    make(map[storetypes.StoreKey]MVView),
		initView: initView,
	}
}

func (mv *MultiMVMemoryView) getViewOrInit(name storetypes.StoreKey) MVView {
	view, ok := mv.views[name]
	if !ok {
		view = mv.initView(name)
		mv.views[name] = view
	}
	return view
}

func (mv *MultiMVMemoryView) GetStore(name storetypes.StoreKey) storetypes.Store {
	return mv.getViewOrInit(name)
}

func (mv *MultiMVMemoryView) GetKVStore(name storetypes.StoreKey) storetypes.KVStore {
	return mv.GetStore(name).(storetypes.KVStore)
}

func (mv *MultiMVMemoryView) GetObjKVStore(name storetypes.StoreKey) storetypes.ObjKVStore {
	return mv.GetStore(name).(storetypes.ObjKVStore)
}

func (mv *MultiMVMemoryView) ReadSet() *MultiReadSet {
	rs := make(MultiReadSet, len(mv.views))
	for key, view := range mv.views {
		rs[mv.stores[key]] = view.ReadSet()
	}
	return &rs
}

func (mv *MultiMVMemoryView) ApplyWriteSet(version TxnVersion) MultiLocations {
	newLocations := make(MultiLocations, len(mv.views))
	for key, view := range mv.views {
		newLocations[mv.stores[key]] = view.ApplyWriteSet(version)
	}
	return newLocations
}