You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.4 KiB
122 lines
2.4 KiB
package types
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
tmpubsub "github.com/tendermint/tmlibs/pubsub"
|
|
)
|
|
|
|
func BenchmarkEventBus(b *testing.B) {
|
|
benchmarks := []struct {
|
|
name string
|
|
numClients int
|
|
randQueries bool
|
|
randEvents bool
|
|
}{
|
|
{"10Clients1Query1Event", 10, false, false},
|
|
{"100Clients", 100, false, false},
|
|
{"1000Clients", 1000, false, false},
|
|
|
|
{"10ClientsRandQueries1Event", 10, true, false},
|
|
{"100Clients", 100, true, false},
|
|
{"1000Clients", 1000, true, false},
|
|
|
|
{"10ClientsRandQueriesRandEvents", 10, true, true},
|
|
{"100Clients", 100, true, true},
|
|
{"1000Clients", 1000, true, true},
|
|
|
|
{"10Clients1QueryRandEvents", 10, false, true},
|
|
{"100Clients", 100, false, true},
|
|
{"1000Clients", 1000, false, true},
|
|
}
|
|
|
|
for _, bm := range benchmarks {
|
|
b.Run(bm.name, func(b *testing.B) {
|
|
benchmarkEventBus(bm.numClients, bm.randQueries, bm.randEvents, b)
|
|
})
|
|
}
|
|
}
|
|
|
|
func benchmarkEventBus(numClients int, randQueries bool, randEvents bool, b *testing.B) {
|
|
// for random* functions
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
eventBus := NewEventBusWithBufferCapacity(0) // set buffer capacity to 0 so we are not testing cache
|
|
eventBus.Start()
|
|
defer eventBus.Stop()
|
|
|
|
ctx := context.Background()
|
|
q := EventQueryNewBlock
|
|
|
|
for i := 0; i < numClients; i++ {
|
|
ch := make(chan interface{})
|
|
go func() {
|
|
for range ch {
|
|
}
|
|
}()
|
|
if randQueries {
|
|
q = randQuery()
|
|
}
|
|
eventBus.Subscribe(ctx, fmt.Sprintf("client-%d", i), q, ch)
|
|
}
|
|
|
|
eventType := EventNewBlock
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if randEvents {
|
|
eventType = randEvent()
|
|
}
|
|
|
|
eventBus.Publish(eventType, "Gamora")
|
|
}
|
|
}
|
|
|
|
var events = []string{EventBond,
|
|
EventUnbond,
|
|
EventRebond,
|
|
EventDupeout,
|
|
EventFork,
|
|
EventNewBlock,
|
|
EventNewBlockHeader,
|
|
EventNewRound,
|
|
EventNewRoundStep,
|
|
EventTimeoutPropose,
|
|
EventCompleteProposal,
|
|
EventPolka,
|
|
EventUnlock,
|
|
EventLock,
|
|
EventRelock,
|
|
EventTimeoutWait,
|
|
EventVote}
|
|
|
|
func randEvent() string {
|
|
return events[rand.Intn(len(events))]
|
|
}
|
|
|
|
var queries = []tmpubsub.Query{EventQueryBond,
|
|
EventQueryUnbond,
|
|
EventQueryRebond,
|
|
EventQueryDupeout,
|
|
EventQueryFork,
|
|
EventQueryNewBlock,
|
|
EventQueryNewBlockHeader,
|
|
EventQueryNewRound,
|
|
EventQueryNewRoundStep,
|
|
EventQueryTimeoutPropose,
|
|
EventQueryCompleteProposal,
|
|
EventQueryPolka,
|
|
EventQueryUnlock,
|
|
EventQueryLock,
|
|
EventQueryRelock,
|
|
EventQueryTimeoutWait,
|
|
EventQueryVote}
|
|
|
|
func randQuery() tmpubsub.Query {
|
|
return queries[rand.Intn(len(queries))]
|
|
}
|