This commit replaces the deep-copy based state revert mechanism with a linear complexity journal. This commit also hides several internal StateDB methods to limit the number of ways in which calling code can use the journal incorrectly. As usual consultation and bug fixes to the initial implementation were provided by @karalabe, @obscuren and @Arachnid. Thank you!
		
			
				
	
	
		
			81 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			81 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2014 The go-ethereum Authors
 | |
| // This file is part of the go-ethereum library.
 | |
| //
 | |
| // The go-ethereum library is free software: you can redistribute it and/or modify
 | |
| // it under the terms of the GNU Lesser General Public License as published by
 | |
| // the Free Software Foundation, either version 3 of the License, or
 | |
| // (at your option) any later version.
 | |
| //
 | |
| // The go-ethereum library is distributed in the hope that it will be useful,
 | |
| // but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 | |
| // GNU Lesser General Public License for more details.
 | |
| //
 | |
| // You should have received a copy of the GNU Lesser General Public License
 | |
| // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
 | |
| 
 | |
| package state
 | |
| 
 | |
| import (
 | |
| 	"encoding/json"
 | |
| 	"fmt"
 | |
| 
 | |
| 	"github.com/ethereum/go-ethereum/common"
 | |
| 	"github.com/ethereum/go-ethereum/rlp"
 | |
| )
 | |
| 
 | |
| type DumpAccount struct {
 | |
| 	Balance  string            `json:"balance"`
 | |
| 	Nonce    uint64            `json:"nonce"`
 | |
| 	Root     string            `json:"root"`
 | |
| 	CodeHash string            `json:"codeHash"`
 | |
| 	Code     string            `json:"code"`
 | |
| 	Storage  map[string]string `json:"storage"`
 | |
| }
 | |
| 
 | |
| type Dump struct {
 | |
| 	Root     string                 `json:"root"`
 | |
| 	Accounts map[string]DumpAccount `json:"accounts"`
 | |
| }
 | |
| 
 | |
| func (self *StateDB) RawDump() Dump {
 | |
| 	dump := Dump{
 | |
| 		Root:     common.Bytes2Hex(self.trie.Root()),
 | |
| 		Accounts: make(map[string]DumpAccount),
 | |
| 	}
 | |
| 
 | |
| 	it := self.trie.Iterator()
 | |
| 	for it.Next() {
 | |
| 		addr := self.trie.GetKey(it.Key)
 | |
| 		var data Account
 | |
| 		if err := rlp.DecodeBytes(it.Value, &data); err != nil {
 | |
| 			panic(err)
 | |
| 		}
 | |
| 
 | |
| 		obj := newObject(nil, common.BytesToAddress(addr), data, nil)
 | |
| 		account := DumpAccount{
 | |
| 			Balance:  data.Balance.String(),
 | |
| 			Nonce:    data.Nonce,
 | |
| 			Root:     common.Bytes2Hex(data.Root[:]),
 | |
| 			CodeHash: common.Bytes2Hex(data.CodeHash),
 | |
| 			Code:     common.Bytes2Hex(obj.Code(self.db)),
 | |
| 			Storage:  make(map[string]string),
 | |
| 		}
 | |
| 		storageIt := obj.getTrie(self.db).Iterator()
 | |
| 		for storageIt.Next() {
 | |
| 			account.Storage[common.Bytes2Hex(self.trie.GetKey(storageIt.Key))] = common.Bytes2Hex(storageIt.Value)
 | |
| 		}
 | |
| 		dump.Accounts[common.Bytes2Hex(addr)] = account
 | |
| 	}
 | |
| 	return dump
 | |
| }
 | |
| 
 | |
| func (self *StateDB) Dump() []byte {
 | |
| 	json, err := json.MarshalIndent(self.RawDump(), "", "    ")
 | |
| 	if err != nil {
 | |
| 		fmt.Println("dump err", err)
 | |
| 	}
 | |
| 
 | |
| 	return json
 | |
| }
 |