Few clean-ups

1 files changed, 2 insertions, 0 deletions

diff --git a/internal/options/options.go b/internal/options/options.go
index f7510ba..8e10686 100644
--- a/internal/options/options.go
+++ b/internal/options/options.go
@@ -10,6 +10,7 @@ type Options struct {
 	NumWorkers      int
 	MetricsInterval time.Duration
 	Timeout         time.Duration
+	MaxResponse     int64
 }
 
 func Parse() (opts Options) {
@@ -17,6 +18,7 @@ func Parse() (opts Options) {
 	flag.IntVar(&opts.NumWorkers, "w", 10, "number of parallel workers")
 	flag.DurationVar(&opts.Timeout, "t", 10*time.Second, "timeout for each website visit")
 	flag.DurationVar(&opts.MetricsInterval, "m", 5*time.Second, "how often to emit metrics")
+	flag.Int64Var(&opts.MaxResponse, "r", 128*1024*1024, "max response body size to accept")
 	flag.Parse()
 	return
 }