95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
// Package verify 校验
|
||
package verify
|
||
|
||
import (
|
||
"bytes"
|
||
"compress/gzip"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"io/ioutil"
|
||
"net/http"
|
||
|
||
"go_dreamfactory/stress/model"
|
||
)
|
||
|
||
// getZipData 处理gzip压缩
|
||
func getZipData(response *http.Response) (body []byte, err error) {
|
||
var reader io.ReadCloser
|
||
switch response.Header.Get("Content-Encoding") {
|
||
case "gzip":
|
||
reader, err = gzip.NewReader(response.Body)
|
||
defer func() {
|
||
_ = reader.Close()
|
||
}()
|
||
default:
|
||
reader = response.Body
|
||
}
|
||
body, err = ioutil.ReadAll(reader)
|
||
response.Body = ioutil.NopCloser(bytes.NewReader(body))
|
||
return
|
||
}
|
||
|
||
// HTTPStatusCode 通过 HTTP 状态码判断是否请求成功
|
||
func HTTPStatusCode(request *model.Request, response *http.Response) (code int, isSucceed bool) {
|
||
defer func() {
|
||
_ = response.Body.Close()
|
||
}()
|
||
code = response.StatusCode
|
||
if code == request.Code {
|
||
isSucceed = true
|
||
}
|
||
// 开启调试模式
|
||
if request.GetDebug() {
|
||
body, err := getZipData(response)
|
||
fmt.Printf("请求结果 httpCode:%d body:%s err:%v \n", response.StatusCode, string(body), err)
|
||
}
|
||
io.Copy(ioutil.Discard, response.Body)
|
||
return
|
||
}
|
||
|
||
/*************************** 返回值为json ********************************/
|
||
|
||
// ResponseJSON 返回数据结构体
|
||
type ResponseJSON struct {
|
||
Code int `json:"code"`
|
||
Msg string `json:"msg"`
|
||
Data interface{} `json:"data"`
|
||
}
|
||
|
||
// HTTPJson 通过返回的Body 判断
|
||
// 返回示例: {"code":200,"msg":"Success","data":{}}
|
||
// code 默认将http code作为返回码，http code 为200时 取body中的返回code
|
||
func HTTPJson(request *model.Request, response *http.Response) (code int, isSucceed bool) {
|
||
defer func() {
|
||
_ = response.Body.Close()
|
||
}()
|
||
code = response.StatusCode
|
||
if code == http.StatusOK {
|
||
body, err := getZipData(response)
|
||
if err != nil {
|
||
code = model.ParseError
|
||
fmt.Printf("请求结果 ioutil.ReadAll err:%v", err)
|
||
} else {
|
||
responseJSON := &ResponseJSON{}
|
||
err = json.Unmarshal(body, responseJSON)
|
||
if err != nil {
|
||
code = model.ParseError
|
||
fmt.Printf("请求结果 json.Unmarshal err:%v", err)
|
||
} else {
|
||
code = responseJSON.Code
|
||
// body 中code返回200为返回数据成功
|
||
if responseJSON.Code == request.Code {
|
||
isSucceed = true
|
||
}
|
||
}
|
||
}
|
||
// 开启调试模式
|
||
if request.GetDebug() {
|
||
fmt.Printf("请求结果 httpCode:%d body:%s err:%v \n", response.StatusCode, string(body), err)
|
||
}
|
||
}
|
||
io.Copy(ioutil.Discard, response.Body)
|
||
return
|
||
}
|