From efad47bf8d20d9cf48f6f635cb60f0f539e61888 Mon Sep 17 00:00:00 2001 From: liuzx Date: Fri, 26 Aug 2022 14:27:20 +0800 Subject: [PATCH 1/2] add sleeptime when delete cloudbrain task --- routers/repo/cloudbrain.go | 1 + 1 file changed, 1 insertion(+) diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 457f275ed..cd5f7430a 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -1251,6 +1251,7 @@ func DeleteJobsByRepoID(repoID int64) { log.Warn("Failed to get cloudBrain info", err) return } + time.Sleep(time.Second * 120) DeleteJobs(cloudBrains) } From 69b31dc48e259c0f7c8fa2c0f2b99b678a616260 Mon Sep 17 00:00:00 2001 From: liuzx Date: Fri, 26 Aug 2022 17:43:43 +0800 Subject: [PATCH 2/2] add stopping status --- models/cloudbrain.go | 2 +- routers/repo/cloudbrain.go | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/models/cloudbrain.go b/models/cloudbrain.go index c1c01b581..dc11d3459 100755 --- a/models/cloudbrain.go +++ b/models/cloudbrain.go @@ -1889,7 +1889,7 @@ func GetCloudbrainCountByUserID(userID int64, jobType string) (int, error) { func GetCloudbrainRunCountByRepoID(repoID int64) (int, error) { count, err := x.In("status", JobWaiting, JobRunning, ModelArtsCreateQueue, ModelArtsCreating, ModelArtsStarting, ModelArtsReadyToStart, ModelArtsResizing, ModelArtsStartQueuing, ModelArtsRunning, ModelArtsRestarting, ModelArtsTrainJobInit, - ModelArtsTrainJobImageCreating, ModelArtsTrainJobSubmitTrying, ModelArtsTrainJobWaiting, ModelArtsTrainJobRunning, + ModelArtsTrainJobImageCreating, ModelArtsTrainJobSubmitTrying, ModelArtsTrainJobWaiting, ModelArtsTrainJobRunning, ModelArtsStopping, ModelArtsResizing, ModelArtsTrainJobScaling, ModelArtsTrainJobCheckInit, ModelArtsTrainJobCheckRunning, ModelArtsTrainJobCheckRunningCompleted).And("repo_id = ?", repoID).Count(new(Cloudbrain)) return int(count), err } diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index cd5f7430a..457f275ed 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -1251,7 +1251,6 @@ func DeleteJobsByRepoID(repoID int64) { log.Warn("Failed to get cloudBrain info", err) return } - time.Sleep(time.Second * 120) DeleteJobs(cloudBrains) }