package task
import ( "bytes" "encoding/json" "fmt" "io/ioutil" "net/http" "net/url" "strings" "time")
var Client http.Client = clients()
// Res 模拟响应结构// @Description:type Res struct { Have string `json:"Have"`}
// Get 获取GET请求// @Description:// @param uri// @param args// @return *http.Requestfunc Get(uri string, args map[string]interface{}) *http.Request { uri = uri + ToValues(args) request, _ := http.NewRequest("get", uri, nil) return request}
// PostForm POST接口form表单// @Description:// @param path// @param args// @return *http.Requestfunc PostForm(path string, args map[string]interface{}) *http.Request { request, _ := http.NewRequest("post", path, strings.NewReader(ToValues(args))) return request}
// PostJson POST请求,JSON参数// @Description:// @param path// @param args// @return *http.Requestfunc PostJson(path string, args map[string]interface{}) *http.Request { marshal, _ := json.Marshal(args) request, _ := http.NewRequest("post", path, bytes.NewReader(marshal))
return request}
// ToValues 将map解析成HTTP参数,用于GET和POST form表单// @Description:// @param args// @return stringfunc ToValues(args map[string]interface{}) string { if args != nil && len(args) > 0 { params := url.Values{} for k, v := range args { params.Set(k, fmt.Sprintf("%v", v)) } return params.Encode() } return ""}
// Response 获取响应详情,默认[]byte格式// @Description:// @param request// @return []bytefunc Response(request *http.Request) []byte { res, err := Client.Do(request) if err != nil { return nil } body, _ := ioutil.ReadAll(res.Body) // 读取响应 body, 返回为 []byte defer res.Body.Close() return body}
// clients 初始化请求客户端// @Description:// @return http.Clientfunc clients() http.Client { return http.Client{ Timeout: time.Duration(5) * time.Second, //超时时间 Transport: &http.Transport{ MaxIdleConnsPerHost: 5, //单个路由最大空闲连接数 MaxConnsPerHost: 100, //单个路由最大连接数 IdleConnTimeout: 90 * time.Second, TLSHandshakeTimeout: 10 * time.Second, ExpectContinueTimeout: 1 * time.Second, }, }}
// ParseRes 解析响应// @Description:// @receiver r// @param resfunc (r *Res) ParseRes(res []byte) { json.Unmarshal(res, r)}
// ParseRes 解析响应,将[]byte转成传入对象// @Description:// @param res// @param r//func ParseRes(res []byte, r interface{}) { json.Unmarshal(res, r)}
评论