Browse Source

新增工作流状态恢复

pull/52/head
JeshuaRen 1 year ago
parent
commit
b916bbac59
2 changed files with 15 additions and 4 deletions
  1. +2
    -0
      sdks/scheduler/jobflow.go
  2. +13
    -4
      sdks/scheduler/models.go

+ 2
- 0
sdks/scheduler/jobflow.go View File

@@ -84,6 +84,7 @@ type JobFlowRunDAO struct {
Description string `gorm:"column:description;size:255" json:"description"`
Content string `gorm:"column:content;type:text" json:"content"`
Status string `gorm:"column:status;type:enum('running','fail','success')" json:"status"`
Token string `gorm:"column:token;size:255" json:"token"`
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime" json:"createdAt"`
FinishAt *time.Time `gorm:"column:finish_at" json:"finishedAt"`
}
@@ -95,6 +96,7 @@ type JobFlowRun struct {
Description string `gorm:"column:description;size:255" json:"description"`
Content FlowData `gorm:"column:content;type:text" json:"content"`
Status string `gorm:"column:status;type:enum('running','fail','success')" json:"status"`
Token string `gorm:"column:token;size:255" json:"token"`
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime" json:"createdAt"`
FinishAt *time.Time `gorm:"column:finish_at" json:"finishedAt"`
}

+ 13
- 4
sdks/scheduler/models.go View File

@@ -630,7 +630,7 @@ type InferencePlatform struct {
}

type JobOutput interface {
Output()
Output2()
}

var JobOutputTypeUnion = types.NewTypeUnion[JobOutput](
@@ -642,7 +642,7 @@ var _ = serder.UseTypeUnionInternallyTagged(&JobOutputTypeUnion, "type")

type JobOutputBase struct{}

func (d *JobOutputBase) Output() {}
func (d *JobOutputBase) Output2() {}

type PublicOutput struct {
serder.Metadata `union:"object"`
@@ -650,10 +650,18 @@ type PublicOutput struct {
Type string `json:"type"`
}

type HPCOutput struct {
serder.Metadata `union:"HPCSlurm"`
JobOutputBase
Type string `json:"type"`
Output string `json:"output"`
}

type AIJobOutput struct {
serder.Metadata `union:"object"`
serder.Metadata `union:"AI"`
JobOutputBase
Type string `json:"type"`
Type string `json:"type"`
Output string `json:"output"`
}

type BindingJobOutput struct {
@@ -668,6 +676,7 @@ type DataReturnJobOutput struct {
JobOutputBase
Type string `json:"type"`
ReportMessage TrainJobStatusReport `json:"report"`
PackageID cdssdk.PackageID `json:"packageID"`
}

type JobStatusReport interface {


Loading…
Cancel
Save