add timeout flag

README.md

@@ -74,12 +74,13 @@ Usage:
   whisperer [flags]
 
 Flags:
-  -a, --agent string     user agent (default "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:67.0) Gecko/20100101 Firefox/67.0")
-  -d, --delay duration   delay between requests
-  -g, --goroutines int   number of goroutines (default 1)
-  -h, --help             help for whisperer
-      --urls string      simple .txt file with URL's to visit (default "./urls.txt")
-  -v, --verbose          enables verbose mode
+  -a, --agent string       user agent (default "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:67.0) Gecko/20100101 Firefox/67.0")
+  -d, --delay duration     delay between requests (default 1s)
+  -g, --goroutines int     number of goroutines (default 1)
+  -h, --help               help for whisperer
+  -t, --timeout duration   max time to wait for a response before canceling the request (default 3s)
+      --urls string        simple .txt file with URL's to visit (default "./urls.txt")
+  -v, --verbose            enables verbose mode
 ```
 
 ## URLs file

cmd/root.go

@@ -18,14 +18,16 @@ var (
 	agent      string
 	delay      time.Duration
 	goroutines int
+	timeout    time.Duration
 	urls       string
 	verbose    bool
 )
 
 func init() {
 	rootCmd.PersistentFlags().StringVarP(&agent, "agent", "a", "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:67.0) Gecko/20100101 Firefox/67.0", "user agent")
-	rootCmd.PersistentFlags().DurationVarP(&delay, "delay", "d", 0, "delay between requests")
+	rootCmd.PersistentFlags().DurationVarP(&delay, "delay", "d", 1*time.Second, "delay between requests")
 	rootCmd.PersistentFlags().IntVarP(&goroutines, "goroutines", "g", 1, "number of goroutines")
+	rootCmd.PersistentFlags().DurationVarP(&timeout, "timeout", "t", 3*time.Second, "max time to wait for a response before canceling the request")
 	rootCmd.PersistentFlags().StringVar(&urls, "urls", "./urls.txt", "simple .txt file with URL's to visit")
 	rootCmd.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, "enables verbose mode")
 }
@@ -45,7 +47,7 @@ var rootCmd = &cobra.Command{
 			return fmt.Errorf("while reading URLs from %q: %v", urls, err)
 		}
 
-		client := &http.Client{}
+		client := &http.Client{Timeout: timeout}
 		sema := make(chan struct{}, goroutines)
 		seed := rand.NewSource(time.Now().Unix())
 		r := rand.New(seed)