From db18632dd211238fadcdec0fab643698be534b62 Mon Sep 17 00:00:00 2001 From: Sonic Date: Fri, 9 Nov 2018 17:24:54 +0800 Subject: core: support extracting governance state from state trie --- core/state/gov.go | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 core/state/gov.go (limited to 'core/state') diff --git a/core/state/gov.go b/core/state/gov.go new file mode 100644 index 000000000..8b94cfab2 --- /dev/null +++ b/core/state/gov.go @@ -0,0 +1,31 @@ +package state + +import ( + "github.com/dexon-foundation/dexon/common" + "github.com/dexon-foundation/dexon/core/types" + "github.com/dexon-foundation/dexon/trie" +) + +func GetGovState(statedb *StateDB, header *types.Header, + addr common.Address) (*types.GovState, error) { + proof, err := statedb.GetProof(addr) + if err != nil { + return nil, err + } + + govState := &types.GovState{ + BlockHash: header.Hash(), + Number: header.Number, + Root: header.Root, + Proof: proof, + } + + if t := statedb.StorageTrie(addr); t != nil { + it := trie.NewIterator(t.NodeIterator(nil)) + for it.Next() { + govState.Storage = append(govState.Storage, + [2][]byte{it.Key, it.Value}) + } + } + return govState, nil +} -- cgit