support --no-stat flag

internal/option.go

@@ -93,6 +93,7 @@ type OutputOptions struct {
 	Quiet       bool   `short:"q" long:"quiet" description:"Bool, Quiet" config:"quiet"`
 	NoColor     bool   `long:"no-color" description:"Bool, no color" config:"no-color"`
 	NoBar       bool   `long:"no-bar" description:"Bool, No progress bar" config:"no-bar"`
+	NoStat      bool   `long:"no-stat" description:"Bool, No stat file output" config:"no-stat"`
 }
 
 type RequestOptions struct {
@@ -719,10 +720,13 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
 	if err != nil {
 		return nil, err
 	}
-	r.StatFile.Mod = os.O_WRONLY | os.O_CREATE
-	err = r.StatFile.Init()
-	if err != nil {
-		return nil, err
+
+	if !opt.NoStat {
+		r.StatFile.Mod = os.O_WRONLY | os.O_CREATE
+		err = r.StatFile.Init()
+		if err != nil {
+			return nil, err
+		}
 	}
 	return r, nil
 }

internal/runner.go

@@ -154,7 +154,9 @@ func (r *Runner) Prepare(ctx context.Context) error {
 		r.Pools, err = ants.NewPoolWithFunc(r.PoolSize, func(i interface{}) {
 			t := i.(*Task)
 			if t.origin != nil && t.origin.End == t.origin.Total {
-				r.StatFile.SafeWrite(t.origin.Json())
+				if r.StatFile != nil {
+					r.StatFile.SafeWrite(t.origin.Json())
+				}
 				r.Done()
 				return
 			}
@@ -231,10 +233,14 @@ Loop:
 			if len(r.taskCh) > 0 {
 				for t := range r.taskCh {
 					stat := pkg.NewStatistor(t.baseUrl)
-					r.StatFile.SafeWrite(stat.Json())
+					if r.StatFile != nil {
+						r.StatFile.SafeWrite(stat.Json())
+					}
 				}
 			}
-			logs.Log.Importantf("already save all stat to %s", r.StatFile.Filename)
+			if r.StatFile != nil {
+				logs.Log.Importantf("already save all stat to %s", r.StatFile.Filename)
+			}
 			break Loop
 		case t, ok := <-r.taskCh:
 			if !ok {