TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

native: cache GetValidators result

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-06-24 13:20:59 +03:00
parent 53f2e130c0
commit 978b3c96fe
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkg/core/native/native_neo.go
View file

@ -4,6 +4,7 @@ import (
"math/big" "math/big"
"sort" "sort"
"strings" "strings"
"sync/atomic"
"github.com/nspcc-dev/neo-go/pkg/core/blockchainer" "github.com/nspcc-dev/neo-go/pkg/core/blockchainer"
"github.com/nspcc-dev/neo-go/pkg/core/dao" "github.com/nspcc-dev/neo-go/pkg/core/dao"
@ -23,6 +24,8 @@ import (
type NEO struct { type NEO struct {
nep5TokenNative nep5TokenNative
GAS *GAS GAS *GAS
validators atomic.Value
} }
// keyWithVotes is a serialized key with votes balance. It's not deserialized // keyWithVotes is a serialized key with votes balance. It's not deserialized
@ -330,6 +333,7 @@ func (n *NEO) ModifyAccountVotes(acc *state.NEOBalanceState, d dao.DAO, value *b
return err return err
} }
} }
n.validators.Store(nil)
return nil return nil
} }
@ -382,9 +386,13 @@ func (n *NEO) getRegisteredValidatorsCall(ic *interop.Context, _ []stackitem.Ite
// GetValidatorsInternal returns a list of current validators. // GetValidatorsInternal returns a list of current validators.
func (n *NEO) GetValidatorsInternal(bc blockchainer.Blockchainer, d dao.DAO) (keys.PublicKeys, error) { func (n *NEO) GetValidatorsInternal(bc blockchainer.Blockchainer, d dao.DAO) (keys.PublicKeys, error) {
if vals := n.validators.Load(); vals != nil {
return vals.(keys.PublicKeys), nil
}
standByValidators := bc.GetStandByValidators() standByValidators := bc.GetStandByValidators()
si := d.GetStorageItem(n.ContractID, validatorsCountKey) si := d.GetStorageItem(n.ContractID, validatorsCountKey)
if si == nil { if si == nil {
n.validators.Store(standByValidators)
return standByValidators, nil return standByValidators, nil
} }
validatorsCount, err := ValidatorsCountFromBytes(si.Value) validatorsCount, err := ValidatorsCountFromBytes(si.Value)
@ -428,6 +436,7 @@ func (n *NEO) GetValidatorsInternal(bc blockchainer.Blockchainer, d dao.DAO) (ke
} }
} }
sort.Sort(result) sort.Sort(result)
n.validators.Store(result)
return result, nil return result, nil
} }