添加文件: xy_sh/client.go
This commit is contained in:
parent
594860eb2b
commit
a3b253000f
|
|
@ -0,0 +1,244 @@
|
|||
package xy_sh
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// Client 供应商接口客户端
|
||||
type Client struct {
|
||||
httpClient *http.Client
|
||||
sm4Key []byte // SM4加密密钥，16字节
|
||||
sm3Salt []byte // SM3签名盐值
|
||||
orderURL string // 下单接口地址
|
||||
queryURL string // 查询接口地址
|
||||
rechargeURL string // 微信立减金充值接口地址
|
||||
}
|
||||
|
||||
// NewClient 创建新的客户端
|
||||
// - sm4Key: SM4加密密钥，必须为16字节
|
||||
// - sm3Salt: SM3签名盐值
|
||||
// - orderURL: 下单接口地址
|
||||
// - queryURL: 查询接口地址
|
||||
// - rechargeURL: 微信立减金充值接口地址
|
||||
func NewClient(sm4Key []byte, sm3Salt []byte, orderURL, queryURL, rechargeURL string) (*Client, error) {
|
||||
if len(sm4Key) != 16 {
|
||||
return nil, fmt.Errorf("SM4密钥长度必须为16字节")
|
||||
}
|
||||
if len(sm3Salt) == 0 {
|
||||
return nil, fmt.Errorf("SM3盐值不能为空")
|
||||
}
|
||||
return &Client{
|
||||
httpClient: &http.Client{},
|
||||
sm4Key: sm4Key,
|
||||
sm3Salt: sm3Salt,
|
||||
orderURL: orderURL,
|
||||
queryURL: queryURL,
|
||||
rechargeURL: rechargeURL,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// SetHTTPClient 设置自定义HTTP客户端
|
||||
func (c *Client) SetHTTPClient(httpClient *http.Client) {
|
||||
c.httpClient = httpClient
|
||||
}
|
||||
|
||||
// doRequest 执行加密请求并解密响应
|
||||
func (c *Client) doRequest(ctx context.Context, url string, bizReq interface{}) (*RawResponse, error) {
|
||||
// 1. 序列化业务参数为JSON
|
||||
bizBody, err := json.Marshal(bizReq)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("序列化业务参数失败: %v", err)
|
||||
}
|
||||
|
||||
// 2. SM4加密业务参数
|
||||
encryptedData, err := SM4Encrypt(bizBody, c.sm4Key)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("SM4加密失败: %v", err)
|
||||
}
|
||||
|
||||
// 3. 生成时间戳和签名
|
||||
timestamp := GenerateTimestampMillis()
|
||||
sign := GenerateSign(timestamp, encryptedData, c.sm3Salt)
|
||||
|
||||
// 4. 构建加密请求体
|
||||
reqBody := EncryptedRequest{EncryptedData: encryptedData}
|
||||
reqBytes, err := json.Marshal(reqBody)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("序列化请求体失败: %v", err)
|
||||
}
|
||||
|
||||
// 5. 创建HTTP请求
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(reqBytes))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建HTTP请求失败: %v", err)
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
req.Header.Set("timestamp", timestamp)
|
||||
req.Header.Set("sign", sign)
|
||||
|
||||
// 6. 发送请求
|
||||
resp, err := c.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("发送HTTP请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// 7. 读取响应体
|
||||
respBytes, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取响应体失败: %v", err)
|
||||
}
|
||||
|
||||
// 8. 解析原始响应
|
||||
var rawResp RawResponse
|
||||
if err := json.Unmarshal(respBytes, &rawResp); err != nil {
|
||||
return nil, fmt.Errorf("解析响应JSON失败: %v", err)
|
||||
}
|
||||
|
||||
return &rawResp, nil
|
||||
}
|
||||
|
||||
// decryptData 解密响应中的data字段
|
||||
func (c *Client) decryptData(encryptedData string, target interface{}) error {
|
||||
if encryptedData == "" {
|
||||
return nil
|
||||
}
|
||||
plaintext, err := SM4Decrypt(encryptedData, c.sm4Key)
|
||||
if err != nil {
|
||||
return fmt.Errorf("SM4解密失败: %v", err)
|
||||
}
|
||||
if err := json.Unmarshal(plaintext, target); err != nil {
|
||||
return fmt.Errorf("解析解密后数据失败: %v", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Order 卡券/直充权益下单
|
||||
// 适用于卡密、直充商品下单
|
||||
func (c *Client) Order(ctx context.Context, req *OrderRequest) (*OrderResponse, error) {
|
||||
rawResp, err := c.doRequest(ctx, c.orderURL, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
resp := &OrderResponse{
|
||||
Code: rawResp.Code,
|
||||
Msg: rawResp.Msg,
|
||||
}
|
||||
|
||||
if rawResp.Data != "" {
|
||||
var data OrderData
|
||||
if err := c.decryptData(rawResp.Data, &data); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
resp.Data = &data
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// QueryOrder 卡券/直充/微信立减金订单查询
|
||||
// 通过供应商订单号查询订单状态
|
||||
func (c *Client) QueryOrder(ctx context.Context, req *QueryRequest) (*QueryResponse, error) {
|
||||
rawResp, err := c.doRequest(ctx, c.queryURL, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
resp := &QueryResponse{
|
||||
Code: rawResp.Code,
|
||||
Msg: rawResp.Msg,
|
||||
}
|
||||
|
||||
if rawResp.Data != "" {
|
||||
var data QueryData
|
||||
if err := c.decryptData(rawResp.Data, &data); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
resp.Data = &data
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// WechatRecharge 微信立减金订单充值
|
||||
func (c *Client) WechatRecharge(ctx context.Context, req *WechatRechargeRequest) (*WechatRechargeResponse, error) {
|
||||
rawResp, err := c.doRequest(ctx, c.rechargeURL, req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
resp := &WechatRechargeResponse{
|
||||
Code: rawResp.Code,
|
||||
Msg: rawResp.Msg,
|
||||
}
|
||||
|
||||
if rawResp.Data != "" {
|
||||
var data WechatRechargeData
|
||||
if err := c.decryptData(rawResp.Data, &data); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
resp.Data = &data
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// SendCallback 发送回调通知（供应商调用行方回调接口）
|
||||
// 返回的 CallbackResponse 中 StatusCode 为 HTTP 状态码，Body 为响应体内容
|
||||
func (c *Client) SendCallback(ctx context.Context, callbackURL string, req *CallbackRequest) (*CallbackResponse, error) {
|
||||
// 1. 序列化业务参数
|
||||
bizBody, err := json.Marshal(req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("序列化回调参数失败: %v", err)
|
||||
}
|
||||
|
||||
// 2. SM4加密
|
||||
encryptedData, err := SM4Encrypt(bizBody, c.sm4Key)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("SM4加密失败: %v", err)
|
||||
}
|
||||
|
||||
// 3. 生成时间戳和签名
|
||||
timestamp := GenerateTimestampMillis()
|
||||
sign := GenerateSign(timestamp, encryptedData, c.sm3Salt)
|
||||
|
||||
// 4. 构建请求体
|
||||
reqBody := EncryptedRequest{EncryptedData: encryptedData}
|
||||
reqBytes, err := json.Marshal(reqBody)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("序列化请求体失败: %v", err)
|
||||
}
|
||||
|
||||
// 5. 创建HTTP请求
|
||||
httpReq, err := http.NewRequestWithContext(ctx, http.MethodPost, callbackURL, bytes.NewReader(reqBytes))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建HTTP请求失败: %v", err)
|
||||
}
|
||||
httpReq.Header.Set("Content-Type", "application/json")
|
||||
httpReq.Header.Set("timestamp", timestamp)
|
||||
httpReq.Header.Set("sign", sign)
|
||||
|
||||
// 6. 发送请求
|
||||
resp, err := c.httpClient.Do(httpReq)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("发送回调请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// 7. 读取响应体
|
||||
respBytes, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取回调响应失败: %v", err)
|
||||
}
|
||||
|
||||
return &CallbackResponse{
|
||||
StatusCode: resp.StatusCode,
|
||||
Body: string(respBytes),
|
||||
}, nil
|
||||
}
|
||||
Loading…
Reference in New Issue