博客

HttpGo工具(golang封装httpclient)

封装httpclient简化开发

依赖库

import . "github.com/NoBugBoy/httpgo/http"

使用方法

req := &Req{}
	body, err := req.Url("https://www.baidu.com").
		Method(http.MethodGet). //请求方式
		Header("user-agent", "Mozilla/5.0..."). //请求头
		Header("content-type", "application/json"). //请求头可以设置多个
		Timeout(3). //请求超时时间
		Retry(3). //请求错误重试次数
		Chunk(). //开启Chunk不会自动关闭response io,需要自己手动读取response body数据并关闭io 参考Test5分块传输
		Params(Query{ //请求参数,所有请求方式通用，如果get参数携带?id=1则优先使用url参数
			"id": 1,
		}).
		ProxyUrl("192.168.1.1:8080"). //配置要使用的代理ip
		ImportProxy(). //引入配置文件中的代理ip并随机使用
		Proxy(). //启用代理模式
		Build(). //创建request,一般不需单独调用，使用方法参考Test1压力测试
		Go(). //发起请求
		Body() //获取返回值string
	if err != nil {
		panic(err)
	}
	fmt.Println(body)
	request := req.Request //保留*http.Request对象以便有需要
	fmt.Println(request)
	response := req.Response //保留*http.Response对象以便有需要
	fmt.Println(response)
	transport := req.TransportSetting() //操作Transport进行参数调整
	fmt.Println(transport)

Build()方法的使用之压力测试，提前将请求对象创建好，然后一口气发起请求

var join sync.WaitGroup

// Test1 压力测试, 注意 ulimit 和 maxfd 的调优 /**
func Test1() {
	arr := make([]*Req, 0)
	for i := 0; i < 1000; i++ {
		join.Add(1)
		req := &Req{}
		x := req.Url("http://localhost:8080/get/1").
			Method(http.MethodGet).
			Header("Connection", "Keep-Alive").
			Header("Content-Type", "application/json").
			Timeout(30).
			Build()
		arr = append(arr, x)
	}
	for _, req := range arr {
	    //发起压力测试请求
		go runAndPrint(req)
	}
	join.Wait()
}

// 并发请求
func runAndPrint(r *Req) {
	defer join.Done()
	r.Go()
	//fmt.Println(.Body())
}

Chunk()模式的使用,可以自己控制如何使用Response

req := &Req{}
	re := req.ImportProxy().
		Method(http.MethodGet).
		Header("Connection", "Keep-Alive").
		Header("Transfer-Encoding", "chunked").
		Url("http://localhost:8080/get").
		Chunk().
		Timeout(30). //超时会关闭
		Go()
	fmt.Println(re.Response.Header)
	data := make([]byte, 1024)
	for {
		read, err := re.Response.Body.Read(data)
		fmt.Println("字节长度 ", read)
		if read > 0 {
			fmt.Print(string(data[:read]))
		}
		if err == io.EOF {
			break
		}
	}
	fmt.Println("Ok")

以上源码和测试均在github仓库中 https://github.com/NoBugBoy/httpgo 欢迎star提issues