prysm-pulse/slasher/db/kv/benchmark_test.go
Shay Zluf fd19fd10a9
New spanner db structure (#6061)
* new spanner db structure

* lint fixes

* go mod fix

* fix iface

* remove unused

* remove extra line

* change from db

* exported field

* exported field

* revert to original

* fix

* ivan feedback

* ivan feedback

* ivan feedback

* revert mod changes

* fix db impl

* gaz

* import fix

* Try to fix tests

* ivan feedback

* new epoch store

* added comment

* fix error

Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
Co-authored-by: Ivan Martinez <ivanthegreatdev@gmail.com>
2020-06-02 10:41:21 -04:00

137 lines
3.5 KiB
Go

package kv
import (
"context"
"flag"
"testing"
"github.com/prysmaticlabs/prysm/slasher/detection/attestations/types"
"github.com/urfave/cli/v2"
)
const (
benchmarkValidator = 300000
)
func BenchmarkStore_SaveEpochSpans(b *testing.B) {
ctx := context.Background()
sigBytes := [2]byte{}
app := cli.App{}
set := flag.NewFlagSet("test", 0)
db := setupDB(b, cli.NewContext(&app, set, nil))
es := EpochStore{}
err := es.SetValidatorSpan(ctx, benchmarkValidator, types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
for i := 0; i < benchmarkValidator; i++ {
err = es.SetValidatorSpan(ctx, uint64(i), types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
}
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
err := db.SaveEpochSpans(ctx, uint64(i%54000), es)
if err != nil {
b.Fatalf("Save validator span map failed: %v", err)
}
}
}
func BenchmarkStore_EpochSpans(b *testing.B) {
app := cli.App{}
set := flag.NewFlagSet("test", 0)
db := setupDB(b, cli.NewContext(&app, set, nil))
ctx := context.Background()
sigBytes := [2]byte{}
es := EpochStore{}
err := es.SetValidatorSpan(ctx, benchmarkValidator, types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
for i := 0; i < benchmarkValidator; i++ {
err = es.SetValidatorSpan(ctx, uint64(i), types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
}
b.Log(len(es))
for i := 0; i < 200; i++ {
err := db.SaveEpochSpans(ctx, uint64(i), es)
if err != nil {
b.Fatalf("Save validator span map failed: %v", err)
}
}
b.Log(db.db.Info())
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := db.EpochSpans(ctx, uint64(i%200))
if err != nil {
b.Fatalf("Read validator span map failed: %v", err)
}
}
}
func BenchmarkStore_GetValidatorSpan(b *testing.B) {
ctx := context.Background()
sigBytes := [2]byte{}
es := EpochStore{}
err := es.SetValidatorSpan(ctx, benchmarkValidator, types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
for i := 0; i < benchmarkValidator; i++ {
err = es.SetValidatorSpan(ctx, uint64(i), types.Span{MinSpan: uint16(i), MaxSpan: uint16(benchmarkValidator - i), SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
}
b.Log(len(es))
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, err := es.GetValidatorSpan(ctx, uint64(i%benchmarkValidator))
if err != nil {
b.Fatalf("Read validator span map failed: %v", err)
}
}
}
func BenchmarkStore_SetValidatorSpan(b *testing.B) {
ctx := context.Background()
sigBytes := [2]byte{}
es := EpochStore{}
err := es.SetValidatorSpan(ctx, benchmarkValidator, types.Span{MinSpan: 1, MaxSpan: 2, SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
for i := 0; i < benchmarkValidator; i++ {
err = es.SetValidatorSpan(ctx, uint64(i), types.Span{MinSpan: uint16(i), MaxSpan: uint16(benchmarkValidator - i), SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Error(err)
}
}
b.Log(len(es))
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
err := es.SetValidatorSpan(ctx, uint64(i%benchmarkValidator), types.Span{MinSpan: uint16(i), MaxSpan: uint16(benchmarkValidator - i), SigBytes: sigBytes, HasAttested: true})
if err != nil {
b.Fatalf("Read validator span map failed: %v", err)
}
}
}