This commit is contained in:
pavelkrolevets 2018-08-22 15:34:37 +08:00
commit 197fcd5445

View file

@ -5,12 +5,12 @@ import (
"errors"
"fmt"
"encoding/binary"
"github.com/pavelkrolevets/go-ethereum/common"
"github.com/pavelkrolevets/go-ethereum/crypto/sha3"
"github.com/pavelkrolevets/go-ethereum/ethdb"
"github.com/pavelkrolevets/go-ethereum/rlp"
"github.com/pavelkrolevets/go-ethereum/trie"
"encoding/binary"
"github.com/pavelkrolevets/go-ethereum/ethdb"
)
type LCPContext struct {
@ -264,8 +264,6 @@ type LCPContextProto struct {
periodBlockHash common.Hash `json:"periodBlockRoot" gencodec:"required"`
maxValidatorsHash common.Hash `json:"maxValidatorRoot" gencodec:"required"`
epochIntervalHash common.Hash `json:"epochIntervalRoot" gencodec:"required"`
}
func (d *LCPContext) ToProto() *LCPContextProto {
@ -389,6 +387,7 @@ func (d *LCPContext) UnDelegate(delegatorAddr, candidateAddr common.Address) err
}
return d.voteTrie.TryDelete(delegator)
}
//function to write tries to memory
func (d *LCPContext) CommitTo(db ethdb.Database) (*LCPContextProto, error) {
epochRoot, err := d.epochTrie.Commit(nil)
@ -451,7 +450,6 @@ func (dc *LCPContext) SetVote(vote *trie.Trie) { dc.voteTrie = vote }
func (dc *LCPContext) SetCandidate(candidate *trie.Trie) { dc.candidateTrie = candidate }
func (dc *LCPContext) SetMintCnt(mintCnt *trie.Trie) { dc.mintCntTrie = mintCnt }
func (dc *LCPContext) GetValidators() ([]common.Address, error) {
var validators []common.Address
key := []byte("validator")
@ -481,7 +479,7 @@ func (dc *LCPContext) SetPeriodBlock(period int64) error {
return nil
}
func (dc *LCPContext) GetPeriodBlock() (int64) {
func (dc *LCPContext) GetPeriodBlock() int64 {
key := []byte("BlockPeriod")
periodRLP := dc.epochTrie.Get(key)
period := int64(binary.LittleEndian.Uint64(periodRLP))
@ -497,7 +495,7 @@ func (dc *LCPContext) SetMaxValidators(maxVal int64) error {
return nil
}
func (dc *LCPContext) GetMaxValidators() (int64) {
func (dc *LCPContext) GetMaxValidators() int64 {
key := []byte("MaxValidators")
maxValRLP := dc.epochTrie.Get(key)
maxVal := int64(binary.LittleEndian.Uint64(maxValRLP))
@ -512,7 +510,7 @@ func (dc *LCPContext) SetEpochInterval(epochInt int64) error {
dc.epochTrie.Update(key, epochIntByte)
return nil
}
func (dc *LCPContext) GetEpochInterval() (int64) {
func (dc *LCPContext) GetEpochInterval() int64 {
key := []byte("EpochInterval")
epochIntRLP := dc.epochTrie.Get(key)
period := int64(binary.LittleEndian.Uint64(epochIntRLP))