GO1.5实现简单的http并发请求,支持:GET、POST、HEAD、PUT

package main

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

func main() {
	connection := flag.Int("c", 200, "-c N")
	timeout := flag.Int("o", 5, "-o N")
	timeover := flag.Int("t", 5, "-t N")
	printresult := flag.Bool("p", false, "-p false")
	method := flag.String("m", "GET", "-m GET")
	url := flag.String("u", "http://127.0.0.1", "-u http://127.0.0.1")
	flag.Parse()
	var Count int32
	defer func() {
		if !*printresult {
			fmt.Println("成功响应:", Count)
		}
	}()
	T := time.Tick(time.Duration(*timeover) * time.Second)
	var result chan string = make(chan string, 10)
	t := time.Duration(*timeout) * time.Second
	Client := http.Client{Timeout: t}
	for i := 0; i < *connection; i++ {
		go func() {
			req, _ := http.NewRequest(*method, *url, nil)
			resp, _ := Client.Do(req)
			defer resp.Body.Close()
			if resp.StatusCode == 200 {
				b, _ := ioutil.ReadAll(resp.Body)
				result <- string(b)
				atomic.AddInt32(&Count, int32(1))
			}
		}()
	}
	for {
		select {
		case x := <-result:
			if *printresult {
				fmt.Print(x)
			}
		case <-T:
			return
		}
	}
}

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值