詳解Golang語言HTTP客戶端實踐
最近在學習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.Request
func 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.Request
func 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.Request
func 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 string
func 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 []byte
func 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.Client
func 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 res
func (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 #gosetup
GOPATH=/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_m
get請求
post請求form表單
post請求json表單Process finished with the exit code 0
測試服務
依舊采用瞭moco_FunTester測試框架實現。
package com.mocofun.moco.main
import com.funtester.utils.ArgsUtil
import 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()
}
}
到此這篇關於詳解Golang語言HTTP客戶端實踐的文章就介紹到這瞭,更多相關Golang HTTP客戶端內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！