Add throughput1 client library

cmd/msak-client/client.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log"
+
+	"github.com/google/uuid"
+	"github.com/m-lab/msak/pkg/client"
+	"github.com/m-lab/msak/pkg/version"
+)
+
+const clientName = "msak-client-go"
+
+var clientVersion = version.Version
+
+var (
+	flagServer   = flag.String("server", "", "Server address")
+	flagStreams  = flag.Int("streams", client.DefaultStreams, "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", client.DefaultLength, "Length of the last stream")
+	flagScheme   = flag.String("scheme", client.DefaultScheme, "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()
+
+	// For a given number of streams, there will be streams-1 delays. This makes
+	// sure that all the streams can at least start with the current configuration.
+	if float64(*flagStreams-1)*flagDelay.Seconds() >= flagDuration.Seconds() {
+		log.Fatal("Invalid configuration: please check streams, delay and duration and make sure they make sense.")
+	}
+
+	config := client.Config{
+		Server:            *flagServer,
+		Scheme:            *flagScheme,
+		NumStreams:        *flagStreams,
+		CongestionControl: *flagCC,
+		Delay:             *flagDelay,
+		Length:            *flagDuration,
+		MeasurementID:     *flagMID,
+		Emitter: client.HumanReadable{
+			Debug: *flagDebug,
+		},
+		NoVerify: *flagNoVerify,
+	}
+
+	cl := client.New(clientName, clientVersion, config)
+
+	cl.Download(context.Background())
+	cl.Upload(context.Background())
+}