Browse Source

新增工作流状态查询功能

pull/52/head
JeshuaRen 6 months ago
parent
commit
a059fc7294
3 changed files with 39 additions and 0 deletions
  1. +9
    -0
      sdks/pcmscheduler/models.go
  2. +28
    -0
      sdks/scheduler/jobflow.go
  3. +2
    -0
      sdks/scheduler/models.go

+ 9
- 0
sdks/pcmscheduler/models.go View File

@@ -56,6 +56,7 @@ const (

FailedStatus = "failed"
SuccessStatus = "success"
SucceededStatus = "succeeded"
UploadingStatus = "uploading"
RunningStatus = "running"

@@ -85,6 +86,14 @@ const (

StartNode = "StartTask"
EndNode = "EndTask"

NodeTypeBinding = "binding"
NodeTypeUpload = "upload"
NodeTypeDataReturn = "data_return"
NodeTypeHPCCreate = "hpc_create"
NodeTypeInference = "inference"
NodeTypeAICreate = "ai_job_create"
NodeTypeAIJobRun = "ai_job_run"
)

type TaskID int64


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

@@ -67,3 +67,31 @@ type JobFlow struct {
UpdatedAt time.Time `json:"updatedAt"`
CreatedAt time.Time `json:"createdAt"`
}

type JobFlowRunStatus struct {
RunID JobSetID `gorm:"column:run_id;primaryKey" json:"runID"`
NodeType string `gorm:"column:node_type;size:100" json:"nodeType"`
NodeID string `gorm:"column:node_id;size:255" json:"nodeID"`
Status string `gorm:"column:status;type:enum('pending','running','fail','success')" json:"status"`
RunLog string `gorm:"column:run_log;type:text" json:"runLog"`
}

type JobFlowRunDAO struct {
ID JobSetID `gorm:"column:id;primaryKey;" json:"runID"`
UserID cdssdk.UserID `gorm:"column:user_id" json:"userID"`
Name string `gorm:"column:name;size:255;not null" json:"name"`
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"`
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime" json:"createdAt"`
}

type JobFlowRun struct {
ID JobSetID `gorm:"column:id;primaryKey;" json:"runID"`
UserID cdssdk.UserID `gorm:"column:user_id" json:"userID"`
Name string `gorm:"column:name;size:255;not null" json:"name"`
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"`
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime" json:"createdAt"`
}

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

@@ -31,6 +31,8 @@ type JobID string

type JobSetID string

type DataID int64

type ImageID int64

// 计算中心ID


Loading…
Cancel
Save