add JRZQ09J8、FLXGDEA8、FLXGDEA9、JRZQ1D09

add external_services log
This commit is contained in:
2025-08-25 15:44:06 +08:00
parent 365a2a8886
commit 267ff92998
80 changed files with 5555 additions and 1254 deletions

View File

@@ -0,0 +1,67 @@
package yushan
import (
"tyapi-server/internal/config"
"tyapi-server/internal/shared/external_logger"
)
// NewYushanServiceWithConfig 使用配置创建羽山服务
func NewYushanServiceWithConfig(cfg *config.Config) (*YushanService, error) {
// 将配置类型转换为通用外部服务日志配置
loggingConfig := external_logger.ExternalServiceLoggingConfig{
Enabled: cfg.Yushan.Logging.Enabled,
LogDir: cfg.Yushan.Logging.LogDir,
ServiceName: "yushan",
UseDaily: cfg.Yushan.Logging.UseDaily,
EnableLevelSeparation: cfg.Yushan.Logging.EnableLevelSeparation,
LevelConfigs: make(map[string]external_logger.ExternalServiceLevelFileConfig),
}
// 转换级别配置
for key, value := range cfg.Yushan.Logging.LevelConfigs {
loggingConfig.LevelConfigs[key] = external_logger.ExternalServiceLevelFileConfig{
MaxSize: value.MaxSize,
MaxBackups: value.MaxBackups,
MaxAge: value.MaxAge,
Compress: value.Compress,
}
}
// 创建通用外部服务日志器
logger, err := external_logger.NewExternalServiceLogger(loggingConfig)
if err != nil {
return nil, err
}
// 创建羽山服务
service := NewYushanService(
cfg.Yushan.URL,
cfg.Yushan.APIKey,
cfg.Yushan.AcctID,
logger,
)
return service, nil
}
// NewYushanServiceWithLogging 使用自定义日志配置创建羽山服务
func NewYushanServiceWithLogging(url, apiKey, acctID string, loggingConfig external_logger.ExternalServiceLoggingConfig) (*YushanService, error) {
// 设置服务名称
loggingConfig.ServiceName = "yushan"
// 创建通用外部服务日志器
logger, err := external_logger.NewExternalServiceLogger(loggingConfig)
if err != nil {
return nil, err
}
// 创建羽山服务
service := NewYushanService(url, apiKey, acctID, logger)
return service, nil
}
// NewYushanServiceSimple 创建简单的羽山服务(无日志)
func NewYushanServiceSimple(url, apiKey, acctID string) *YushanService {
return NewYushanService(url, apiKey, acctID, nil)
}

View File

@@ -2,8 +2,10 @@ package yushan
import (
"bytes"
"context"
"crypto/aes"
"crypto/cipher"
"crypto/md5"
"crypto/rand"
"encoding/base64"
"encoding/hex"
@@ -15,6 +17,8 @@ import (
"strings"
"time"
"tyapi-server/internal/shared/external_logger"
"github.com/tidwall/gjson"
)
@@ -32,21 +36,37 @@ type YushanConfig struct {
type YushanService struct {
config YushanConfig
logger *external_logger.ExternalServiceLogger
}
// NewWestDexService 是一个构造函数,用于初始化 WestDexService
func NewYushanService(url, apiKey, acctID string) *YushanService {
// NewYushanService 是一个构造函数,用于初始化 YushanService
func NewYushanService(url, apiKey, acctID string, logger *external_logger.ExternalServiceLogger) *YushanService {
return &YushanService{
config: YushanConfig{
URL: url,
ApiKey: apiKey,
AcctID: acctID,
},
logger: logger,
}
}
// CallAPI 调用西部数据的 API
func (y *YushanService) CallAPI(code string, params map[string]interface{}) (respBytes []byte, err error) {
// CallAPI 调用羽山数据的 API
func (y *YushanService) CallAPI(ctx context.Context, code string, params map[string]interface{}) (respBytes []byte, err error) {
startTime := time.Now()
requestID := y.generateRequestID()
// 从ctx中获取transactionId
var transactionID string
if ctxTransactionID, ok := ctx.Value("transaction_id").(string); ok {
transactionID = ctxTransactionID
}
// 记录请求日志
if y.logger != nil {
y.logger.LogRequest(requestID, code, y.config.URL, y.buildLogData(params, transactionID))
}
// 获取当前时间戳
unixMilliseconds := time.Now().UnixNano() / int64(time.Millisecond)
@@ -64,13 +84,21 @@ func (y *YushanService) CallAPI(code string, params map[string]interface{}) (res
// 将请求数据转换为 JSON 字节数组
messageBytes, err := json.Marshal(reqData)
if err != nil {
return nil, fmt.Errorf("%w: %s", ErrSystem, err.Error())
err = fmt.Errorf("%w: %s", ErrSystem, err.Error())
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
// 获取 API 密钥
key, err := hex.DecodeString(y.config.ApiKey)
if err != nil {
return nil, fmt.Errorf("%w: %s", ErrSystem, err.Error())
err = fmt.Errorf("%w: %s", ErrSystem, err.Error())
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
// 使用 AES CBC 加密请求数据
@@ -80,10 +108,16 @@ func (y *YushanService) CallAPI(code string, params map[string]interface{}) (res
content := base64.StdEncoding.EncodeToString(cipherText)
// 发起 HTTP 请求
client := &http.Client{}
req, err := http.NewRequest("POST", y.config.URL, strings.NewReader(content))
client := &http.Client{
Timeout: 20 * time.Second,
}
req, err := http.NewRequestWithContext(ctx, "POST", y.config.URL, strings.NewReader(content))
if err != nil {
return nil, fmt.Errorf("%w: %s", ErrSystem, err.Error())
err = fmt.Errorf("%w: %s", ErrSystem, err.Error())
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("ACCT_ID", y.config.AcctID)
@@ -91,13 +125,20 @@ func (y *YushanService) CallAPI(code string, params map[string]interface{}) (res
// 执行请求
resp, err := client.Do(req)
if err != nil {
return nil, fmt.Errorf("%w: %s", ErrSystem, err.Error())
err = fmt.Errorf("%w: %s", ErrSystem, err.Error())
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
defer resp.Body.Close()
// 读取响应体
body, err := io.ReadAll(resp.Body)
if err != nil {
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
@@ -108,12 +149,22 @@ func (y *YushanService) CallAPI(code string, params map[string]interface{}) (res
} else {
sDec, err := base64.StdEncoding.DecodeString(string(body))
if err != nil {
return nil, fmt.Errorf("%w: %s", ErrSystem, err.Error())
err = fmt.Errorf("%w: %s", ErrSystem, err.Error())
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
respData = y.AES_CBC_Decrypt(sDec, key)
}
retCode := gjson.GetBytes(respData, "retcode").String()
// 记录响应日志
if y.logger != nil {
duration := time.Since(startTime)
y.logger.LogResponse(requestID, code, resp.StatusCode, respData, duration)
}
if retCode == "100000" {
// retcode 为 100000表示查询为空
return nil, ErrNotFound
@@ -121,13 +172,41 @@ func (y *YushanService) CallAPI(code string, params map[string]interface{}) (res
// retcode 为 000000表示有数据返回 retdata
retData := gjson.GetBytes(respData, "retdata")
if !retData.Exists() {
return nil, fmt.Errorf("%w: %s", ErrDatasource, "羽山请求retdata为空")
err = fmt.Errorf("%w: %s", ErrDatasource, "羽山请求retdata为空")
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
return []byte(retData.Raw), nil
} else {
return nil, fmt.Errorf("%w: %s", ErrDatasource, "羽山请求未知的状态码")
err = fmt.Errorf("%w: %s", ErrDatasource, "羽山请求未知的状态码")
if y.logger != nil {
y.logger.LogError(requestID, code, err, y.buildLogData(params, transactionID))
}
return nil, err
}
}
// generateRequestID 生成请求ID
func (y *YushanService) generateRequestID() string {
timestamp := time.Now().UnixNano()
hash := md5.Sum([]byte(fmt.Sprintf("%d_%s", timestamp, y.config.ApiKey)))
return fmt.Sprintf("yushan_%x", hash[:8])
}
// buildLogData 构建包含transactionId的日志数据
func (y *YushanService) buildLogData(data map[string]interface{}, transactionID string) map[string]interface{} {
if transactionID == "" {
return data
}
logData := data
if logData == nil {
logData = make(map[string]interface{})
}
logData["transaction_id"] = transactionID
return logData
}
// GenerateRandomString 生成一个32位的随机字符串订单号

View File

@@ -0,0 +1,83 @@
package yushan
import (
"testing"
"time"
)
func TestGenerateRequestID(t *testing.T) {
service := &YushanService{
config: YushanConfig{
ApiKey: "test_api_key_123",
},
}
id1 := service.generateRequestID()
// 等待一小段时间确保时间戳不同
time.Sleep(time.Millisecond)
id2 := service.generateRequestID()
if id1 == "" || id2 == "" {
t.Error("请求ID生成失败")
}
if id1 == id2 {
t.Error("不同时间生成的请求ID应该不同")
}
// 验证ID格式
if len(id1) < 20 { // yushan_ + 8位十六进制 + 其他
t.Errorf("请求ID长度不足实际: %s", id1)
}
}
func TestGenerateRandomString(t *testing.T) {
service := &YushanService{}
str1, err := service.GenerateRandomString()
if err != nil {
t.Fatalf("生成随机字符串失败: %v", err)
}
str2, err := service.GenerateRandomString()
if err != nil {
t.Fatalf("生成随机字符串失败: %v", err)
}
if str1 == "" || str2 == "" {
t.Error("随机字符串为空")
}
if str1 == str2 {
t.Error("两次生成的随机字符串应该不同")
}
// 验证长度16字节 = 32位十六进制字符
if len(str1) != 32 || len(str2) != 32 {
t.Error("随机字符串长度应该是32位")
}
}
func TestIsJSON(t *testing.T) {
testCases := []struct {
input string
expected bool
}{
{"{}", true},
{"[]", true},
{"{\"key\": \"value\"}", true},
{"[1, 2, 3]", true},
{"invalid json", false},
{"", false},
{"{invalid}", false},
}
for _, tc := range testCases {
result := IsJSON(tc.input)
if result != tc.expected {
t.Errorf("输入: %s, 期望: %v, 实际: %v", tc.input, tc.expected, result)
}
}
}