cloudbrain restart

modules/cloudbrain/cloudbrain.go

@@ -2,6 +2,7 @@ package cloudbrain
 
 import (
 	"code.gitea.io/gitea/modules/storage"
+	"encoding/json"
 	"errors"
 	"strconv"
 
@@ -108,6 +109,9 @@ func GenerateTask(ctx *context.Context, jobName, image, command, uuid, codePath,
 		uuid
 
 	var resourceSpec *models.ResourceSpec
+	if ResourceSpecs == nil {
+		json.Unmarshal([]byte(setting.ResourceSpecs), &ResourceSpecs)
+	}
 	for _, spec := range ResourceSpecs.ResourceSpec {
 		if resourceSpecId == spec.Id {
 			resourceSpec = spec
@@ -226,6 +230,9 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain) error {
 	jobName := task.JobName
 
 	var resourceSpec *models.ResourceSpec
+	if ResourceSpecs == nil {
+		json.Unmarshal([]byte(setting.ResourceSpecs), &ResourceSpecs)
+	}
 	for _, spec := range ResourceSpecs.ResourceSpec {
 		if task.ResourceSpecId == spec.Id {
 			resourceSpec = spec

routers/repo/cloudbrain.go

@@ -256,7 +256,7 @@ func CloudBrainRestart(ctx *context.Context) {
 		3、更新此任务的状态
 	*/
 
-	//todo: 是否启用事务?
+	//todo: 是否启用事务？处理时间太长，容易卡住
 
 	var jobID = ctx.Params(":jobid")
 	task, err := models.GetCloudbrainByJobID(jobID)