roysc/plugeth
1
0
Fork 0
forked from cerc-io/plugeth
Code Pull Requests Activity

Merge pull request #15288 from karalabe/trie-hash-benchmark

Browse Source 
trie: make hasher benchmark meaningful post-caches
This commit is contained in:
Péter Szilágyi 2017-10-13 15:50:05 +03:00 committed by GitHub
commit 41b7745529
1 changed files with 34 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
trie/trie_test.go
View File

@ -22,6 +22,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"math/big"
"math/rand" "math/rand"
"os" "os"
"reflect" "reflect"
@ -30,7 +31,9 @@ import (
"github.com/davecgh/go-spew/spew" "github.com/davecgh/go-spew/spew"
"github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/ethdb" "github.com/ethereum/go-ethereum/ethdb"
"github.com/ethereum/go-ethereum/rlp"
) )
func init() { func init() {
@ -505,8 +508,6 @@ func BenchmarkGet(b *testing.B) { benchGet(b, false) }
func BenchmarkGetDB(b *testing.B) { benchGet(b, true) } func BenchmarkGetDB(b *testing.B) { benchGet(b, true) }
func BenchmarkUpdateBE(b *testing.B) { benchUpdate(b, binary.BigEndian) } func BenchmarkUpdateBE(b *testing.B) { benchUpdate(b, binary.BigEndian) }
func BenchmarkUpdateLE(b *testing.B) { benchUpdate(b, binary.LittleEndian) } func BenchmarkUpdateLE(b *testing.B) { benchUpdate(b, binary.LittleEndian) }
func BenchmarkHashBE(b *testing.B) { benchHash(b, binary.BigEndian) }
func BenchmarkHashLE(b *testing.B) { benchHash(b, binary.LittleEndian) }
const benchElemCount = 20000 const benchElemCount = 20000
@ -549,19 +550,40 @@ func benchUpdate(b *testing.B, e binary.ByteOrder) *Trie {
return trie return trie
} }
func benchHash(b *testing.B, e binary.ByteOrder) { // Benchmarks the trie hashing. Since the trie caches the result of any operation,
trie := newEmpty() // we cannot use b.N as the number of hashing rouns, since all rounds apart from
k := make([]byte, 32) // the first one will be NOOP. As such, we'll use b.N as the number of account to
for i := 0; i < benchElemCount; i++ { // insert into the trie before measuring the hashing.
e.PutUint64(k, uint64(i)) func BenchmarkHash(b *testing.B) {
trie.Update(k, k) // Make the random benchmark deterministic
} random := rand.New(rand.NewSource(0))
b.ResetTimer() // Create a realistic account trie to hash
for i := 0; i < b.N; i++ { addresses := make([][20]byte, b.N)
trie.Hash() for i := 0; i < len(addresses); i++ {
for j := 0; j < len(addresses[i]); j++ {
addresses[i][j] = byte(random.Intn(256))
} }
} }
accounts := make([][]byte, len(addresses))
for i := 0; i < len(accounts); i++ {
var (
nonce = uint64(random.Int63())
balance = new(big.Int).Rand(random, new(big.Int).Exp(common.Big2, common.Big256, nil))
root = emptyRoot
code = crypto.Keccak256(nil)
)
accounts[i], _ = rlp.EncodeToBytes([]interface{}{nonce, balance, root, code})
}
// Insert the accounts into the trie and hash it
trie := newEmpty()
for i := 0; i < len(addresses); i++ {
trie.Update(crypto.Keccak256(addresses[i][:]), accounts[i])
}
b.ResetTimer()
b.ReportAllocs()
trie.Hash()
}
func tempDB() (string, Database) { func tempDB() (string, Database) {
dir, err := ioutil.TempDir("", "trie-bench") dir, err := ioutil.TempDir("", "trie-bench")