UDS throughput benchmark (Golang)

uds_test.go

package test

import (
	"fmt"
	"net"
	"os"
	"testing"
	"time"
)

var (
	done    chan bool
	msgSize int = 1024
)

func read(n net.Listener) {
	defer n.Close()
	defer os.Remove("/tmp/uds_test.sock")

	c, err := n.Accept()
	if err != nil {
		panic(err)
	}

	buf := make([]byte, msgSize)
	msgs := int64(1)
	start := time.Now().UnixNano()
	for {
		_, err := c.Read(buf[:])
		if err != nil {
			panic(err)
		}

		if msgs%100000 == 0 {
			sample := time.Now().UnixNano()
			fmt.Printf("Throughput: %f/sec (%d message size)\n", float64(msgs*int64(time.Second))/float64((sample-start)), msgSize)
		}
		msgs++
	}
}

func write() {
	c, err := net.Dial("unix", "/tmp/uds_test.sock")
	defer c.Close()

	buf := make([]byte, msgSize)

	for {
		_, err = c.Write(buf)
		if err != nil {
			panic(err)
		}
	}
	done <- true
}

func TestThroughput(t *testing.T) {
	writers := 2
	readers := 2
	done = make(chan bool, writers)
	n, err := net.Listen("unix", "/tmp/uds_test.sock")
	if err != nil {
		panic(err)
	}
	for i := 0; i < readers; i++ {
		go read(n)
	}

	for i := 0; i < writers; i++ {
		go write()
	}
	<-done
}