217 lines
7.0 KiB
Go
217 lines
7.0 KiB
Go
package service
|
||
|
||
import (
|
||
"bdqr-server/app/main/model"
|
||
"context"
|
||
"database/sql"
|
||
"encoding/json"
|
||
"fmt"
|
||
"time"
|
||
|
||
"github.com/Masterminds/squirrel"
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
// TianyuanapiCallLogService 天元API调用记录服务
|
||
type TianyuanapiCallLogService struct {
|
||
tianyuanapiCallLogModel model.TianyuanapiCallLogModel
|
||
featureModel model.FeatureModel
|
||
}
|
||
|
||
// NewTianyuanapiCallLogService 创建天元API调用记录服务
|
||
func NewTianyuanapiCallLogService(
|
||
tianyuanapiCallLogModel model.TianyuanapiCallLogModel,
|
||
featureModel model.FeatureModel,
|
||
) *TianyuanapiCallLogService {
|
||
return &TianyuanapiCallLogService{
|
||
tianyuanapiCallLogModel: tianyuanapiCallLogModel,
|
||
featureModel: featureModel,
|
||
}
|
||
}
|
||
|
||
// CallLogOptions 调用记录选项
|
||
type CallLogOptions struct {
|
||
FeatureID string // 功能ID
|
||
ApiID string // API标识（如：YYSYBE08）
|
||
OrderID string // 订单ID（可选）
|
||
QueryID string // 查询ID（可选）
|
||
CallStatus int64 // 调用状态：0=失败，1=成功
|
||
ResponseTime int64 // 响应耗时（毫秒）
|
||
ErrorCode string // 错误码（失败时）
|
||
ErrorMessage string // 错误信息（失败时）
|
||
RequestParams interface{} // 请求参数（可选）
|
||
ResponseData interface{} // 响应数据（可选）
|
||
TransactionID string // 天元API流水号
|
||
}
|
||
|
||
// RecordCall 记录天元API调用
|
||
func (s *TianyuanapiCallLogService) RecordCall(ctx context.Context, opts CallLogOptions) error {
|
||
// 1. 获取feature的成本价
|
||
costPrice := 0.00
|
||
if opts.CallStatus == 1 { // 只有成功才计算成本
|
||
if opts.FeatureID == "" {
|
||
logx.Infof("记录API调用时feature_id为空，api_id=%s，无法获取成本价", opts.ApiID)
|
||
} else {
|
||
feature, err := s.featureModel.FindOne(ctx, opts.FeatureID)
|
||
if err == nil {
|
||
costPrice = feature.CostPrice
|
||
logx.Infof("记录API调用 - feature_id=%s, api_id=%s, cost_price=%f", opts.FeatureID, opts.ApiID, costPrice)
|
||
} else {
|
||
logx.Errorf("查询feature成本价失败，feature_id=%s, api_id=%s, err=%v", opts.FeatureID, opts.ApiID, err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 2. 转换参数和响应为JSON字符串
|
||
var requestParamsStr, responseDataStr *string
|
||
if opts.RequestParams != nil {
|
||
if bytes, err := json.Marshal(opts.RequestParams); err == nil {
|
||
jsonStr := string(bytes)
|
||
requestParamsStr = &jsonStr
|
||
}
|
||
}
|
||
if opts.ResponseData != nil {
|
||
// 只记录响应数据的前1000个字符，避免存储过大
|
||
if bytes, err := json.Marshal(opts.ResponseData); err == nil {
|
||
jsonStr := string(bytes)
|
||
if len(jsonStr) > 1000 {
|
||
jsonStr = jsonStr[:1000] + "...[truncated]"
|
||
}
|
||
responseDataStr = &jsonStr
|
||
}
|
||
}
|
||
|
||
// 3. 构建调用记录
|
||
callTime := time.Now()
|
||
deleteTime := sql.NullTime{}
|
||
callLog := &model.TianyuanapiCallLog{
|
||
FeatureId: opts.FeatureID,
|
||
ApiId: opts.ApiID,
|
||
OrderId: sql.NullString{},
|
||
QueryId: sql.NullString{},
|
||
CallStatus: opts.CallStatus,
|
||
CallTime: callTime,
|
||
ResponseTime: sql.NullInt64{},
|
||
CostPrice: costPrice,
|
||
ErrorCode: sql.NullString{},
|
||
ErrorMessage: sql.NullString{},
|
||
RequestParams: sql.NullString{},
|
||
ResponseData: sql.NullString{},
|
||
TransactionId: sql.NullString{},
|
||
CreateTime: callTime,
|
||
UpdateTime: callTime,
|
||
DeleteTime: deleteTime,
|
||
DelState: 0,
|
||
Version: 0,
|
||
}
|
||
|
||
// 4. 填充可选字段
|
||
if opts.OrderID != "" {
|
||
callLog.OrderId = sql.NullString{String: opts.OrderID, Valid: true}
|
||
}
|
||
if opts.QueryID != "" {
|
||
callLog.QueryId = sql.NullString{String: opts.QueryID, Valid: true}
|
||
}
|
||
if opts.ResponseTime > 0 {
|
||
callLog.ResponseTime = sql.NullInt64{Int64: opts.ResponseTime, Valid: true}
|
||
}
|
||
if opts.ErrorCode != "" {
|
||
callLog.ErrorCode = sql.NullString{String: opts.ErrorCode, Valid: true}
|
||
}
|
||
if opts.ErrorMessage != "" {
|
||
callLog.ErrorMessage = sql.NullString{String: opts.ErrorMessage, Valid: true}
|
||
}
|
||
if requestParamsStr != nil {
|
||
callLog.RequestParams = sql.NullString{String: *requestParamsStr, Valid: true}
|
||
}
|
||
if responseDataStr != nil {
|
||
callLog.ResponseData = sql.NullString{String: *responseDataStr, Valid: true}
|
||
}
|
||
if opts.TransactionID != "" {
|
||
callLog.TransactionId = sql.NullString{String: opts.TransactionID, Valid: true}
|
||
}
|
||
|
||
// 5. 插入记录
|
||
_, err := s.tianyuanapiCallLogModel.Insert(ctx, nil, callLog)
|
||
if err != nil {
|
||
logx.Errorf("插入天元API调用记录失败，feature_id=%s, api_id=%s, err=%v", opts.FeatureID, opts.ApiID, err)
|
||
return fmt.Errorf("插入调用记录失败: %w", err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// GetStatistics 获取统计信息
|
||
type StatisticsFilter struct {
|
||
FeatureID string // 功能ID
|
||
ApiID string // API标识
|
||
StartDate time.Time // 开始日期
|
||
EndDate time.Time // 结束日期
|
||
}
|
||
|
||
// Statistics 统计信息
|
||
type Statistics struct {
|
||
TotalCalls int64 // 总调用次数
|
||
SuccessCalls int64 // 成功次数
|
||
FailedCalls int64 // 失败次数
|
||
TotalCost float64 // 总成本（成功调用的成本之和）
|
||
}
|
||
|
||
// GetStatistics 获取统计信息
|
||
func (s *TianyuanapiCallLogService) GetStatistics(ctx context.Context, filter StatisticsFilter) (*Statistics, error) {
|
||
builder := s.tianyuanapiCallLogModel.SelectBuilder()
|
||
|
||
// 添加过滤条件
|
||
if filter.FeatureID != "" {
|
||
builder = builder.Where(squirrel.Eq{"feature_id": filter.FeatureID})
|
||
}
|
||
if filter.ApiID != "" {
|
||
builder = builder.Where(squirrel.Eq{"api_id": filter.ApiID})
|
||
}
|
||
if !filter.StartDate.IsZero() {
|
||
builder = builder.Where(squirrel.GtOrEq{"call_time": filter.StartDate})
|
||
logx.Infof("API成本统计 - 开始时间: %v", filter.StartDate)
|
||
}
|
||
if !filter.EndDate.IsZero() {
|
||
builder = builder.Where(squirrel.Lt{"call_time": filter.EndDate})
|
||
logx.Infof("API成本统计 - 结束时间: %v", filter.EndDate)
|
||
}
|
||
|
||
// 统计总调用次数
|
||
totalCalls, err := s.tianyuanapiCallLogModel.FindCount(ctx, builder, "id")
|
||
if err != nil {
|
||
return nil, fmt.Errorf("统计总调用次数失败: %w", err)
|
||
}
|
||
logx.Infof("API成本统计 - 总调用次数: %d", totalCalls)
|
||
|
||
// 统计成功次数
|
||
successBuilder := builder.Where(squirrel.Eq{"call_status": 1})
|
||
successCalls, err := s.tianyuanapiCallLogModel.FindCount(ctx, successBuilder, "id")
|
||
if err != nil {
|
||
return nil, fmt.Errorf("统计成功次数失败: %w", err)
|
||
}
|
||
logx.Infof("API成本统计 - 成功调用次数: %d", successCalls)
|
||
|
||
// 统计失败次数
|
||
failedCalls := totalCalls - successCalls
|
||
|
||
// 统计总成本（仅成功调用）
|
||
// 先打印SQL以便调试（复制builder避免影响后续查询）
|
||
debugBuilder := successBuilder
|
||
query, values, _ := debugBuilder.Columns("IFNULL(SUM(cost_price),0)").Where("del_state = ?", 0).ToSql()
|
||
logx.Infof("API成本统计 - SQL: %s, 参数: %v", query, values)
|
||
|
||
totalCost, err := s.tianyuanapiCallLogModel.FindSum(ctx, successBuilder, "cost_price")
|
||
if err != nil {
|
||
return nil, fmt.Errorf("统计总成本失败: %w", err)
|
||
}
|
||
logx.Infof("API成本统计 - 总成本: %f", totalCost)
|
||
|
||
return &Statistics{
|
||
TotalCalls: totalCalls,
|
||
SuccessCalls: successCalls,
|
||
FailedCalls: failedCalls,
|
||
TotalCost: totalCost,
|
||
}, nil
|
||
}
|