xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2026-05-11 04:04:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

check user power for create ai3d job

Browse Source
This commit is contained in:
RockYang
2025-09-13 10:29:40 +08:00
parent 330ac397d4
commit 359999c045
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/service/ai3d/service.go
View File

@@ -45,6 +45,13 @@ func NewService(db *gorm.DB, redisCli *redis.Client, tencentClient *Tencent3DCli
// CreateJob 创建3D生成任务 // CreateJob 创建3D生成任务
func (s *Service) CreateJob(userId uint, request vo.AI3DJobParams) (*model.AI3DJob, error) { 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 { switch request.Type {
case types.AI3DTaskTypeGitee: case types.AI3DTaskTypeGitee:
if s.giteeClient == nil { if s.giteeClient == nil {