fix
This commit is contained in:
416
app/main/api/internal/service/tianyuanapi_sdk/client.go
Normal file
416
app/main/api/internal/service/tianyuanapi_sdk/client.go
Normal file
@@ -0,0 +1,416 @@
|
||||
package tianyuanapi
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/aes"
|
||||
"crypto/cipher"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
// API调用相关错误类型
|
||||
var (
|
||||
ErrQueryEmpty = errors.New("查询为空")
|
||||
ErrSystem = errors.New("接口异常")
|
||||
ErrDecryptFail = errors.New("解密失败")
|
||||
ErrRequestParam = errors.New("请求参数结构不正确")
|
||||
ErrInvalidParam = errors.New("参数校验不正确")
|
||||
ErrInvalidIP = errors.New("未经授权的IP")
|
||||
ErrMissingAccessId = errors.New("缺少Access-Id")
|
||||
ErrInvalidAccessId = errors.New("未经授权的AccessId")
|
||||
ErrFrozenAccount = errors.New("账户已冻结")
|
||||
ErrArrears = errors.New("账户余额不足，无法请求")
|
||||
ErrProductNotFound = errors.New("产品不存在")
|
||||
ErrProductDisabled = errors.New("产品已停用")
|
||||
ErrNotSubscribed = errors.New("未订阅此产品")
|
||||
ErrBusiness = errors.New("业务失败")
|
||||
)
|
||||
|
||||
// 错误码映射 - 严格按照用户要求
|
||||
var ErrorCodeMap = map[error]int{
|
||||
ErrQueryEmpty: 1000,
|
||||
ErrSystem: 1001,
|
||||
ErrDecryptFail: 1002,
|
||||
ErrRequestParam: 1003,
|
||||
ErrInvalidParam: 1003,
|
||||
ErrInvalidIP: 1004,
|
||||
ErrMissingAccessId: 1005,
|
||||
ErrInvalidAccessId: 1006,
|
||||
ErrFrozenAccount: 1007,
|
||||
ErrArrears: 1007,
|
||||
ErrProductNotFound: 1008,
|
||||
ErrProductDisabled: 1008,
|
||||
ErrNotSubscribed: 1008,
|
||||
ErrBusiness: 2001,
|
||||
}
|
||||
|
||||
// ApiCallOptions API调用选项
|
||||
type ApiCallOptions struct {
|
||||
Json bool `json:"json,omitempty"` // 是否返回JSON格式
|
||||
}
|
||||
|
||||
// Client 天元API客户端
|
||||
type Client struct {
|
||||
accessID string
|
||||
key string
|
||||
baseURL string
|
||||
timeout time.Duration
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// Config 客户端配置
|
||||
type Config struct {
|
||||
AccessID string // 访问ID
|
||||
Key string // AES密钥（16进制）
|
||||
BaseURL string // API基础URL
|
||||
Timeout time.Duration // 超时时间
|
||||
}
|
||||
|
||||
// Request 请求参数
|
||||
type Request struct {
|
||||
InterfaceName string `json:"interfaceName"` // 接口名称
|
||||
Params map[string]interface{} `json:"params"` // 请求参数
|
||||
Timeout int `json:"timeout"` // 超时时间（毫秒）
|
||||
Options *ApiCallOptions `json:"options"` // 调用选项
|
||||
}
|
||||
|
||||
// ApiResponse HTTP API响应
|
||||
type ApiResponse struct {
|
||||
Code int `json:"code"`
|
||||
Message string `json:"message"`
|
||||
TransactionID string `json:"transaction_id"` // 流水号
|
||||
Data string `json:"data"` // 加密的数据
|
||||
}
|
||||
|
||||
// Response Call方法的响应
|
||||
type Response struct {
|
||||
Code int `json:"code"`
|
||||
Message string `json:"message"`
|
||||
Success bool `json:"success"`
|
||||
TransactionID string `json:"transaction_id"` // 流水号
|
||||
Data map[string]interface{} `json:"data"` // 解密后的数据
|
||||
Timeout int64 `json:"timeout"` // 请求耗时（毫秒）
|
||||
Error string `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
// NewClient 创建新的客户端实例
|
||||
func NewClient(config Config) (*Client, error) {
|
||||
// 参数校验
|
||||
if config.AccessID == "" {
|
||||
return nil, fmt.Errorf("accessID不能为空")
|
||||
}
|
||||
if config.Key == "" {
|
||||
return nil, fmt.Errorf("key不能为空")
|
||||
}
|
||||
if config.BaseURL == "" {
|
||||
config.BaseURL = "http://127.0.0.1:8080"
|
||||
}
|
||||
if config.Timeout == 0 {
|
||||
config.Timeout = 60 * time.Second
|
||||
}
|
||||
|
||||
// 验证密钥格式
|
||||
if _, err := hex.DecodeString(config.Key); err != nil {
|
||||
return nil, fmt.Errorf("无效的密钥格式，必须是16进制字符串: %v", err)
|
||||
}
|
||||
|
||||
return &Client{
|
||||
accessID: config.AccessID,
|
||||
key: config.Key,
|
||||
baseURL: config.BaseURL,
|
||||
timeout: config.Timeout,
|
||||
client: &http.Client{
|
||||
Timeout: config.Timeout,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// Call 调用API接口
|
||||
func (c *Client) Call(req Request) (*Response, error) {
|
||||
startTime := time.Now()
|
||||
|
||||
// 参数校验
|
||||
if err := c.validateRequest(req); err != nil {
|
||||
return nil, fmt.Errorf("请求参数校验失败: %v", err)
|
||||
}
|
||||
|
||||
// 加密参数
|
||||
jsonData, err := json.Marshal(req.Params)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("参数序列化失败: %v", err)
|
||||
}
|
||||
|
||||
encryptedData, err := c.encrypt(string(jsonData))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("数据加密失败: %v", err)
|
||||
}
|
||||
|
||||
// 构建请求体
|
||||
requestBody := map[string]interface{}{
|
||||
"data": encryptedData,
|
||||
}
|
||||
|
||||
// 添加选项
|
||||
if req.Options != nil {
|
||||
requestBody["options"] = req.Options
|
||||
} else {
|
||||
// 默认选项
|
||||
defaultOptions := &ApiCallOptions{
|
||||
Json: true,
|
||||
}
|
||||
requestBody["options"] = defaultOptions
|
||||
}
|
||||
|
||||
requestBodyBytes, err := json.Marshal(requestBody)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("请求体序列化失败: %v", err)
|
||||
}
|
||||
|
||||
// 创建HTTP请求
|
||||
url := fmt.Sprintf("%s/api/v1/%s", c.baseURL, req.InterfaceName)
|
||||
|
||||
httpReq, err := http.NewRequest("POST", url, bytes.NewBuffer(requestBodyBytes))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建HTTP请求失败: %v", err)
|
||||
}
|
||||
|
||||
// 设置请求头
|
||||
httpReq.Header.Set("Content-Type", "application/json")
|
||||
httpReq.Header.Set("Access-Id", c.accessID)
|
||||
httpReq.Header.Set("User-Agent", "TianyuanAPI-Go-SDK/1.0.0")
|
||||
|
||||
// 发送请求
|
||||
resp, err := c.client.Do(httpReq)
|
||||
if err != nil {
|
||||
endTime := time.Now()
|
||||
requestTime := endTime.Sub(startTime).Milliseconds()
|
||||
return &Response{
|
||||
Success: false,
|
||||
Message: "请求失败",
|
||||
Error: err.Error(),
|
||||
Timeout: requestTime,
|
||||
}, nil
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// 读取响应
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
endTime := time.Now()
|
||||
requestTime := endTime.Sub(startTime).Milliseconds()
|
||||
return &Response{
|
||||
Success: false,
|
||||
Message: "读取响应失败",
|
||||
Error: err.Error(),
|
||||
Timeout: requestTime,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// 解析HTTP API响应
|
||||
var apiResp ApiResponse
|
||||
if err := json.Unmarshal(body, &apiResp); err != nil {
|
||||
endTime := time.Now()
|
||||
requestTime := endTime.Sub(startTime).Milliseconds()
|
||||
return &Response{
|
||||
Success: false,
|
||||
Message: "响应解析失败",
|
||||
Error: err.Error(),
|
||||
Timeout: requestTime,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// 计算请求耗时
|
||||
endTime := time.Now()
|
||||
requestTime := endTime.Sub(startTime).Milliseconds()
|
||||
|
||||
// 构建Call方法的响应
|
||||
response := &Response{
|
||||
Code: apiResp.Code,
|
||||
Message: apiResp.Message,
|
||||
Success: apiResp.Code == 0,
|
||||
TransactionID: apiResp.TransactionID,
|
||||
Timeout: requestTime,
|
||||
}
|
||||
|
||||
// 如果有加密数据，尝试解密
|
||||
if apiResp.Data != "" {
|
||||
decryptedData, err := c.decrypt(apiResp.Data)
|
||||
if err == nil {
|
||||
var decryptedMap map[string]interface{}
|
||||
if json.Unmarshal([]byte(decryptedData), &decryptedMap) == nil {
|
||||
response.Data = decryptedMap
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 根据响应码返回对应的错误
|
||||
if apiResp.Code != 0 {
|
||||
err := GetErrorByCode(apiResp.Code)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
|
||||
// CallInterface 简化接口调用方法
|
||||
func (c *Client) CallInterface(interfaceName string, params map[string]interface{}, options ...*ApiCallOptions) (*Response, error) {
|
||||
var opts *ApiCallOptions
|
||||
if len(options) > 0 {
|
||||
opts = options[0]
|
||||
}
|
||||
|
||||
req := Request{
|
||||
InterfaceName: interfaceName,
|
||||
Params: params,
|
||||
Timeout: 60000,
|
||||
Options: opts,
|
||||
}
|
||||
|
||||
return c.Call(req)
|
||||
}
|
||||
|
||||
// validateRequest 校验请求参数
|
||||
func (c *Client) validateRequest(req Request) error {
|
||||
if req.InterfaceName == "" {
|
||||
return fmt.Errorf("interfaceName不能为空")
|
||||
}
|
||||
if req.Params == nil {
|
||||
return fmt.Errorf("params不能为空")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// encrypt AES CBC加密
|
||||
func (c *Client) encrypt(plainText string) (string, error) {
|
||||
keyBytes, err := hex.DecodeString(c.key)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
block, err := aes.NewCipher(keyBytes)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 生成随机IV
|
||||
iv := make([]byte, aes.BlockSize)
|
||||
if _, err := io.ReadFull(rand.Reader, iv); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 填充数据
|
||||
paddedData := c.pkcs7Pad([]byte(plainText), aes.BlockSize)
|
||||
|
||||
// 加密
|
||||
ciphertext := make([]byte, len(iv)+len(paddedData))
|
||||
copy(ciphertext, iv)
|
||||
|
||||
mode := cipher.NewCBCEncrypter(block, iv)
|
||||
mode.CryptBlocks(ciphertext[len(iv):], paddedData)
|
||||
|
||||
return base64.StdEncoding.EncodeToString(ciphertext), nil
|
||||
}
|
||||
|
||||
// decrypt AES CBC解密
|
||||
func (c *Client) decrypt(encryptedText string) (string, error) {
|
||||
keyBytes, err := hex.DecodeString(c.key)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
ciphertext, err := base64.StdEncoding.DecodeString(encryptedText)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
block, err := aes.NewCipher(keyBytes)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
if len(ciphertext) < aes.BlockSize {
|
||||
return "", fmt.Errorf("密文太短")
|
||||
}
|
||||
|
||||
iv := ciphertext[:aes.BlockSize]
|
||||
ciphertext = ciphertext[aes.BlockSize:]
|
||||
|
||||
if len(ciphertext)%aes.BlockSize != 0 {
|
||||
return "", fmt.Errorf("密文长度不是块大小的倍数")
|
||||
}
|
||||
|
||||
mode := cipher.NewCBCDecrypter(block, iv)
|
||||
mode.CryptBlocks(ciphertext, ciphertext)
|
||||
|
||||
// 去除填充
|
||||
unpaddedData, err := c.pkcs7Unpad(ciphertext)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return string(unpaddedData), nil
|
||||
}
|
||||
|
||||
// pkcs7Pad PKCS7填充
|
||||
func (c *Client) pkcs7Pad(data []byte, blockSize int) []byte {
|
||||
padding := blockSize - len(data)%blockSize
|
||||
padtext := bytes.Repeat([]byte{byte(padding)}, padding)
|
||||
return append(data, padtext...)
|
||||
}
|
||||
|
||||
// pkcs7Unpad PKCS7去除填充
|
||||
func (c *Client) pkcs7Unpad(data []byte) ([]byte, error) {
|
||||
length := len(data)
|
||||
if length == 0 {
|
||||
return nil, fmt.Errorf("数据为空")
|
||||
}
|
||||
unpadding := int(data[length-1])
|
||||
if unpadding > length {
|
||||
return nil, fmt.Errorf("无效的填充")
|
||||
}
|
||||
return data[:length-unpadding], nil
|
||||
}
|
||||
|
||||
// GetErrorByCode 根据错误码获取错误
|
||||
func GetErrorByCode(code int) error {
|
||||
// 对于有多个错误对应同一错误码的情况，返回第一个
|
||||
switch code {
|
||||
case 1000:
|
||||
return ErrQueryEmpty
|
||||
case 1001:
|
||||
return ErrSystem
|
||||
case 1002:
|
||||
return ErrDecryptFail
|
||||
case 1003:
|
||||
return ErrRequestParam
|
||||
case 1004:
|
||||
return ErrInvalidIP
|
||||
case 1005:
|
||||
return ErrMissingAccessId
|
||||
case 1006:
|
||||
return ErrInvalidAccessId
|
||||
case 1007:
|
||||
return ErrFrozenAccount
|
||||
case 1008:
|
||||
return ErrProductNotFound
|
||||
case 2001:
|
||||
return ErrBusiness
|
||||
default:
|
||||
return fmt.Errorf("未知错误码: %d", code)
|
||||
}
|
||||
}
|
||||
|
||||
// GetCodeByError 根据错误获取错误码
|
||||
func GetCodeByError(err error) int {
|
||||
if code, exists := ErrorCodeMap[err]; exists {
|
||||
return code
|
||||
}
|
||||
return -1
|
||||
}
|
||||
Reference in New Issue
Block a user