Add throughput1 client library

cmd/msak-client/client.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"context"
+	"crypto/tls"
+	"flag"
+	"log"
+	"os"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/m-lab/msak/pkg/client"
+	"github.com/m-lab/msak/pkg/version"
+)
+
+var (
+	clientName    = "msak-client-go"
+	clientVersion = version.Version
+)
+
+var (
+	flagServer   = flag.String("server", "", "Server address")
+	flagStreams  = flag.Int("streams", 2, "Number of streams")
+	flagCC       = flag.String("cc", "bbr", "Congestion control algorithm to use")
+	flagDelay    = flag.Duration("delay", 0, "Delay between each stream")
+	flagDuration = flag.Duration("duration", 3*time.Second, "Length of the last stream")
+	flagScheme   = flag.String("scheme", "ws", "Websocket scheme (wss or ws)")
+	flagMID      = flag.String("mid", uuid.NewString(), "Measurement ID to use")
+	flagNoVerify = flag.Bool("no-verify", false, "Skip TLS certificate verification")
+	flagDebug    = flag.Bool("debug", false, "Enable debug logging")
+)
+
+func main() {
+	flag.Parse()
+
+	if float64(*flagStreams-1)*flagDelay.Seconds() >= flagDuration.Seconds() {
+		log.Print("Invalid configuration: please check streams, delay and duration and make sure they make sense.")
+		os.Exit(1)
+	}
+
+	cl := client.New(clientName, clientVersion)
+	cl.Dialer.TLSClientConfig = &tls.Config{
+		InsecureSkipVerify: *flagNoVerify,
+	}
+	cl.Server = *flagServer
+	cl.CongestionControl = *flagCC
+	cl.NumStreams = *flagStreams
+	cl.Scheme = *flagScheme
+	cl.MeasurementID = *flagMID
+	cl.Length = *flagDuration
+	cl.Delay = *flagDelay
+
+	cl.Emitter = &client.HumanReadable{
+		Debug: *flagDebug,
+	}
+
+	cl.Download(context.Background())
+	cl.Upload(context.Background())
+}