写点什么

Golang 语言 HTTP 客户端实践

用户头像
FunTester
关注
发布于: 刚刚

最近在学习 Golang 语言,中间遇到一个前辈指点,有一个学习原则:Learning By Doing。跟我之前学习 Java 的经验高度契合。在前一段时间学习洼坑中挣扎了好几天,差点就忘记这个重要的成功经验。


那么那什么来做练习呢?当然结合当下的工作啦,所以我列了一个路线给自己,那就是从接口测试开始学起来,从功能测试到性能测试,然后掌握基本 Server 开发技能。


首先,得先把 HTTP 接口测试常用的几个功能实现了,主要是获取 HTTPrequest 对象,发送请求解析响应以及 HttpClient 的基本配置。


这里实现比较简单和粗浅,让我想起 FunTester 测试框架一开始的时候,也是从封装 HttpClient.jar 提供的 API 开始的,感慨万千。


这里我从了 Golang SDK 自带的net/http包提供的 HTTP 相关 API,虽然本身提供了http.PostForm()http.Post()以及http.Get()封装好的方法,但是在处理 HTTPrequest 的 header 和 cookie 处理缺乏灵活性,所以我重新将net/http封装的 API 进行第二次封装。其中存在几处遗漏,比如 HTTPstatus 的判断以及 header 中Content-Type自动处理,这个以后再丰富,暂时的目标就是能用。

HTTP 客户端封装

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)}
复制代码

测试脚本

package main
import ( "fmt" "funtester/src/task" "io" "log" "net/http" "os" "time")
const ( a = iota b c d e)
func init() { os.Mkdir("./log/", 0777) os.Mkdir("./long/", 0777) file := "./log/" + string(time.Now().Format("20060102")) + ".log" openFile, _ := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666) writer := io.MultiWriter(os.Stdout, openFile) log.SetOutput(writer) log.SetFlags(log.LstdFlags | log.Lshortfile | log.Ldate)}
func main() {
url := "http://localhost:12345/test" args := map[string]interface{}{ "name": "FunTester", "fun": "fdsafj", } cookie := &http.Cookie{ Name: "token", Value: "fsjej09u0934jtej", } get := task.Get(url, args) get.Header.Add("user_agent", "FunTester") get.AddCookie(cookie) response := task.Response(get) fmt.Println(string(response)) form := task.PostForm(url, args) bytes := task.Response(form) fmt.Println(string(bytes)) json := task.PostJson(url, args) res := task.Response(json) fmt.Println(string(res))
}
复制代码

控制台输出

GOROOT=/usr/local/go #gosetupGOPATH=/Users/oker/go #gosetup/usr/local/go/bin/go build -o /private/var/folders/7b/0djfgf7j7p9ch_hgm9wx9n6w0000gn/T/GoLand/___go_build_funtester_src_m funtester/src/m #gosetup/private/var/folders/7b/0djfgf7j7p9ch_hgm9wx9n6w0000gn/T/GoLand/___go_build_funtester_src_mget请求post请求form表单post请求json表单
Process finished with the exit code 0
复制代码

测试服务

依旧采用了moco_FunTester测试框架实现。


package com.mocofun.moco.main
import com.funtester.utils.ArgsUtilimport com.mocofun.moco.MocoServer
class Share extends MocoServer {
static void main(String[] args) { def util = new ArgsUtil(args) // def server = getServerNoLog(util.getIntOrdefault(0,12345)) def server = getServer(util.getIntOrdefault(0, 12345)) server.get(urlStartsWith("/test")).response("get请求") server.post(both(urlStartsWith("/test"), existForm("fun"))).response("post请求form表单") server.post(both(urlStartsWith("/test"), existParams("fun"))).response("post请求json表单") server.get(urlStartsWith("/qps")).response(qps(textRes("恭喜到达QPS!"), 1))// server.response(delay(jsonRes(getJson("Have=Fun ~ Tester !")), 1000)) server.response("Have Fun ~ Tester !") def run = run(server) waitForKey("fan") run.stop() }}
复制代码

Have Fun ~ Tester !

发布于: 刚刚阅读数: 2
用户头像

FunTester

关注

公众号:FunTester,650+原创,欢迎关注 2020.10.20 加入

Have Fun,Tester! 公众号FunTester,坚持原创文章的测试人。 FunTester测试框架作者,DCS_FunTester分布式性能测试框架作者。

评论

发布
暂无评论
Golang语言HTTP客户端实践