status-im
/
go-libp2p
mirror of https://github.com/status-im/go-libp2p.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

better benchmark tests

This commit is contained in:
Aarsh Shah 2020-04-29 15:20:21 +05:30
parent 4cca3868d6
commit 1c8319d290
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
p2p/security/noise/benchmark_test.go
View File

@ -181,27 +181,31 @@ func benchDataTransfer(b *benchenv, dataSize int64, m testMode) {
var totalBytes int64
var totalTime time.Duration
plainTextBufs := make([][]byte, 62)
rbufs := make(map[int][]byte)
for i := 0; i < len(plainTextBufs); i++ {
// plaintext will be 2 KB to 63 KB
plainTextBufs[i] = make([]byte, (i+2)*1024)
switch m {
case readBufferGtEncMsg:
rbufs[i] = make([]byte, len(plainTextBufs[i])+poly1305.TagSize+1)
case readBufferGtPlainText:
rbufs[i] = make([]byte, len(plainTextBufs[i])+1)
case readBufferLtPlainText:
rbufs[i] = make([]byte, len(plainTextBufs[i])-2)
}
}
b.ResetTimer()
b.ReportAllocs()
for i := 0; i < b.N; i++ {
initSession, respSession := b.connect(true)
b.StopTimer()
var rbuf []byte
plainTextBuf := make([]byte, randInRange(2*1024, 63*1024))
switch m {
case readBufferGtEncMsg:
rbuf = make([]byte, len(plainTextBuf)+poly1305.TagSize+1)
case readBufferGtPlainText:
rbuf = make([]byte, len(plainTextBuf)+1)
case readBufferLtPlainText:
rbuf = make([]byte, len(plainTextBuf)-2)
}
b.StartTimer()
start := time.Now()
err := pipeRandom(b.rndSrc, initSession, respSession, dataSize, plainTextBuf, rbuf)
bufi := i % len(plainTextBufs)
err := pipeRandom(b.rndSrc, initSession, respSession, dataSize, plainTextBufs[bufi], rbufs[bufi])
if err != nil {
b.Fatalf("error sending random data: %s", err)
}