red-future/model-gateway
4
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

refactor(task): 优化任务处理中的余额检查逻辑

Browse Source
This commit is contained in:
WangLiZhao
2026-07-01 18:47:13 +08:00
parent d18016e698
commit a332aa438b
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
service/task/worker.go
+6 -8
View File
@@ -38,21 +38,19 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.ModelGatewayTa
rawData []byte rawData []byte
result map[string]any result map[string]any
err error err error
surplus float64
) )
g.Log().Infof(ctx, "[handleOne] 开始 taskId=%s model=%s", task.TaskID, task.ModelName) g.Log().Infof(ctx, "[handleOne] 开始 taskId=%s model=%s", task.TaskID, task.ModelName)
// ============================================ // ============================================
// 1) 查询余额 // 1) 查询余额
// ============================================ // ============================================
var surplus float64 surplus, _ = gateway.GetTenantSurplus(ctx, task.TenantId)
if len(model.BillingConfig) > 0 { if surplus <= 0 {
surplus, _ = gateway.GetTenantSurplus(ctx, task.TenantId) w.failTask(ctx, task, startTime, "租户余额不足")
if surplus <= 0 { return
w.failTask(ctx, task, startTime, "租户余额不足")
return
}
g.Log().Infof(ctx, "[handleOne] 当前余额 tenantId=%d surplus=%.2f", task.TenantId, surplus)
} }
g.Log().Infof(ctx, "[handleOne] 当前余额 tenantId=%d surplus=%.2f", task.TenantId, surplus)
// ============================================ // ============================================
// 2) 调用模型 // 2) 调用模型