|
|
package service_gateway
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
"github.com/beego/beego/v2/client/httplib"
|
|
|
)
|
|
|
|
|
|
type MessageCode struct {
|
|
|
Code int `json:"code"`
|
|
|
Msg string `json:"msg"`
|
|
|
}
|
|
|
|
|
|
//GatewayResponse 统一消息返回格式
|
|
|
type GatewayResponse struct {
|
|
|
MessageCode
|
|
|
Data json.RawMessage `json:"data"`
|
|
|
}
|
|
|
|
|
|
type HttpLibBaseServiceGateway struct {
|
|
|
ConnectTimeout time.Duration
|
|
|
ReadWriteTimeout time.Duration
|
|
|
}
|
|
|
|
|
|
func (serviceGateway *HttpLibBaseServiceGateway) CreateRequest(url string, method string) *httplib.BeegoHTTPRequest {
|
|
|
var request *httplib.BeegoHTTPRequest
|
|
|
switch method {
|
|
|
case "get":
|
|
|
request = httplib.Get(url)
|
|
|
|
|
|
case "post":
|
|
|
request = httplib.Post(url)
|
|
|
|
|
|
case "put":
|
|
|
request = httplib.Put(url)
|
|
|
|
|
|
case "delete":
|
|
|
request = httplib.Delete(url)
|
|
|
|
|
|
case "head":
|
|
|
request = httplib.Head(url)
|
|
|
|
|
|
default:
|
|
|
request = httplib.Get(url)
|
|
|
}
|
|
|
return request.SetTimeout(serviceGateway.ConnectTimeout, serviceGateway.ReadWriteTimeout)
|
|
|
}
|
|
|
|
|
|
func (serviceGateway *HttpLibBaseServiceGateway) responseHandle(response map[string]interface{}) (map[string]interface{}, error) {
|
|
|
data := make(map[string]interface{})
|
|
|
var err error
|
|
|
if code, ok := response["code"]; ok {
|
|
|
code := code.(float64)
|
|
|
if code == 0 {
|
|
|
if response["data"] == nil {
|
|
|
data = nil
|
|
|
} else {
|
|
|
data = response["data"].(map[string]interface{})
|
|
|
}
|
|
|
} else {
|
|
|
msg := response["msg"].(string)
|
|
|
err = fmt.Errorf(strings.Join([]string{strconv.FormatFloat(code, 'f', -1, 64), msg}, " "))
|
|
|
}
|
|
|
} else {
|
|
|
jsonBytes, marshalErr := json.Marshal(response)
|
|
|
if marshalErr != nil {
|
|
|
err = marshalErr
|
|
|
}
|
|
|
err = fmt.Errorf("无法解析的网关服务数据返回格式:%s", string(jsonBytes))
|
|
|
}
|
|
|
return data, err
|
|
|
}
|
|
|
|
|
|
func (serviceGateway *HttpLibBaseServiceGateway) GetResponseData(result GatewayResponse, data interface{}) error {
|
|
|
if result.Code != 0 {
|
|
|
return fmt.Errorf(result.Msg)
|
|
|
}
|
|
|
if data == nil {
|
|
|
return nil
|
|
|
}
|
|
|
err := json.Unmarshal(result.Data, data)
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
return nil
|
|
|
} |
...
|
...
|
|