浏览代码

Merge branch 'fix-2026' of https://git.openi.org.cn/OpenI/aiforge into fix-2026

pull/2292/head
liuzx 3 年前
父节点
当前提交
a5dfb5686a
共有 2 个文件被更改,包括 13 次插入1 次删除
  1. +4
    -0
      models/cloudbrain.go
  2. +9
    -1
      routers/repo/cloudbrain.go

+ 4
- 0
models/cloudbrain.go 查看文件

@@ -1559,6 +1559,10 @@ func GetStoppedJobWithNoDurationJob() ([]*Cloudbrain, error) {
Limit(100).
Find(&cloudbrains)
}
func GetStoppedJobWithNoStartTimeEndTime() ([]*Cloudbrain, error) {
cloudbrains := make([]*Cloudbrain, 0)
return cloudbrains, x.SQL("select * from cloudbrain where status in (?,?,?,?,?,?,?) and (start_time is null or end_time is null) limit 100", ModelArtsTrainJobCompleted, ModelArtsTrainJobFailed, ModelArtsTrainJobKilled, ModelArtsStopped, JobStopped, JobFailed, JobSucceeded).Find(&cloudbrains)
}

func GetCloudbrainCountByUserID(userID int64, jobType string) (int, error) {
count, err := x.In("status", JobWaiting, JobRunning).And("job_type = ? and user_id = ? and type = ?", jobType, userID, TypeCloudBrainOne).Count(new(Cloudbrain))


+ 9
- 1
routers/repo/cloudbrain.go 查看文件

@@ -1485,11 +1485,19 @@ func SyncCloudbrainStatus() {
}

func HandleTaskWithNoDuration(ctx *context.Context) {
mode := ctx.Query("mode")
log.Info("HandleTaskWithNoDuration start")
count := 0
start := time.Now().Unix()
for {
cloudBrains, err := models.GetStoppedJobWithNoDurationJob()
var cloudBrains []*models.Cloudbrain
var err error
if mode == "1" {
cloudBrains, err = models.GetStoppedJobWithNoStartTimeEndTime()
} else {
cloudBrains, err = models.GetStoppedJobWithNoDurationJob()
}

if err != nil {
log.Error("HandleTaskWithNoTrainJobDuration failed:", err.Error())
break


正在加载...
取消
保存