refactor: move prombench to promrw

Author: ernado

cmd/promrw/README.md

@@ -9,7 +9,7 @@ go run ./cmd/promrw record -o /tmp/requests.rwq --d 10m --addr="127.0.0.1:8080"
 
 Start load generator:
 ```bash
-go run ./cmd/prombench --targetsCount=100 --scrapeInterval=1s http://127.0.0.1:8080
+go run ./cmd/promrw bench --targetsCount=100 --scrapeInterval=1s http://127.0.0.1:8080
 ```
 
 Prometheus remote write requests will be recorded to `/tmp/requests.rwq` file.

cmd/promrw/main.go

@@ -13,11 +13,14 @@ import (
 func main() {
 	rootCmd := &cobra.Command{
 		Use:   "promrw",
-		Short: "promrw is a tool for recording and sending prometheus remote write requests",
+		Short: "promrw is a benchmarking suite for prometheus remote write",
+
+		SilenceUsage: true,
 	}
 	rootCmd.AddCommand(
-		newRecorderCommand(),
-		newSenderCommand(),
+		newRecordCommand(),
+		newReplayCommand(),
+		newBenchCommand(),
 	)
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
 	defer stop()

cmd/promrw/recorder.go renamed to cmd/promrw/record.go

@@ -20,7 +20,7 @@ import (
 	"golang.org/x/sync/errgroup"
 )
 
-type Recorder struct {
+type Record struct {
 	Addr     string
 	Duration time.Duration
 	Output   string
@@ -30,7 +30,7 @@ type Recorder struct {
 	bytes    atomic.Uint64
 }
 
-func (r *Recorder) read(req *http.Request) ([]byte, error) {
+func (r *Record) read(req *http.Request) ([]byte, error) {
 	if t := req.Header.Get("Content-Type"); t != "application/x-protobuf" {
 		return nil, errors.Errorf("unsupported content type %q", t)
 	}
@@ -71,7 +71,7 @@ func fmtInt(v int) string {
 	return s
 }
 
-func (r *Recorder) Run(ctx context.Context) (rerr error) {
+func (r *Record) Run(ctx context.Context) (rerr error) {
 	fmt.Println("listening on", "http://"+r.Addr)
 	fmt.Println("writing to", r.Output, "for", r.Duration)
 	f, err := os.Create(r.Output)
@@ -158,8 +158,8 @@ func (r *Recorder) Run(ctx context.Context) (rerr error) {
 	return g.Wait()
 }
 
-func newRecorderCommand() *cobra.Command {
-	var recorder Recorder
+func newRecordCommand() *cobra.Command {
+	var recorder Record
 	cmd := &cobra.Command{
 		Use:   "record",
 		Args:  cobra.NoArgs,

cmd/promrw/sender.go renamed to cmd/promrw/replay.go

@@ -17,14 +17,14 @@ import (
 	"golang.org/x/sync/errgroup"
 )
 
-type Sender struct {
+type Replay struct {
 	Target  string
 	Source  string
 	Workers int
 }
 
-func (s *Sender) Run(ctx context.Context) error {
-	f, err := os.Open(s.Source)
+func (r *Replay) Run(ctx context.Context) error {
+	f, err := os.Open(r.Source)
 	if err != nil {
 		return errors.Wrap(err, "open file")
 	}
@@ -50,7 +50,7 @@ func (s *Sender) Run(ctx context.Context) error {
 	fn := func(data []byte) error {
 		ctx, cancel := context.WithTimeout(ctx, time.Minute)
 		defer cancel()
-		req, err := http.NewRequestWithContext(ctx, http.MethodPost, s.Target, bytes.NewReader(data))
+		req, err := http.NewRequestWithContext(ctx, http.MethodPost, r.Target, bytes.NewReader(data))
 		if err != nil {
 			return errors.Wrap(err, "create request")
 		}
@@ -75,7 +75,7 @@ func (s *Sender) Run(ctx context.Context) error {
 
 	g, ctx := errgroup.WithContext(ctx)
 	inputs := make(chan []byte)
-	for i := 0; i < s.Workers; i++ {
+	for i := 0; i < r.Workers; i++ {
 		g.Go(func() error {
 			for {
 				select {
@@ -121,18 +121,18 @@ func (s *Sender) Run(ctx context.Context) error {
 	return nil
 }
 
-func newSenderCommand() *cobra.Command {
-	var sender Sender
+func newReplayCommand() *cobra.Command {
+	var replay Replay
 	cmd := &cobra.Command{
-		Use:   "send",
+		Use:   "replay",
 		Short: "Send recorded requests to remote write server",
 		Args:  cobra.NoArgs,
 		RunE: func(cmd *cobra.Command, args []string) error {
-			return sender.Run(cmd.Context())
+			return replay.Run(cmd.Context())
 		},
 	}
-	cmd.Flags().StringVar(&sender.Target, "target", "http://127.0.0.1:19291", "Target server")
-	cmd.Flags().StringVarP(&sender.Source, "input", "i", "requests.rwq", "Source file")
-	cmd.Flags().IntVarP(&sender.Workers, "workers", "j", 8, "Number of workers")
+	cmd.Flags().StringVar(&replay.Target, "target", "http://127.0.0.1:19291", "Target server")
+	cmd.Flags().StringVarP(&replay.Source, "input", "i", "requests.rwq", "Source file")
+	cmd.Flags().IntVarP(&replay.Workers, "workers", "j", 8, "Number of workers")
 	return cmd
 }