280 lines
8.5 KiB
Go
280 lines
8.5 KiB
Go
package xingwei
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"crypto/md5"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"strconv"
|
||
"sync/atomic"
|
||
"time"
|
||
|
||
"tyapi-server/internal/shared/external_logger"
|
||
)
|
||
|
||
// 行为数据API状态码常量
|
||
const (
|
||
CodeSuccess = 200 // 操作成功
|
||
CodeSystemError = 500 // 系统内部错误
|
||
CodeMerchantError = 3001 // 商家相关报错（商家不存在、商家被禁用、商家余额不足）
|
||
CodeAccountExpired = 3002 // 账户已过期
|
||
CodeIPWhitelistMissing = 3003 // 未添加ip白名单
|
||
CodeUnauthorized = 3004 // 未授权调用该接口
|
||
CodeProductIDError = 4001 // 产品id错误
|
||
CodeInterfaceDisabled = 4002 // 接口被停用
|
||
CodeQueryException = 5001 // 接口查询异常，请联系技术人员
|
||
CodeNotFound = 6000 // 未查询到结果
|
||
)
|
||
|
||
var (
|
||
ErrDatasource = errors.New("数据源异常")
|
||
ErrSystem = errors.New("系统异常")
|
||
ErrNotFound = errors.New("未查询到结果")
|
||
|
||
// 请求ID计数器，确保唯一性
|
||
requestIDCounter int64
|
||
)
|
||
|
||
// XingweiResponse 行为数据API响应结构
|
||
type XingweiResponse struct {
|
||
Msg string `json:"msg"`
|
||
Code int `json:"code"`
|
||
Data interface{} `json:"data"`
|
||
}
|
||
|
||
// XingweiErrorCode 行为数据错误码定义
|
||
type XingweiErrorCode struct {
|
||
Code int
|
||
Message string
|
||
}
|
||
|
||
// 行为数据错误码映射
|
||
var XingweiErrorCodes = map[int]XingweiErrorCode{
|
||
CodeSuccess: {Code: CodeSuccess, Message: "操作成功"},
|
||
CodeSystemError: {Code: CodeSystemError, Message: "系统内部错误"},
|
||
CodeMerchantError: {Code: CodeMerchantError, Message: "商家相关报错（商家不存在、商家被禁用、商家余额不足）"},
|
||
CodeAccountExpired: {Code: CodeAccountExpired, Message: "账户已过期"},
|
||
CodeIPWhitelistMissing: {Code: CodeIPWhitelistMissing, Message: "未添加ip白名单"},
|
||
CodeUnauthorized: {Code: CodeUnauthorized, Message: "未授权调用该接口"},
|
||
CodeProductIDError: {Code: CodeProductIDError, Message: "产品id错误"},
|
||
CodeInterfaceDisabled: {Code: CodeInterfaceDisabled, Message: "接口被停用"},
|
||
CodeQueryException: {Code: CodeQueryException, Message: "接口查询异常，请联系技术人员"},
|
||
CodeNotFound: {Code: CodeNotFound, Message: "未查询到结果"},
|
||
}
|
||
|
||
// GetXingweiErrorMessage 根据错误码获取错误消息
|
||
func GetXingweiErrorMessage(code int) string {
|
||
if errorCode, exists := XingweiErrorCodes[code]; exists {
|
||
return errorCode.Message
|
||
}
|
||
return fmt.Sprintf("未知错误码: %d", code)
|
||
}
|
||
|
||
type XingweiConfig struct {
|
||
URL string
|
||
ApiID string
|
||
ApiKey string
|
||
}
|
||
|
||
type XingweiService struct {
|
||
config XingweiConfig
|
||
logger *external_logger.ExternalServiceLogger
|
||
}
|
||
|
||
// NewXingweiService 是一个构造函数，用于初始化 XingweiService
|
||
func NewXingweiService(url, apiID, apiKey string, logger *external_logger.ExternalServiceLogger) *XingweiService {
|
||
return &XingweiService{
|
||
config: XingweiConfig{
|
||
URL: url,
|
||
ApiID: apiID,
|
||
ApiKey: apiKey,
|
||
},
|
||
logger: logger,
|
||
}
|
||
}
|
||
|
||
// generateRequestID 生成请求ID
|
||
func (x *XingweiService) generateRequestID() string {
|
||
timestamp := time.Now().UnixNano()
|
||
// 使用原子计数器确保唯一性
|
||
counter := atomic.AddInt64(&requestIDCounter, 1)
|
||
hash := md5.Sum([]byte(fmt.Sprintf("%d_%d_%s", timestamp, counter, x.config.ApiID)))
|
||
return fmt.Sprintf("xingwei_%x", hash[:8])
|
||
}
|
||
|
||
// createSign 创建签名：使用MD5算法将apiId、timestamp、apiKey字符串拼接生成sign
|
||
// 参考Java示例：DigestUtils.md5Hex(apiId + timestamp + apiKey)
|
||
func (x *XingweiService) createSign(timestamp int64) string {
|
||
signStr := x.config.ApiID + strconv.FormatInt(timestamp, 10) + x.config.ApiKey
|
||
hash := md5.Sum([]byte(signStr))
|
||
return fmt.Sprintf("%x", hash)
|
||
}
|
||
|
||
// CallAPI 调用行为数据的 API
|
||
func (x *XingweiService) CallAPI(ctx context.Context, projectID string, params map[string]interface{}) (resp []byte, err error) {
|
||
startTime := time.Now()
|
||
requestID := x.generateRequestID()
|
||
timestamp := time.Now().UnixMilli()
|
||
|
||
// 从ctx中获取transactionId
|
||
var transactionID string
|
||
if ctxTransactionID, ok := ctx.Value("transaction_id").(string); ok {
|
||
transactionID = ctxTransactionID
|
||
}
|
||
|
||
// 记录请求日志
|
||
if x.logger != nil {
|
||
x.logger.LogRequest(requestID, transactionID, "xingwei_api", x.config.URL, params)
|
||
}
|
||
|
||
// 将请求参数转换为JSON
|
||
jsonData, marshalErr := json.Marshal(params)
|
||
if marshalErr != nil {
|
||
err = errors.Join(ErrSystem, marshalErr)
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 创建HTTP POST请求
|
||
req, newRequestErr := http.NewRequestWithContext(ctx, "POST", x.config.URL, bytes.NewBuffer(jsonData))
|
||
if newRequestErr != nil {
|
||
err = errors.Join(ErrSystem, newRequestErr)
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Set("Content-Type", "application/json")
|
||
req.Header.Set("timestamp", strconv.FormatInt(timestamp, 10))
|
||
req.Header.Set("sign", x.createSign(timestamp))
|
||
req.Header.Set("API-ID", x.config.ApiID)
|
||
req.Header.Set("project_id", projectID)
|
||
|
||
// 创建HTTP客户端
|
||
client := &http.Client{
|
||
Timeout: 20 * time.Second,
|
||
}
|
||
|
||
// 发送请求
|
||
httpResp, clientDoErr := client.Do(req)
|
||
if clientDoErr != nil {
|
||
err = errors.Join(ErrSystem, clientDoErr)
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
defer func(Body io.ReadCloser) {
|
||
closeErr := Body.Close()
|
||
if closeErr != nil {
|
||
// 记录关闭错误
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", errors.Join(ErrSystem, fmt.Errorf("关闭响应体失败: %w", closeErr)), params)
|
||
}
|
||
}
|
||
}(httpResp.Body)
|
||
|
||
// 计算请求耗时
|
||
duration := time.Since(startTime)
|
||
|
||
// 读取响应体
|
||
bodyBytes, ReadErr := io.ReadAll(httpResp.Body)
|
||
if ReadErr != nil {
|
||
err = errors.Join(ErrSystem, ReadErr)
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 记录响应日志
|
||
if x.logger != nil {
|
||
x.logger.LogResponse(requestID, transactionID, "xingwei_api", httpResp.StatusCode, bodyBytes, duration)
|
||
}
|
||
|
||
// 检查HTTP状态码
|
||
if httpResp.StatusCode != http.StatusOK {
|
||
err = errors.Join(ErrSystem, fmt.Errorf("行为数据请求失败，状态码: %d", httpResp.StatusCode))
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 解析响应结构
|
||
var xingweiResp XingweiResponse
|
||
if err := json.Unmarshal(bodyBytes, &xingweiResp); err != nil {
|
||
err = errors.Join(ErrSystem, fmt.Errorf("响应解析失败: %w", err))
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 检查业务状态码
|
||
switch xingweiResp.Code {
|
||
case CodeSuccess:
|
||
// 成功响应，返回data字段
|
||
if xingweiResp.Data == nil {
|
||
return []byte("{}"), nil
|
||
}
|
||
|
||
// 将data转换为JSON字节
|
||
dataBytes, err := json.Marshal(xingweiResp.Data)
|
||
if err != nil {
|
||
err = errors.Join(ErrSystem, fmt.Errorf("data字段序列化失败: %w", err))
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api", err, params)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
return dataBytes, nil
|
||
|
||
case CodeNotFound:
|
||
// 未查询到结果，返回查空错误
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api",
|
||
errors.Join(ErrNotFound, fmt.Errorf("未查询到结果")), params)
|
||
}
|
||
return nil, errors.Join(ErrNotFound, fmt.Errorf("未查询到结果"))
|
||
|
||
case CodeSystemError:
|
||
// 系统内部错误
|
||
errorMsg := GetXingweiErrorMessage(xingweiResp.Code)
|
||
systemErr := fmt.Errorf("行为数据系统错误[%d]: %s", xingweiResp.Code, errorMsg)
|
||
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api",
|
||
errors.Join(ErrSystem, systemErr), params)
|
||
}
|
||
|
||
return nil, errors.Join(ErrSystem, systemErr)
|
||
|
||
default:
|
||
// 其他业务错误
|
||
errorMsg := GetXingweiErrorMessage(xingweiResp.Code)
|
||
businessErr := fmt.Errorf("行为数据业务错误[%d]: %s", xingweiResp.Code, errorMsg)
|
||
|
||
if x.logger != nil {
|
||
x.logger.LogError(requestID, transactionID, "xingwei_api",
|
||
errors.Join(ErrDatasource, businessErr), params)
|
||
}
|
||
|
||
return nil, errors.Join(ErrDatasource, businessErr)
|
||
}
|
||
}
|
||
|
||
// GetConfig 获取配置信息
|
||
func (x *XingweiService) GetConfig() XingweiConfig {
|
||
return x.config
|
||
}
|