89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package state
|
|
|
|
import (
|
|
"github.com/tendermint/basecoin/types"
|
|
. "github.com/tendermint/go-common"
|
|
"github.com/tendermint/go-wire"
|
|
eyes "github.com/tendermint/merkleeyes/client"
|
|
)
|
|
|
|
type State struct {
|
|
chainID string
|
|
eyesCli *eyes.Client
|
|
checkCache map[string]checkAccount
|
|
|
|
LastBlockHeight uint64
|
|
LastBlockHash []byte
|
|
GasLimit int64
|
|
}
|
|
|
|
func NewState(eyesCli *eyes.Client) *State {
|
|
s := &State{
|
|
chainID: "",
|
|
eyesCli: eyesCli,
|
|
checkCache: make(map[string]checkAccount),
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (s *State) SetChainID(chainID string) {
|
|
s.chainID = chainID
|
|
}
|
|
|
|
func (s *State) GetChainID() string {
|
|
if s.chainID == "" {
|
|
PanicSanity("Expected to have set SetChainID")
|
|
}
|
|
return s.chainID
|
|
}
|
|
|
|
//----------------------------------------
|
|
// CheckTx state
|
|
|
|
type checkAccount struct {
|
|
sequence int
|
|
balance int64
|
|
}
|
|
|
|
func (s *State) GetCheckAccount(addr []byte, defaultSequence int, defaultBalance int64) (sequence int, balance int64) {
|
|
cAcc, ok := s.checkCache[string(addr)]
|
|
if !ok {
|
|
return defaultSequence, defaultBalance
|
|
}
|
|
return cAcc.sequence, cAcc.balance
|
|
}
|
|
|
|
func (s *State) SetCheckAccount(addr []byte, sequence int, balance int64) {
|
|
s.checkCache[string(addr)] = checkAccount{sequence, balance}
|
|
}
|
|
|
|
func (s *State) ResetCacheState() {
|
|
s.checkCache = make(map[string]checkAccount)
|
|
}
|
|
|
|
//----------------------------------------
|
|
|
|
func (s *State) GetAccount(addr []byte) *types.Account {
|
|
res := s.eyesCli.GetSync(addr)
|
|
if res.IsErr() {
|
|
panic("Error loading account: " + res.Error())
|
|
}
|
|
if len(res.Data) == 0 {
|
|
return nil
|
|
}
|
|
var acc types.Account
|
|
err := wire.ReadBinaryBytes(res.Data, &acc)
|
|
if err != nil {
|
|
panic("Error reading account: " + err.Error())
|
|
}
|
|
return &acc
|
|
}
|
|
|
|
func (s *State) SetAccount(address []byte, acc *types.Account) {
|
|
accBytes := wire.BinaryBytes(acc)
|
|
res := s.eyesCli.SetSync(address, accBytes)
|
|
if res.IsErr() {
|
|
panic("Error storing account: " + res.Error())
|
|
}
|
|
}
|