diff --git a/service/task/worker.go b/service/task/worker.go index c8e131d..467be4a 100644 --- a/service/task/worker.go +++ b/service/task/worker.go @@ -107,6 +107,7 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.ModelGatewayTa // 计费处理 if len(model.BillingConfig) > 0 && len(task.BillingData) > 0 { + fmt.Printf("[handleOne] 计费处理 BillingConfig=%s, BillingData=%s", model.BillingConfig, task.BillingData) // 取请求阶段数据作为基础 billingInput := make(map[string]any) for k, v := range task.BillingData[0] { @@ -122,10 +123,11 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.ModelGatewayTa if billingResult != nil { task.BillingData[0] = billingResult } - + fmt.Printf("[handleOne] 计费处理 BillingResult=%s", billingResult) if billingResult != nil { task.BillingData[0] = billingResult totalFee := gconv.Float64(billingResult["total_fee"]) + fmt.Printf("[handleOne] 计费处理 totalFee=%f", totalFee) if totalFee > 0 { _ = gateway.DeductBalance(util.AsyncCtx(ctx), model.TenantId, -totalFee) } @@ -324,6 +326,8 @@ func (w *asyncWorker) parseAndRetry(ctx context.Context, body map[string]any, mo // 计费 if len(model.BillingConfig) > 0 && len(task.BillingData) > 0 { + fmt.Println("task.BillingData", task.BillingData) + fmt.Println("task.BillingData", task.BillingData) requestData := task.BillingData[0] retryData := make(map[string]any) for k, v := range requestData { @@ -334,9 +338,11 @@ func (w *asyncWorker) parseAndRetry(ctx context.Context, body map[string]any, mo retryData[k] = v } billingResult := util.CalculateBilling(model.BillingConfig, retryData) + fmt.Println("billingResult", billingResult) if billingResult != nil { task.BillingData = append(task.BillingData, billingResult) totalFee := gconv.Float64(billingResult["total_fee"]) + fmt.Println("totalFee", totalFee) if totalFee > 0 { _ = gateway.DeductBalance(util.AsyncCtx(ctx), task.TenantId, -totalFee) }