分享
GO1.5实现简单的http并发请求,支持:GET、POST、HEAD、PUT
fyxichen · · 8075 次点击 · · 开始浏览这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
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
}
}
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
有疑问加站长微信联系(非本文作者)
入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889
关注微信8075 次点击 ∙ 1 赞
添加一条新回复
(您需要 后才能回复 没有账号 ?)
- 请尽量让自己的回复能够对别人有帮助
- 支持 Markdown 格式, **粗体**、~~删除线~~、
`单行代码` - 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
- 图片支持拖拽、截图粘贴等方式上传
收入到我管理的专栏 新建专栏
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
}
}
}
版权声明:本文为博主原创文章,未经博主允许不得转载。