start writing test client

Signed-off-by: Alec Holmes <alecholmez@me.com>

Author: alecholmez

benchmarks/client/README.md

@@ -0,0 +1,3 @@
+# Benchmark Client
+
+This test client provides simulation of various workloads when communicating with the go-control-plane management server.

benchmarks/client/main.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	"log"
+	"os"
+	"runtime"
+	"time"
+
+	"github.com/envoyproxy/go-control-plane/benchmarks/client/xds"
+	"github.com/urfave/cli/v2"
+)
+
+func main() {
+	// Statically set the max procs
+	runtime.GOMAXPROCS(runtime.NumCPU())
+
+	app := &cli.App{
+		Name:  "xds-benchmark",
+		Usage: "xds benchmarking tool that simulates client workload",
+		Commands: []*cli.Command{
+			{
+				Name:    "run",
+				Aliases: []string{"r"},
+				Usage:   "run the benchmark with the provided duration",
+				Action: func(c *cli.Context) error {
+					arg := c.Args().First()
+					dur, err := time.ParseDuration(arg)
+					if err != nil {
+						return err
+					}
+
+					sess, err := xds.NewSession("localhost:50000")
+					if err != nil {
+						return err
+					}
+
+					return sess.Simulate(dur)
+				},
+			},
+		},
+	}
+
+	err := app.Run(os.Args)
+	if err != nil {
+		log.Fatal(err)
+	}
+}

benchmarks/client/xds/opts.go

@@ -0,0 +1,48 @@
+package xds
+
+// Options are configuration settings for the discovery object
+type Options struct {
+	NodeID        string
+	Zone          string
+	Cluster       string
+	ResourceNames []string // List of Envoy resource names to subscribe to
+	ResourceType  string   // ex: type.googleapis.com/envoy.api.v2.ClusterLoadAssignment
+}
+
+// Option follows the functional opts pattern
+type Option func(*Options)
+
+// WithNode will inject the node id into the configuration object
+func WithNode(id string) Option {
+	return func(o *Options) {
+		o.NodeID = id
+	}
+}
+
+// WithZone will specificy which zone to use in the xDS discovery request
+func WithZone(zone string) Option {
+	return func(o *Options) {
+		o.Zone = zone
+	}
+}
+
+// WithCluster will specificy which cluster the request is announcing as
+func WithCluster(cluster string) Option {
+	return func(o *Options) {
+		o.Cluster = cluster
+	}
+}
+
+// WithResourceNames will inject a list of resources the user wants to place watches on
+func WithResourceNames(names []string) Option {
+	return func(o *Options) {
+		o.ResourceNames = names
+	}
+}
+
+// WithResourceType will inject the specific resource type that a user wants to stream
+func WithResourceType(resource string) Option {
+	return func(o *Options) {
+		o.ResourceType = resource
+	}
+}

benchmarks/client/xds/xds.go

@@ -0,0 +1,45 @@
+package xds
+
+import (
+	"time"
+
+	"google.golang.org/grpc"
+)
+
+// Sess holds a grpc connection as well as config options to use during the simulation
+type Sess struct {
+	Session *grpc.ClientConn
+	Opts    Options
+}
+
+// NewSession will dial a new benchmarking session with the configured options
+func NewSession(url string, opts ...Option) (*Sess, error) {
+	var options Options
+	for _, o := range opts {
+		o(&options)
+	}
+
+	conn, err := grpc.Dial(url, grpc.WithBlock(), grpc.WithInsecure())
+	if err != nil {
+		return nil, err
+	}
+
+	return &Sess{
+		Session: conn,
+		Opts:    options,
+	}, nil
+}
+
+// Simulate will start an xDS stream which provides simulatest clients communicating with an xDS server
+func (s *Sess) Simulate(target time.Duration) error {
+	// Create a loop that will continually do work until the elapsed time as passed
+	for timeout := time.After(target); ; {
+		select {
+		case <-timeout:
+			return nil
+		default:
+			// Do some work
+
+		}
+	}
+}

benchmarks/client/xds/xds_test.go

@@ -0,0 +1 @@
+package xds

go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/pkg/profile v1.6.0
 	github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4
 	github.com/stretchr/testify v1.7.0
+	github.com/urfave/cli/v2 v2.3.0
 	go.opentelemetry.io/proto/otlp v0.7.0
 	google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013
 	google.golang.org/grpc v1.36.0

go.sum

@@ -47,10 +47,16 @@ github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZN
 github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4 h1:gQz4mCbXsO+nc9n1hCxHcGA3Zx3Eo+UHZoInFGUIXNM=
 github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
 github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=
+github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
+github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
+github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/urfave/cli/v2 v2.3.0 h1:qph92Y649prgesehzOrQjdWyxFOp/QVM+6imKHad91M=
+github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI=
 go.opentelemetry.io/proto/otlp v0.7.0 h1:rwOQPCuKAKmwGKq2aVNnYIibI6wnV7EvzgfTCzcdGg8=
 go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=