check user power for create ai3d job

This commit is contained in:
RockYang
2025-09-13 10:29:40 +08:00
parent 330ac397d4
commit 359999c045

View File

@@ -45,6 +45,13 @@ func NewService(db *gorm.DB, redisCli *redis.Client, tencentClient *Tencent3DCli
// CreateJob 创建3D生成任务
func (s *Service) CreateJob(userId uint, request vo.AI3DJobParams) (*model.AI3DJob, error) {
// 检查用户算力
var user model.User
s.db.Where("id = ?", userId).First(&user)
if user.Power < request.Power {
return nil, fmt.Errorf("用户算力不足")
}
switch request.Type {
case types.AI3DTaskTypeGitee:
if s.giteeClient == nil {