140 lines
3.5 KiB
Go
140 lines
3.5 KiB
Go
// Package rpc ...
|
||
//
|
||
// Description : rpc ...
|
||
//
|
||
// Author : go_developer@163.com<白茶清欢>
|
||
//
|
||
// Date : 2023-02-14 14:33
|
||
package rpc
|
||
|
||
import (
|
||
"bytes"
|
||
"compress/gzip"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"strings"
|
||
|
||
"git.zhangdeman.cn/zhangdeman/util"
|
||
|
||
"github.com/ddliu/go-httpclient"
|
||
)
|
||
|
||
var (
|
||
// HttpClient http 客户端
|
||
HttpClient *httpClient
|
||
)
|
||
|
||
func init() {
|
||
HttpClient = &httpClient{}
|
||
}
|
||
|
||
type httpClient struct {
|
||
}
|
||
|
||
// Send 统一发送请求方法
|
||
//
|
||
// Author : go_developer@163.com<白茶清欢>
|
||
//
|
||
// Date : 14:37 2023/2/14
|
||
func (hc *httpClient) Send(requestURL string, method string, header map[string]string, param map[string]interface{}, options httpclient.Map) ([]byte, error) {
|
||
var (
|
||
err error
|
||
response *httpclient.Response
|
||
c *httpclient.HttpClient
|
||
)
|
||
if nil == param {
|
||
param = make(map[string]interface{})
|
||
}
|
||
if nil == header {
|
||
header = make(map[string]string)
|
||
}
|
||
if nil == options {
|
||
options = map[interface{}]interface{}{}
|
||
}
|
||
c = httpclient.NewHttpClient()
|
||
c.Headers = header
|
||
c.WithOptions(options)
|
||
byteData, _ := json.Marshal(param)
|
||
paramMap := map[string]string{}
|
||
for k, v := range param {
|
||
paramMap[k] = fmt.Sprintf("%v", v)
|
||
}
|
||
switch strings.ToUpper(method) {
|
||
case http.MethodGet:
|
||
response, err = c.Get(requestURL, paramMap)
|
||
case http.MethodConnect:
|
||
response, err = c.Connect(requestURL, paramMap)
|
||
case http.MethodHead:
|
||
response, err = c.Head(requestURL)
|
||
case http.MethodOptions:
|
||
response, err = c.Options(requestURL, paramMap)
|
||
case http.MethodPatch:
|
||
response, err = c.Patch(requestURL, paramMap)
|
||
case http.MethodPut:
|
||
response, err = c.Put(requestURL, bytes.NewReader(byteData))
|
||
case http.MethodDelete:
|
||
response, err = c.Delete(requestURL, paramMap)
|
||
case http.MethodPost:
|
||
response, err = c.Post(requestURL, bytes.NewReader(byteData))
|
||
case http.MethodTrace:
|
||
response, err = c.Trace(requestURL, paramMap)
|
||
default:
|
||
return nil, errors.New(method + " is not invalid")
|
||
}
|
||
if nil != err {
|
||
return nil, errors.New(" request send fail : " + err.Error())
|
||
}
|
||
|
||
if response.StatusCode != http.StatusOK {
|
||
return nil, fmt.Errorf("%v : %v", response.StatusCode, response.Status)
|
||
}
|
||
|
||
return hc.getResponseBody(response)
|
||
}
|
||
|
||
// getResponseBody ...
|
||
//
|
||
// Author : go_developer@163.com<白茶清欢>
|
||
//
|
||
// Date : 15:12 2023/2/14
|
||
func (hc *httpClient) getResponseBody(response *httpclient.Response) ([]byte, error) {
|
||
var (
|
||
responseBody []byte
|
||
err error
|
||
gzipInstance *gzip.Reader
|
||
buf bytes.Buffer
|
||
)
|
||
if responseBody, err = io.ReadAll(response.Body); nil != err {
|
||
return nil, err
|
||
}
|
||
// 判断返回值是否经过gzip压缩
|
||
responseGzip := response.Header.Get("content-encoding")
|
||
if strings.ToLower(responseGzip) == "gzip" {
|
||
// 压缩过的数据，在处理response body
|
||
if gzipInstance, err = gzip.NewReader(bytes.NewReader(responseBody)); nil != err {
|
||
return nil, err
|
||
}
|
||
defer func() {
|
||
_ = gzipInstance.Close()
|
||
}()
|
||
if _, err = io.Copy(&buf, gzipInstance); nil != err {
|
||
return nil, err
|
||
}
|
||
responseBody = buf.Bytes()
|
||
}
|
||
// 判断是否为gbk编码,若是,转为utf8
|
||
isGBK := strings.Contains(strings.ToLower(response.Header.Get("content-type")), "gbk")
|
||
if isGBK {
|
||
return []byte(util.String.Convert(string(responseBody), "gbk", "utf-8")), nil
|
||
}
|
||
// 判断是否为gb2312,若是,转为utf8
|
||
isGB2312 := strings.Contains(strings.ToLower(response.Header.Get("content-type")), "gb2312")
|
||
if isGB2312 {
|
||
return []byte(util.String.Convert(string(responseBody), "gb2312", "utf-8")), nil
|
||
}
|
||
return responseBody, nil
|
||
}
|