main.go

package main

import (
	"errors"
	"fmt"
	"os"
	"time"

	"github.com/gonetx/httpit/pit"
	"github.com/spf13/cobra"
)

func main() {
	if err := rootCmd.Execute(); err != nil {
		_, _ = fmt.Fprintln(os.Stderr, err)
	}
}

var config pit.Config

func init() {
	rootCmd.Flags().SortFlags = false
	rootCmd.Flags().IntVarP(&config.Connections, "connections", "c", 128, "Maximum number of concurrent connections")
	rootCmd.Flags().IntVarP(&config.Count, "requests", "n", 0, "Number of requests (if specified, then ignore the --duration)")
	rootCmd.Flags().IntVar(&config.Qps, "qps", 0, "Highest qps value for a fixed benchmark (if specified, then ignore the -n|--requests)")
	rootCmd.Flags().DurationVarP(&config.Duration, "duration", "d", time.Second*10, "Duration of test")
	rootCmd.Flags().DurationVarP(&config.Timeout, "timeout", "t", time.Second*3, "Socket/request timeout")
	rootCmd.Flags().StringVarP(&config.Method, "method", "X", "GET", "Http request method")
	rootCmd.Flags().StringSliceVarP(&config.Headers, "header", "H", nil, headersUsage)
	rootCmd.Flags().StringVar(&config.Host, "host", "", "Override request host")
	rootCmd.Flags().BoolVarP(&config.DisableKeepAlives, "disableKeepAlives", "a", false, "Disable HTTP keep-alive, if true, will set header Connection: close")
	rootCmd.Flags().StringVarP(&config.Body, "body", "b", "", "Http request body string")
	rootCmd.Flags().StringVarP(&config.File, "file", "f", "", "Read http request body from file path")
	rootCmd.Flags().BoolVarP(&config.Stream, "stream", "s", false, "Use stream body to reduce memory usage")
	rootCmd.Flags().BoolVarP(&config.JSON, "json", "J", false, "Send json request by setting the Content-Type header to application/json")
	rootCmd.Flags().BoolVarP(&config.Form, "form", "F", false, "Send form request by setting the Content-Type header to application/x-www-form-urlencoded")
	rootCmd.Flags().BoolVarP(&config.Insecure, "insecure", "k", false, "Controls whether a client verifies the server's certificate chain and host name")
	rootCmd.Flags().StringVar(&config.Cert, "cert", "", "Path to the client's TLS Certificate")
	rootCmd.Flags().StringVar(&config.Key, "key", "", "Path to the client's TLS Certificate Private Key")
	rootCmd.Flags().StringVar(&config.HttpProxy, "httpProxy", "", "Http proxy address")
	rootCmd.Flags().StringVar(&config.SocksProxy, "socksProxy", "", "Socks proxy address")
	rootCmd.Flags().BoolVarP(&config.Pipeline, "pipeline", "p", false, "Use fasthttp pipeline client")
	rootCmd.Flags().BoolVar(&config.Follow, "follow", false, "Follow 30x Location redirects for debug mode")
	rootCmd.Flags().IntVar(&config.MaxRedirects, "maxRedirects", 0, "Max redirect count of following 30x, default is 30 (work with --follow)")
	rootCmd.Flags().BoolVarP(&config.Debug, "debug", "D", false, "Send request once and show request and response detail")
	rootCmd.Flags().BoolVar(&config.Http2, "http2", false, "Use HTTP/2.0")
}

var rootCmd = &cobra.Command{
	Use:           usage,
	Example:       example,
	Short:         "httpit is a rapid http benchmark tool",
	Version:       pit.Version,
	Args:          rootArgs,
	Run:           rootRun,
	SilenceErrors: true,
}

func rootArgs(_ *cobra.Command, args []string) error {
	if len(args) == 0 {
		return errors.New("missing url")
	}
	return nil
}

func rootRun(cmd *cobra.Command, args []string) {
	config.Url = args[0]
	config.Args = args[1:]
	if err := pit.New(config).Run(); err != nil {
		cmd.PrintErrln(err)
	}
}

const (
	usage   = `httpit [url|:port|/path] [k:v|k:=v ...]`
	example = `	httpit https://www.google.com -c1 -n5   =>   httpit -X GET https://www.google.com -c1 -n5
	httpit :3000 -c1 -n5                    =>   httpit -X GET http://localhost:3000 -c1 -n5
	httpit /foo -c1 -n5                     =>   httpit -X GET http://localhost/foo -c1 -n5
	httpit :3000 -c1 -n5 foo:=bar           =>   httpit -X GET http://localhost:3000 -c1 -n5 -H "Content-Type: application/json" -b='{"foo":"bar"}'
	httpit :3000 -c1 -n5 foo=bar            =>   httpit -X POST http://localhost:3000 -c1 -n5 -H "Content-Type: application/x-www-form-urlencoded" -b="foo=bar"`
	headersUsage = `HTTP request header with format "K: V", can be repeated
Examples:
	-H "k1: v1" -H k2:v2
	-H "k3: v3, k4: v4"`
)