first commit
This commit is contained in:
395
pkg/httpclient/client.go
Normal file
395
pkg/httpclient/client.go
Normal file
@ -0,0 +1,395 @@
|
||||
package httpclient
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/spf13/cast"
|
||||
"net/http"
|
||||
httpURL "net/url"
|
||||
"time"
|
||||
|
||||
"git.bvbej.com/bvbej/base-golang/pkg/trace"
|
||||
)
|
||||
|
||||
const (
|
||||
// DefaultTTL 一次http请求最长执行1分钟
|
||||
DefaultTTL = time.Minute
|
||||
)
|
||||
|
||||
// Get 请求
|
||||
func Get(url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
return withoutBody(http.MethodGet, url, form, options...)
|
||||
}
|
||||
|
||||
// Delete delete 请求
|
||||
func Delete(url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
return withoutBody(http.MethodDelete, url, form, options...)
|
||||
}
|
||||
|
||||
func withoutBody(method, url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
if url == "" {
|
||||
return nil, errors.New("url required")
|
||||
}
|
||||
|
||||
if len(form) > 0 {
|
||||
if url, err = addFormValuesIntoURL(url, form); err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
ts := time.Now()
|
||||
|
||||
opt := getOption()
|
||||
defer func() {
|
||||
if opt.trace != nil {
|
||||
opt.dialog.Success = err == nil
|
||||
opt.dialog.CostSeconds = time.Since(ts).Seconds()
|
||||
opt.trace.AppendDialog(opt.dialog)
|
||||
}
|
||||
|
||||
releaseOption(opt)
|
||||
}()
|
||||
|
||||
for _, f := range options {
|
||||
f(opt)
|
||||
}
|
||||
opt.header["Content-Type"] = []string{"application/x-www-form-urlencoded; charset=utf-8"}
|
||||
if opt.trace != nil {
|
||||
opt.header[trace.Header] = []string{opt.trace.ID()}
|
||||
}
|
||||
|
||||
ttl := opt.ttl
|
||||
if ttl <= 0 {
|
||||
ttl = DefaultTTL
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), ttl)
|
||||
defer cancel()
|
||||
|
||||
if opt.dialog != nil {
|
||||
decodedURL, _ := httpURL.QueryUnescape(url)
|
||||
opt.dialog.Request = &trace.Request{
|
||||
TTL: ttl.String(),
|
||||
Method: method,
|
||||
DecodedURL: decodedURL,
|
||||
Header: opt.header,
|
||||
}
|
||||
}
|
||||
|
||||
retryTimes := opt.retryTimes
|
||||
if retryTimes <= 0 {
|
||||
retryTimes = DefaultRetryTimes
|
||||
}
|
||||
|
||||
retryDelay := opt.retryDelay
|
||||
if retryDelay <= 0 {
|
||||
retryDelay = DefaultRetryDelay
|
||||
}
|
||||
|
||||
var httpCode int
|
||||
|
||||
defer func() {
|
||||
if opt.alarmObject == nil {
|
||||
return
|
||||
}
|
||||
|
||||
if opt.alarmVerify != nil && !opt.alarmVerify(body) && err == nil {
|
||||
return
|
||||
}
|
||||
|
||||
info := &struct {
|
||||
TraceID string `json:"trace_id"`
|
||||
Request struct {
|
||||
Method string `json:"method"`
|
||||
URL string `json:"url"`
|
||||
} `json:"request"`
|
||||
Response struct {
|
||||
HTTPCode int `json:"http_code"`
|
||||
Body string `json:"body"`
|
||||
} `json:"response"`
|
||||
Error string `json:"error"`
|
||||
}{}
|
||||
|
||||
if opt.trace != nil {
|
||||
info.TraceID = opt.trace.ID()
|
||||
}
|
||||
info.Request.Method = method
|
||||
info.Request.URL = url
|
||||
info.Response.HTTPCode = httpCode
|
||||
info.Response.Body = string(body)
|
||||
info.Error = ""
|
||||
if err != nil {
|
||||
info.Error = fmt.Sprintf("%+v", err)
|
||||
}
|
||||
|
||||
raw, _ := json.MarshalIndent(info, "", " ")
|
||||
onFailedAlarm(opt.alarmTitle, raw, opt.logger, opt.alarmObject)
|
||||
|
||||
}()
|
||||
|
||||
for k := 0; k < retryTimes; k++ {
|
||||
body, httpCode, err = doHTTP(ctx, method, url, nil, opt)
|
||||
if shouldRetry(ctx, httpCode) || (opt.retryVerify != nil && opt.retryVerify(body)) {
|
||||
time.Sleep(retryDelay)
|
||||
continue
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// PostForm post form 请求
|
||||
func PostForm(url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
return withFormBody(http.MethodPost, url, form, options...)
|
||||
}
|
||||
|
||||
// PostJSON post json 请求
|
||||
func PostJSON(url string, raw json.RawMessage, options ...Option) (body []byte, err error) {
|
||||
return withJSONBody(http.MethodPost, url, raw, options...)
|
||||
}
|
||||
|
||||
// PutForm put form 请求
|
||||
func PutForm(url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
return withFormBody(http.MethodPut, url, form, options...)
|
||||
}
|
||||
|
||||
// PutJSON put json 请求
|
||||
func PutJSON(url string, raw json.RawMessage, options ...Option) (body []byte, err error) {
|
||||
return withJSONBody(http.MethodPut, url, raw, options...)
|
||||
}
|
||||
|
||||
// PatchFrom patch form 请求
|
||||
func PatchFrom(url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
return withFormBody(http.MethodPatch, url, form, options...)
|
||||
}
|
||||
|
||||
// PatchJSON patch json 请求
|
||||
func PatchJSON(url string, raw json.RawMessage, options ...Option) (body []byte, err error) {
|
||||
return withJSONBody(http.MethodPatch, url, raw, options...)
|
||||
}
|
||||
|
||||
func withFormBody(method, url string, form httpURL.Values, options ...Option) (body []byte, err error) {
|
||||
if url == "" {
|
||||
return nil, errors.New("url required")
|
||||
}
|
||||
if len(form) == 0 {
|
||||
return nil, errors.New("form required")
|
||||
}
|
||||
|
||||
ts := time.Now()
|
||||
|
||||
opt := getOption()
|
||||
defer func() {
|
||||
if opt.trace != nil {
|
||||
opt.dialog.Success = err == nil
|
||||
opt.dialog.CostSeconds = time.Since(ts).Seconds()
|
||||
opt.trace.AppendDialog(opt.dialog)
|
||||
}
|
||||
|
||||
releaseOption(opt)
|
||||
}()
|
||||
|
||||
for _, f := range options {
|
||||
f(opt)
|
||||
}
|
||||
opt.header["Content-Type"] = []string{"application/x-www-form-urlencoded; charset=utf-8"}
|
||||
if opt.trace != nil {
|
||||
opt.header[trace.Header] = []string{opt.trace.ID()}
|
||||
}
|
||||
|
||||
ttl := opt.ttl
|
||||
if ttl <= 0 {
|
||||
ttl = DefaultTTL
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), ttl)
|
||||
defer cancel()
|
||||
|
||||
formValue := form.Encode()
|
||||
if opt.dialog != nil {
|
||||
decodedURL, _ := httpURL.QueryUnescape(url)
|
||||
opt.dialog.Request = &trace.Request{
|
||||
TTL: ttl.String(),
|
||||
Method: method,
|
||||
DecodedURL: decodedURL,
|
||||
Header: opt.header,
|
||||
Body: formValue,
|
||||
}
|
||||
}
|
||||
|
||||
retryTimes := opt.retryTimes
|
||||
if retryTimes <= 0 {
|
||||
retryTimes = DefaultRetryTimes
|
||||
}
|
||||
|
||||
retryDelay := opt.retryDelay
|
||||
if retryDelay <= 0 {
|
||||
retryDelay = DefaultRetryDelay
|
||||
}
|
||||
|
||||
var httpCode int
|
||||
|
||||
defer func() {
|
||||
if opt.alarmObject == nil {
|
||||
return
|
||||
}
|
||||
|
||||
if opt.alarmVerify != nil && !opt.alarmVerify(body) && err == nil {
|
||||
return
|
||||
}
|
||||
|
||||
info := &struct {
|
||||
TraceID string `json:"trace_id"`
|
||||
Request struct {
|
||||
Method string `json:"method"`
|
||||
URL string `json:"url"`
|
||||
} `json:"request"`
|
||||
Response struct {
|
||||
HTTPCode int `json:"http_code"`
|
||||
Body string `json:"body"`
|
||||
} `json:"response"`
|
||||
Error string `json:"error"`
|
||||
}{}
|
||||
|
||||
if opt.trace != nil {
|
||||
info.TraceID = opt.trace.ID()
|
||||
}
|
||||
info.Request.Method = method
|
||||
info.Request.URL = url
|
||||
info.Response.HTTPCode = httpCode
|
||||
info.Response.Body = string(body)
|
||||
info.Error = ""
|
||||
if err != nil {
|
||||
info.Error = fmt.Sprintf("%+v", err)
|
||||
}
|
||||
|
||||
raw, _ := json.MarshalIndent(info, "", " ")
|
||||
onFailedAlarm(opt.alarmTitle, raw, opt.logger, opt.alarmObject)
|
||||
|
||||
}()
|
||||
|
||||
for k := 0; k < retryTimes; k++ {
|
||||
body, httpCode, err = doHTTP(ctx, method, url, []byte(formValue), opt)
|
||||
if shouldRetry(ctx, httpCode) || (opt.retryVerify != nil && opt.retryVerify(body)) {
|
||||
time.Sleep(retryDelay)
|
||||
continue
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func withJSONBody(method, url string, raw json.RawMessage, options ...Option) (body []byte, err error) {
|
||||
if url == "" {
|
||||
return nil, errors.New("url required")
|
||||
}
|
||||
if len(raw) == 0 {
|
||||
return nil, errors.New("raw required")
|
||||
}
|
||||
|
||||
ts := time.Now()
|
||||
|
||||
opt := getOption()
|
||||
defer func() {
|
||||
if opt.trace != nil {
|
||||
opt.dialog.Success = err == nil
|
||||
opt.dialog.CostSeconds = time.Since(ts).Seconds()
|
||||
opt.trace.AppendDialog(opt.dialog)
|
||||
}
|
||||
|
||||
releaseOption(opt)
|
||||
}()
|
||||
|
||||
for _, f := range options {
|
||||
f(opt)
|
||||
}
|
||||
opt.header["Content-Type"] = []string{"application/json; charset=utf-8"}
|
||||
if opt.trace != nil {
|
||||
opt.header[trace.Header] = []string{opt.trace.ID()}
|
||||
}
|
||||
|
||||
ttl := opt.ttl
|
||||
if ttl <= 0 {
|
||||
ttl = DefaultTTL
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), ttl)
|
||||
defer cancel()
|
||||
|
||||
if opt.dialog != nil {
|
||||
decodedURL, _ := httpURL.QueryUnescape(url)
|
||||
opt.dialog.Request = &trace.Request{
|
||||
TTL: ttl.String(),
|
||||
Method: method,
|
||||
DecodedURL: decodedURL,
|
||||
Header: opt.header,
|
||||
Body: cast.ToString(raw),
|
||||
}
|
||||
}
|
||||
|
||||
retryTimes := opt.retryTimes
|
||||
if retryTimes <= 0 {
|
||||
retryTimes = DefaultRetryTimes
|
||||
}
|
||||
|
||||
retryDelay := opt.retryDelay
|
||||
if retryDelay <= 0 {
|
||||
retryDelay = DefaultRetryDelay
|
||||
}
|
||||
|
||||
var httpCode int
|
||||
|
||||
defer func() {
|
||||
if opt.alarmObject == nil {
|
||||
return
|
||||
}
|
||||
|
||||
if opt.alarmVerify != nil && !opt.alarmVerify(body) && err == nil {
|
||||
return
|
||||
}
|
||||
|
||||
info := &struct {
|
||||
TraceID string `json:"trace_id"`
|
||||
Request struct {
|
||||
Method string `json:"method"`
|
||||
URL string `json:"url"`
|
||||
} `json:"request"`
|
||||
Response struct {
|
||||
HTTPCode int `json:"http_code"`
|
||||
Body string `json:"body"`
|
||||
} `json:"response"`
|
||||
Error string `json:"error"`
|
||||
}{}
|
||||
|
||||
if opt.trace != nil {
|
||||
info.TraceID = opt.trace.ID()
|
||||
}
|
||||
info.Request.Method = method
|
||||
info.Request.URL = url
|
||||
info.Response.HTTPCode = httpCode
|
||||
info.Response.Body = string(body)
|
||||
info.Error = ""
|
||||
if err != nil {
|
||||
info.Error = fmt.Sprintf("%+v", err)
|
||||
}
|
||||
|
||||
raw, _ := json.MarshalIndent(info, "", " ")
|
||||
onFailedAlarm(opt.alarmTitle, raw, opt.logger, opt.alarmObject)
|
||||
|
||||
}()
|
||||
|
||||
for k := 0; k < retryTimes; k++ {
|
||||
body, httpCode, err = doHTTP(ctx, method, url, raw, opt)
|
||||
if shouldRetry(ctx, httpCode) || (opt.retryVerify != nil && opt.retryVerify(body)) {
|
||||
time.Sleep(retryDelay)
|
||||
continue
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user