-r print response

hey.go

@@ -58,6 +58,8 @@ var (
 	z = flag.Duration("z", 0, "")
 	p = flag.Duration("p", time.Second, "")
 
+	r = flag.Bool("r", false, "")
+
 	h2   = flag.Bool("h2", false, "")
 	cpus = flag.Int("cpus", runtime.GOMAXPROCS(-1), "")
 
@@ -94,6 +96,7 @@ Options:
   -x  HTTP Proxy address as host:port.
   -h2 Enable HTTP/2.
   -p  Preview results interval. Default is 1s.
+  -r  Print http response
 
   -host	HTTP Host header.
 
@@ -124,6 +127,7 @@ func main() {
 	q := *q
 	dur := *z
 	report := *p
+	debug := *r
 
 	if dur > 0 {
 		num = math.MaxInt32
@@ -238,6 +242,7 @@ func main() {
 		ProxyAddr:          proxyURL,
 		Output:             *output,
 		ReportPreview:      report,
+		Debug:              debug,
 	}
 	w.Init()
 

requester/requester.go

@@ -22,6 +22,7 @@ import (
 	"github.com/gosuri/uilive"
 	"io"
 	"io/ioutil"
+	"log"
 	"net/http"
 	"net/http/httptrace"
 	"net/url"
@@ -99,6 +100,8 @@ type Work struct {
 	stopCh   chan struct{}
 	start    time.Duration
 
+	Debug bool
+
 	report *report
 	// Preview output the current result in advance
 	ReportPreview time.Duration
@@ -223,7 +226,12 @@ func (b *Work) makeRequest(c *http.Client) {
 	if err == nil {
 		size = resp.ContentLength
 		code = resp.StatusCode
-		io.Copy(ioutil.Discard, resp.Body)
+		if b.Debug {
+			res, _ := io.ReadAll(resp.Body)
+			log.Printf("res %s", string(res))
+		} else {
+			_, _ = io.Copy(io.Discard, resp.Body)
+		}
 		resp.Body.Close()
 	}
 	t := now()