#1852

record the createTime before calling the cloudbrain task api

modules/cloudbrain/cloudbrain.go

@@ -1,6 +1,7 @@
 package cloudbrain
 
 import (
+	"code.gitea.io/gitea/modules/timeutil"
 	"encoding/json"
 	"errors"
 	"strconv"
@@ -194,6 +195,7 @@ func GenerateTask(ctx *context.Context, displayJobName, jobName, image, command,
 		datasetName = attach.Name
 	}
 
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := CreateJob(jobName, models.CreateJobParams{
 		JobName:    jobName,
 		RetryCount: 1,
@@ -294,6 +296,7 @@ func GenerateTask(ctx *context.Context, displayJobName, jobName, image, command,
 		BootFile:             bootFile,
 		DatasetName:          datasetName,
 		Parameters:           params,
+		CreatedUnix:          createTime,
 	})
 
 	if err != nil {
@@ -341,6 +344,7 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
 		return errors.New("no such resourceSpec")
 	}
 
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := CreateJob(jobName, models.CreateJobParams{
 		JobName:    jobName,
 		RetryCount: 1,
@@ -432,6 +436,7 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
 		GpuQueue:        task.GpuQueue,
 		ResourceSpecId:  task.ResourceSpecId,
 		ComputeResource: task.ComputeResource,
+		CreatedUnix:     createTime,
 	}
 
 	err = models.RestartCloudbrain(task, newTask)

modules/modelarts/modelarts.go

@@ -1,6 +1,7 @@
 package modelarts
 
 import (
+	"code.gitea.io/gitea/modules/timeutil"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -197,6 +198,7 @@ func GenerateTask(ctx *context.Context, jobName, uuid, description, flavor strin
 	if poolInfos == nil {
 		json.Unmarshal([]byte(setting.PoolInfos), &poolInfos)
 	}
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := CreateJob(models.CreateNotebookParams{
 		JobName:     jobName,
 		Description: description,
@@ -235,6 +237,7 @@ func GenerateTask(ctx *context.Context, jobName, uuid, description, flavor strin
 		Type:            models.TypeCloudBrainTwo,
 		Uuid:            uuid,
 		ComputeResource: models.NPUResource,
+		CreatedUnix:     createTime,
 	})
 
 	if err != nil {
@@ -254,7 +257,7 @@ func GenerateNotebook2(ctx *context.Context, displayJobName, jobName, uuid, desc
 		log.Error("GetNotebookImageName failed: %v", err.Error())
 		return err
 	}
-
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := createNotebook2(models.CreateNotebook2Params{
 		JobName:     jobName,
 		Description: description,
@@ -288,6 +291,7 @@ func GenerateNotebook2(ctx *context.Context, displayJobName, jobName, uuid, desc
 		ComputeResource: models.NPUResource,
 		Image:           imageName,
 		Description:     description,
+		CreatedUnix:     createTime,
 	})
 
 	if err != nil {
@@ -304,6 +308,7 @@ func GenerateNotebook2(ctx *context.Context, displayJobName, jobName, uuid, desc
 }
 
 func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (err error) {
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := createTrainJob(models.CreateTrainJobParams{
 		JobName:     req.JobName,
 		Description: req.Description,
@@ -364,6 +369,7 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (err error
 		EngineName:        req.EngineName,
 		VersionCount:      req.VersionCount,
 		TotalVersionCount: req.TotalVersionCount,
+		CreatedUnix:       createTime,
 	})
 
 	if err != nil {
@@ -375,6 +381,7 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (err error
 }
 
 func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, jobId string) (err error) {
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := createTrainJobVersion(models.CreateTrainJobVersionParams{
 		Description: req.Description,
 		Config: models.TrainJobVersionConfig{
@@ -451,6 +458,7 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, job
 		EngineName:        req.EngineName,
 		TotalVersionCount: VersionTaskList[0].TotalVersionCount + 1,
 		VersionCount:      VersionListCount + 1,
+		CreatedUnix:       createTime,
 	})
 	if err != nil {
 		log.Error("CreateCloudbrain(%s) failed:%v", req.JobName, err.Error())
@@ -526,6 +534,7 @@ func GetOutputPathByCount(TotalVersionCount int) (VersionOutputPath string) {
 }
 
 func GenerateInferenceJob(ctx *context.Context, req *GenerateInferenceJobReq) (err error) {
+	createTime := timeutil.TimeStampNow()
 	jobResult, err := createInferenceJob(models.CreateInferenceJobParams{
 		JobName:     req.JobName,
 		Description: req.Description,
@@ -591,6 +600,7 @@ func GenerateInferenceJob(ctx *context.Context, req *GenerateInferenceJobReq) (e
 		ModelVersion:      req.ModelVersion,
 		CkptName:          req.CkptName,
 		ResultUrl:         req.ResultUrl,
+		CreatedUnix:       createTime,
 	})
 
 	if err != nil {