在github.com/valyala/fasthttp用到了对象池，为了在高性能测试中减少内存的使用，fasthttp使用了两个对象池（我只看了这俩）：requestPool sync.Pool和responsePool sync.Pool，当然fasthttp也提供了正常的对象创建API，后面我在案例中也会写到。
基础API演示
首先分享一下基础的用法封装：
PS：这个属于练习版本，所以没写多少注释。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
package ft
import ( "encoding/json" "fmt" "funtester/task" "github.com/valyala/fasthttp" )
func FastGet(url string, args map[string]interface{}) ([]byte, error) { uri := url + "?" + task.ToValues(args) _, resp, err := fasthttp.Get(nil, uri) if err != nil { fmt.Println("请求失败:", err.Error()) return nil, err } return resp, err }
func FastPostForm(url string, args map[string]interface{}) ([]byte, error) {
// 填充表单，类似于net/url params := &fasthttp.Args{} for s, i2 := range args { sprintf := fmt.Sprintf("%v", i2) params.Add(s, sprintf) } _, resp, err := fasthttp.Post(nil, url, params) if err != nil { fmt.Println("请求失败:", err.Error()) return nil, err } return resp, nil }
func FastPostJson(url string, args map[string]interface{}) ([]byte, error) {
req := &fasthttp.Request{} req.SetRequestURI(url)
marshal, _ := json.Marshal(args) req.SetBody(marshal)
// 默认是application/x-www-form-urlencoded,其实无所谓 req.Header.SetContentType("application/json") req.Header.SetMethod("POST")
resp := &fasthttp.Response{} if err := fasthttp.Do(req, resp); err != nil { fmt.Println("请求失败:", err.Error()) return nil, err } return resp.Body(), nil } |
其中两点主要注意：
FastGet、FastPostForm使用的fasthttp提供的默认获取请求的方式，FastPostJson使用了自定义请求和获取响应的方式
关于请求头中的req.Header.SetContentType方法，其实无所谓，服务端都可以解析
高性能API演示
下面分享使用更高的性能（基于对象池）的API创建请求和获取响应的方式：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
package task
import ( "crypto/tls" "encoding/json" "fmt" "github.com/valyala/fasthttp" "log" "time" )
var FastClient fasthttp.Client = fastClient()
// FastGet 获取GET请求对象,没有进行资源回收 // @Description: // @param url // @param args // @return *fasthttp.Request func FastGet(url string, args map[string]interface{}) *fasthttp.Request { req := fasthttp.AcquireRequest() req.Header.SetMethod("GET") values := ToValues(args) req.SetRequestURI(url + "?" + values) return req }
// FastPostJson POST请求JSON参数,没有进行资源回收 // @Description: // @param url // @param args // @return *fasthttp.Request func FastPostJson(url string, args map[string]interface{}) *fasthttp.Request { req := fasthttp.AcquireRequest() // 默认是application/x-www-form-urlencoded req.Header.SetContentType("application/json") req.Header.SetMethod("POST") req.SetRequestURI(url) marshal, _ := json.Marshal(args) req.SetBody(marshal) return req }
// FastPostForm POST请求表单传参,没有进行资源回收 // @Description: // @param url // @param args // @return *fasthttp.Request func FastPostForm(url string, args map[string]interface{}) *fasthttp.Request { req := fasthttp.AcquireRequest() // 默认是application/x-www-form-urlencoded //req.Header.SetContentType("application/json") req.Header.SetMethod("POST") req.SetRequestURI(url) marshal, _ := json.Marshal(args) req.BodyWriter().Write([]byte(ToValues(args))) req.BodyWriter().Write(marshal) return req }
// FastResponse 获取响应,保证资源回收 // @Description: // @param request // @return []byte // @return error func FastResponse(request *fasthttp.Request) ([]byte, error) { response := fasthttp.AcquireResponse() defer fasthttp.ReleaseResponse(response) defer fasthttp.ReleaseRequest(request) if err := FastClient.Do(request, response); err != nil { log.Println("响应出错了") return nil, err } return response.Body(), nil }
// DoGet 发送GET请求,获取响应 // @Description: // @param url // @param args // @return []byte // @return error func DoGet(url string, args map[string]interface{}) ([]byte, error) { req := fasthttp.AcquireRequest() defer fasthttp.ReleaseRequest(req) // 用完需要释放资源 req.Header.SetMethod("GET") values := ToValues(args) req.SetRequestURI(url + "?" + values) resp := fasthttp.AcquireResponse() defer fasthttp.ReleaseResponse(resp) // 用完需要释放资源 if err := FastClient.Do(req, resp); err != nil { fmt.Println("请求失败:", err.Error()) return nil, err } return resp.Body(), nil }
// fastClient 获取fast客户端 // @Description: // @return fasthttp.Client func fastClient() fasthttp.Client { return fasthttp.Client{ Name: "FunTester", NoDefaultUserAgentHeader: true, TLSConfig: &tls.Config{InsecureSkipVerify: true}, MaxConnsPerHost: 2000, MaxIdleConnDuration: 5 * time.Second, MaxConnDuration: 5 * time.Second, ReadTimeout: 5 * time.Second, WriteTimeout: 5 * time.Second, MaxConnWaitTimeout: 5 * time.Second, } } |
测试服务
用的还是moco_FunTester测试框架，脚本如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
package com.mocofun.moco.main
import com.funtester.utils.ArgsUtil import com.mocofun.moco.MocoServer import org.apache.tools.ant.taskdefs.condition.And
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(both(urlStartsWith("/test"),existArgs("code"))).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() } } |
Golang单元测试
第一次写Golang单测，有点不适应，搞了很久才通。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
package test
import ( "funtester/ft" "funtester/task" "log" "testing" )
const url = "http://localhost:12345/test"
func args() map[string]interface{} { return map[string]interface{}{ "code": 32, "fun": 32, "msg": "324", } }
func TestGet(t *testing.T) { get := task.FastGet(url, args()) res, err := task.FastResponse(get) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "get请求" { t.Fail() } }
func TestPostJson(t *testing.T) { post := task.FastPostJson(url, args()) res, err := task.FastResponse(post) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "post请求json表单" { t.Fail() } }
func TestPostForm(t *testing.T) { post := task.FastPostForm(url, args()) res, err := task.FastResponse(post) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "post请求form表单" { t.Fail() } }
func TestGetNor(t *testing.T) { res, err := ft.FastGet(url, args()) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "get请求" { t.Fail() } }
func TestPostJsonNor(t *testing.T) { res, err := ft.FastPostJson(url, args()) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "post请求json表单" { t.Fail() } }
func TestPostFormNor(t *testing.T) { res, err := ft.FastPostForm(url, args()) if err != nil { t.Fail() } v := string(res) log.Println(v) if v != "post请求form表单" { t.Fail() } } |
测试报告
用的自带的控制台输出内容：
=== RUN TestGet
2021/10/18 18:56:49 get请求
--- PASS: TestGet (0.01s)
=== RUN TestPostJson
2021/10/18 18:56:49 post请求json表单
--- PASS: TestPostJson (0.00s)
=== RUN TestPostForm
2021/10/18 18:56:49 post请求form表单
--- PASS: TestPostForm (0.00s)
=== RUN TestGetNor
2021/10/18 18:56:49 get请求
--- PASS: TestGetNor (0.00s)
=== RUN TestPostJsonNor
2021/10/18 18:56:49 post请求json表单
--- PASS: TestPostJsonNor (0.00s)
=== RUN TestPostFormNor
2021/10/18 18:56:49 post请求form表单
--- PASS: TestPostFormNor (0.00s)
=== RUN TestStageJSON
