Skip to content

Commit

Permalink
Adding large benchmark with some debug output
Browse files Browse the repository at this point in the history
  • Loading branch information
Preetha Appan committed Oct 5, 2017
1 parent d1caa6c commit f5742cb
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions bench_test.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package yamux

import (
"fmt"
"testing"
)

Expand Down Expand Up @@ -79,3 +80,49 @@ func BenchmarkSendRecv(b *testing.B) {
}
<-doneCh
}

func BenchmarkSendRecvLarge(b *testing.B) {
client, server := testClientServer()
defer client.Close()
defer server.Close()

const sendSize = 512 * 1024 * 1024
const recvSize = 4 * 1024

sendBuf := make([]byte, sendSize)
recvBuf := make([]byte, recvSize)

b.ResetTimer()
recvDone := make(chan struct{})

go func() {
stream, err := server.AcceptStream()
if err != nil {
return
}
defer stream.Close()
for i := 0; i < b.N; i++ {
for j := 0; j < sendSize/recvSize; j++ {
if _, err := stream.Read(recvBuf); err != nil {
b.Fatalf("err: %v", err)
}
}

fmt.Printf("Capacity of rcv buffer = %v, length of rcv window = %v\n", stream.recvBuf.Cap(), stream.recvWindow)

}
close(recvDone)
}()

stream, err := client.Open()
if err != nil {
b.Fatalf("err: %v", err)
}
defer stream.Close()
for i := 0; i < b.N; i++ {
if _, err := stream.Write(sendBuf); err != nil {
b.Fatalf("err: %v", err)
}
}
<-recvDone
}

0 comments on commit f5742cb

Please sign in to comment.