http_body_test.go

gistfile1.txt

package main

import (
	"flag"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"sync"
)

func main() {
	url := flag.String("url", "", "url")
	concurent := flag.Int("c", 1, "concurent")
	cnt := flag.Int("n", 1, "count")
	body := flag.Bool("body", false, "body")
	clo := flag.Bool("close", false, "close")
	flag.Parse()
	fmt.Printf("url:%v\nread body:%v\nclose body:%v\n", *url, *body, *clo)
	wg := sync.WaitGroup{}
	taskChan := make(chan struct{}, *concurent)
	closeChan := make(chan struct{})
	fn := func() {
		for {
			select {
			case <-taskChan:
			case <-closeChan:
				return
			}
			res, err := http.Get(*url)
			if err != nil {
				panic(err)
			}
			if *body {
				io.Copy(ioutil.Discard, res.Body)
			}
			if *clo {
				res.Body.Close()
			}
			wg.Done()
		}
	}
	wg.Add(*cnt)
	for i := 0; i < *concurent; i++ {
		go fn()
	}
	for i := 0; i < *cnt; i++ {
		taskChan <- struct{}{}
	}
	wg.Wait()
	close(closeChan)
	fmt.Println("Press the Enter Key to terminate")
	fmt.Scanln() // wait for Enter Key
}