add sleep and proxy flag

Author: poolqa

bot/bot.go

@@ -13,11 +13,12 @@ import (
 )
 
 type OrderBot struct {
-	ID      int64
-	Client  *fasthttp.Client
-	procCnt int64
-	doneCnt int64
-	gId     string
+	ID            int64
+	Client        *fasthttp.Client
+	SleepInterval time.Duration
+	procCnt       int64
+	doneCnt       int64
+	gId           string
 }
 
 func (bot *OrderBot) GetRoutineId() string {
@@ -58,6 +59,7 @@ func (bot *OrderBot) StandBy(ctx context.Context, wg *sync.WaitGroup, runCnt int
 		err := bot.Client.Do(req, resp)
 		tps.ETime = time.Now()
 		if err != nil {
+			fmt.Printf("error:%v\n", err)
 			tps.RespStatus = fasthttp.StatusBadRequest
 		} else {
 			tps.RespStatus = resp.StatusCode()
@@ -67,6 +69,9 @@ func (bot *OrderBot) StandBy(ctx context.Context, wg *sync.WaitGroup, runCnt int
 
 		performance.Push(tps)
 		bot.doneCnt++
+		if bot.SleepInterval > 0 {
+			time.Sleep(bot.SleepInterval)
+		}
 	}
 	wg.Done()
 }

flagArg/flag.go

@@ -24,6 +24,8 @@ func InitFlag(f *FlagArg) {
 	//flag.BoolVar(&f.ReportDetail, "r", false, "show those routine's requests detail at report log,\nbut it wall use more and more memory.")
 	flag.BoolVar(&f.PreHeat, "w", false, "waiting all routines stand-by.")
 	flag.Int64Var(&f.PrintInterval, "i", 1, "print interval for process request count time.")
+	flag.Int64Var(&f.SleepInterval, "sleep", 0, "sleep msec at every request is finished.")
+	flag.StringVar(&f.ProxyAddr, "proxy", "", "proxy address.")
 	flag.BoolVar(&f.Help, "h", false, "this help")
 
 	flag.Usage = f.Usage
@@ -40,6 +42,8 @@ type FlagArg struct {
 	DebugLog      bool
 	ReportDetail  bool
 	PrintInterval int64
+	SleepInterval int64
+	ProxyAddr     string
 	Help          bool
 }
 

main.go

@@ -2,9 +2,11 @@ package main
 
 import (
 	"context"
+	"crypto/tls"
 	"flag"
 	"fmt"
 	"github.com/valyala/fasthttp"
+	"github.com/valyala/fasthttp/fasthttpproxy"
 	"golang.org/x/text/language"
 	"golang.org/x/text/message"
 	"math/rand"
@@ -62,7 +64,15 @@ func main() {
 
 	BotArray = make([]BotInterface, flagArg.FlagArgs.RoutineCount)
 
-	httpClient := &fasthttp.Client{MaxConnsPerHost: int(flagArg.FlagArgs.RoutineCount)}
+	httpClient := &fasthttp.Client{
+		MaxConnsPerHost: int(flagArg.FlagArgs.RoutineCount),
+		TLSConfig: &tls.Config{
+			InsecureSkipVerify: true,
+		},
+	}
+	if flagArg.FlagArgs.ProxyAddr != "" {
+		httpClient.Dial = fasthttpproxy.FasthttpHTTPDialer(flagArg.FlagArgs.ProxyAddr)
+	}
 	wg := &sync.WaitGroup{}
 	done := make(chan struct{})
 	ctx, runRoutine := context.WithCancel(context.Background())
@@ -75,8 +85,9 @@ func main() {
 	// request routine start
 	for idx := int64(0); idx < flagArg.FlagArgs.RoutineCount; idx++ {
 		BotArray[idx] = &bot.OrderBot{
-			ID:     idx + 1,
-			Client: httpClient,
+			ID:            idx + 1,
+			Client:        httpClient,
+			SleepInterval: time.Duration(flagArg.FlagArgs.SleepInterval) * time.Millisecond,
 		}
 		wg.Add(1)
 		go BotArray[idx].StandBy(ctx, wg, flagArg.FlagArgs.RunCount)