cwinfo/yggdrasil-go
5
0
Fork 0
mirror of https://github.com/cwinfo/yggdrasil-go.git synced 2024-09-20 04:52:33 +00:00
Code Releases Activity

Add a simple transfer benchmark

Browse Source
This commit is contained in:
Aleksander Mistewicz 2019-09-28 14:20:57 +02:00
parent 8677a042cf
commit 21b236771b
1 changed files with 58 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/yggdrasil/core_test.go
View File

@ -30,7 +30,7 @@ func GetLoggerWithPrefix(prefix string) *log.Logger {
return l return l
} }
func CreateAndConnectTwo(t *testing.T) (*Core, *Core) { func CreateAndConnectTwo(t testing.TB) (*Core, *Core) {
nodeA := Core{} nodeA := Core{}
_, err := nodeA.Start(GenerateConfig(), GetLoggerWithPrefix("A: ")) _, err := nodeA.Start(GenerateConfig(), GetLoggerWithPrefix("A: "))
if err != nil { if err != nil {
@ -74,7 +74,7 @@ func TestCore_Start_Connect(t *testing.T) {
CreateAndConnectTwo(t) CreateAndConnectTwo(t)
} }
func CreateEchoListener(t *testing.T, nodeA *Core, bufLen int) chan struct{} { func CreateEchoListener(t testing.TB, nodeA *Core, bufLen int, repeats int) chan struct{} {
// Listen // Listen
listener, err := nodeA.ConnListen() listener, err := nodeA.ConnListen()
if err != nil { if err != nil {
@ -91,6 +91,8 @@ func CreateEchoListener(t *testing.T, nodeA *Core, bufLen int) chan struct{} {
} }
defer conn.Close() defer conn.Close()
buf := make([]byte, bufLen) buf := make([]byte, bufLen)
for i := 0; i < repeats; i++ {
n, err := conn.Read(buf) n, err := conn.Read(buf)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
@ -104,6 +106,7 @@ func CreateEchoListener(t *testing.T, nodeA *Core, bufLen int) chan struct{} {
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }
}
done <- struct{}{} done <- struct{}{}
}() }()
@ -114,7 +117,7 @@ func TestCore_Start_Transfer(t *testing.T) {
nodeA, nodeB := CreateAndConnectTwo(t) nodeA, nodeB := CreateAndConnectTwo(t)
msgLen := 1500 msgLen := 1500
done := CreateEchoListener(t, nodeA, msgLen) done := CreateEchoListener(t, nodeA, msgLen, 1)
if !WaitConnected(nodeA, nodeB) { if !WaitConnected(nodeA, nodeB) {
t.Fatal("nodes did not connect") t.Fatal("nodes did not connect")
@ -146,3 +149,43 @@ func TestCore_Start_Transfer(t *testing.T) {
} }
<-done <-done
} }
func BenchmarkCore_Start_Transfer(b *testing.B) {
nodeA, nodeB := CreateAndConnectTwo(b)
msgLen := 1500 // typical MTU
done := CreateEchoListener(b, nodeA, msgLen, b.N)
if !WaitConnected(nodeA, nodeB) {
b.Fatal("nodes did not connect")
}
// Dial
dialer, err := nodeB.ConnDialer()
if err != nil {
b.Fatal(err)
}
conn, err := dialer.Dial("nodeid", nodeA.NodeID().String())
if err != nil {
b.Fatal(err)
}
defer conn.Close()
msg := make([]byte, msgLen)
rand.Read(msg)
buf := make([]byte, msgLen)
b.SetBytes(int64(b.N * msgLen))
b.ResetTimer()
for i := 0; i < b.N; i++ {
conn.Write(msg)
if err != nil {
b.Fatal(err)
}
_, err = conn.Read(buf)
if err != nil {
b.Fatal(err)
}
}
<-done
}