yggdrasil-go/misc/tests/bandwidth.go

package main

import "fmt"
import "net"
import "time"

func main() {
	addr, err := net.ResolveTCPAddr("tcp", "[::1]:9001")
	if err != nil {
		panic(err)
	}
	listener, err := net.ListenTCP("tcp", addr)
	if err != nil {
		panic(err)
	}
	defer listener.Close()

	packetSize := 65535
	numPackets := 65535

	go func() {
		send, err := net.DialTCP("tcp", nil, addr)
		if err != nil {
			panic(err)
		}
		defer send.Close()
		msg := make([]byte, packetSize)
		for idx := 0; idx < numPackets; idx++ {
			send.Write(msg)
		}
	}()

	start := time.Now()
	//msg := make([]byte, 1280)
	sock, err := listener.AcceptTCP()
	if err != nil {
		panic(err)
	}
	defer sock.Close()
	read := 0
	buf := make([]byte, packetSize)
	for {
		n, err := sock.Read(buf)
		read += n
		if err != nil {
			break
		}
	}
	timed := time.Since(start)

	fmt.Printf("%f packets per second\n", float64(numPackets)/timed.Seconds())
	fmt.Printf("%f bits/sec\n", 8*float64(read)/timed.Seconds())
}