remote-task-excutor-cli/pkg/models/task.go

package models

import "time"

const (
	MainTaskID       = "1"
	DataReturnTaskID = "4"
	AITaskType       = "AI"
	DataReturnType   = "DataReturn"

	JobSetIDKey   = "jobSetID"
	LocalJobIDKey = "localJobID"
	ResponseOK    = "OK"
)

// SubtaskRequest 表示子任务请求
type SubtaskRequest struct {
	UserID     int        `json:"userID"`
	JobSetInfo JobSetInfo `json:"jobSetInfo"`
}

// JobSetInfo 表示任务集信息
type JobSetInfo struct {
	Jobs []Job `json:"jobs"`
}

// Job 表示单个任务
type Job struct {
	LocalJobID   string        `json:"localJobID"`
	Name         string        `json:"name,omitempty"`
	Description  string        `json:"description,omitempty"`
	Type         string        `json:"type"`
	JobSetID     string        `json:"jobSetID,omitempty"` // 停止任务时需要
	Files        *JobFiles     `json:"files,omitempty"`
	JobResources *JobResources `json:"jobResources,omitempty"`
	TargetJob    []TargetJob   `json:"targetJob,omitempty"`
	OnlyCreate   bool          `json:"onlyCreate,omitempty"`
}

// JobFiles 表示任务关联的文件
type JobFiles struct {
	Dataset  FileBinding  `json:"dataset"`
	Model    FileBinding  `json:"model,omitempty"`
	Image    ImageBinding `json:"image"`
	SubModel FileBinding  `json:"sub_model,omitempty"`
}

// FileBinding 表示文件绑定信息
type FileBinding struct {
	Type      string `json:"type"`
	BindingID int    `json:"bindingID"`
}

// ImageBinding 表示镜像绑定信息
type ImageBinding struct {
	Type    string `json:"type"`
	ImageID int    `json:"imageID"`
}

// JobResources 表示任务资源
type JobResources struct {
	ScheduleStrategy string    `json:"scheduleStrategy"`
	Clusters         []Cluster `json:"clusters"`
}

// Cluster 表示集群信息
type Cluster struct {
	ClusterID string          `json:"clusterID"`
	Runtime   Runtime         `json:"runtime"`
	Code      CodeInfo        `json:"code"`
	Resources []ResourcesItem `json:"resources"`
}

// Runtime 表示运行时配置
type Runtime struct {
	Envs   map[string]string `json:"envs"`
	Params map[string]string `json:"params"`
}

// CodeInfo 表示代码信息
type CodeInfo struct {
	Type      string `json:"type"`
	BindingID int    `json:"bindingID"`
}

// Resource 表示资源需求
type Resource struct {
	Type   string `json:"type"`
	Name   string `json:"name"`
	Number int    `json:"number"`
}

// TargetJob 表示目标任务
type TargetJob struct {
	TargetJobID string      `json:"targetJobID"`
	InputParams InputParams `json:"inputParams"`
}

// InputParams 表示输入参数
type InputParams struct {
	PackageName string `json:"PackageName"`
	ClusterID   string `json:"ClusterID"`
	Output      string `json:"Output"`
}

// SubmitTaskResponse 任务提交响应
type SubmitTaskResponse struct {
	Code    string `json:"code"`    // 状态码（字符串类型）
	Message string `json:"message"` // 消息
	Data    Data   `json:"data"`    // 响应数据
}

// Data 响应数据
type Data struct {
	JobSetID string `json:"jobSetID"` // 任务集ID
	Message  string `json:"message"`  // 数据层消息
}

type TaskQueryRequest struct {
	JobSetID   string `json:"jobSetID"`   // 用户ID
	LocalJobID string `json:"localJobID"` // 页码
}

// TaskDetailResponse 任务详情查询响应
type TaskDetailResponse struct {
	Code    string         `json:"code"`    // 状态码
	Message string         `json:"message"` // 消息
	Data    TaskDetailData `json:"data"`    // 任务详情数据
}

// TaskDetailData 任务详情数据
type TaskDetailData struct {
	Name            string        `json:"name"`            // 任务名称
	Description     string        `json:"description"`     // 任务描述
	StartTime       string        `json:"startTime"`       // 开始时间
	EndTime         string        `json:"endTime"`         // 结束时间
	Strategy        int           `json:"strategy"`        // 策略
	SynergyStatus   int           `json:"synergyStatus"`   // 协同状态
	ClusterInfos    []ClusterItem `json:"clusterInfos"`    // 集群信息列表
	SubTaskInfos    []SubTaskInfo `json:"subTaskInfos"`    // 子任务信息列表
	TaskTypeDict    string        `json:"taskTypeDict"`    // 任务类型字典
	AdapterTypeDict string        `json:"adapterTypeDict"` // 适配器类型字典
}

// SubTaskInfo 子任务信息
type SubTaskInfo struct {
	ID          string `json:"id"`          // 子任务ID
	Name        string `json:"name"`        // 子任务名称
	ClusterID   string `json:"clusterId"`   // 集群ID
	ClusterName string `json:"clusterName"` // 集群名称
	Status      string `json:"status"`      // 任务状态
	Remark      string `json:"remark"`      // 备注
	InferURL    string `json:"inferUrl"`    // 推理URL
	WorkDir     string `json:"workDir"`     // 工作目录
	AppName     string `json:"appName"`     // 应用名称
}

// TaskLogResponse 任务日志响应
type TaskLogResponse struct {
	Code    string `json:"code"`
	Message string `json:"message"`
	Data    string `json:"data"`
}

// TaskResultDetailResponse 任务详情响应
type TaskResultDetailResponse struct {
	Code    string               `json:"code"`    // 状态码
	Message string               `json:"message"` // 消息
	Data    TaskResultDetailData `json:"data"`    // 任务详情数据
}

// TaskResultDetailData 任务详情数据
type TaskResultDetailData struct {
	PcmJobData PcmJobData `json:"pcmJobData"` // PCM任务数据
}

// PcmJobData PCM任务数据
type PcmJobData struct {
	TaskID        string        `json:"taskID"`        // 任务ID
	JobSubmitInfo JobSubmitInfo `json:"jobSubmitInfo"` // 任务提交信息
	ResultFiles   []ResultFile  `json:"resultFiles"`   // 结果文件列表
	Status        string        `json:"status"`        // 任务状态
	ErrorMsg      string        `json:"errorMsg"`      // 错误消息
}

// JobSubmitInfo 任务提交信息
type JobSubmitInfo struct {
	Type string              `json:"type"` // 类型
	Info JobSubmitInfoDetail `json:"info"` // 任务详情
}

// JobSubmitInfoDetail 任务提交详情
type JobSubmitInfoDetail struct {
	Name            string          `json:"name"`            // 任务名称
	Description     string          `json:"description"`     // 任务描述
	JobResources    JobResources    `json:"jobResources"`    // 任务资源
	DataDistributes DataDistributes `json:"dataDistributes"` // 数据分布
}

// DataDistributes 数据分布
type DataDistributes struct {
	Dataset []DatasetDistribute `json:"dataset"` // 数据集分布
	Code    []CodeDistribute    `json:"code"`    // 代码分布
	Image   []ImageDistribute   `json:"image"`   // 镜像分布
	Model   []ModelDistribute   `json:"model"`   // 模型分布
}

// DatasetDistribute 数据集分布
type DatasetDistribute struct {
	DataName  string           `json:"dataName"`  // 数据名称
	PackageID int              `json:"packageID"` // 包ID
	Clusters  []ClusterStorage `json:"clusters"`  // 集群存储
}

// CodeDistribute 代码分布
type CodeDistribute struct {
	DataName  string           `json:"dataName"`  // 数据名称
	PackageID int              `json:"packageID"` // 包ID
	Output    string           `json:"output"`    // 输出路径
	Clusters  []ClusterStorage `json:"clusters"`  // 集群存储
}

// ImageDistribute 镜像分布
type ImageDistribute struct {
	DataName  string           `json:"dataName"`  // 数据名称
	PackageID int              `json:"packageID"` // 包ID
	Clusters  []ClusterStorage `json:"clusters"`  // 集群存储
}

// ModelDistribute 模型分布
type ModelDistribute struct {
	DataName  string           `json:"dataName"`  // 数据名称
	PackageID int              `json:"packageID"` // 包ID
	Clusters  []ClusterStorage `json:"clusters"`  // 集群存储
}

// ClusterStorage 集群存储信息
type ClusterStorage struct {
	ClusterID string `json:"clusterID"` // 集群ID
	StorageID int    `json:"storageID"` // 存储ID
	JsonData  string `json:"jsonData"`  // JSON数据
}

// ResultFile 结果文件
type ResultFile struct {
	ClusterID string         `json:"clusterID"` // 集群ID
	Objects   []ResultObject `json:"objects"`   // 结果对象列表
}

// ResultObject 结果对象
type ResultObject struct {
	ObjectID   int        `json:"objectID"`   // 对象ID
	PackageID  int        `json:"packageID"`  // 包ID
	Path       string     `json:"path"`       // 路径
	Size       string     `json:"size"`       // 大小
	FileHash   string     `json:"fileHash"`   // 文件哈希
	Redundancy Redundancy `json:"redundancy"` // 冗余配置
	CreateTime time.Time  `json:"createTime"` // 创建时间
	UpdateTime time.Time  `json:"updateTime"` // 更新时间
}

// InferenceTaskStatusRequest 推理任务状态查询请求
type InferenceTaskStatusRequest struct {
	LocalJobID string `json:"localJobID" binding:"required"` // 本地任务ID
	JobSetID   string `json:"jobSetID" binding:"required"`   // 任务集ID
}

// InferenceTaskStatusResponse 推理任务状态查询响应
type InferenceTaskStatusResponse struct {
	Code int                     `json:"code"` // 状态码
	Msg  string                  `json:"msg"`  // 消息
	Data InferenceTaskStatusData `json:"data"` // 任务状态数据
}

// InferenceTaskStatusData 推理任务状态数据
type InferenceTaskStatusData struct {
	Status string `json:"status"` // 任务状态
	URL    string `json:"url"`    // 推理URL
}

// InferenceAsyncSubmitResponse 异步提交推理任务响应
type InferenceAsyncSubmitResponse struct {
	Code int                          `json:"code"`
	Msg  string                       `json:"msg"`
	Data InferenceAsyncSubmitRespData `json:"data"`
}

type InferenceAsyncSubmitRespData struct {
	TaskID string `json:"task_id"`
}

// InferenceAsyncTaskStatusRequest 异步任务查询请求
type InferenceAsyncTaskStatusRequest struct {
	TaskID string `json:"task_id" binding:"required"`
}

// InferenceAsyncTaskStatusResponse 异步任务查询响应
type InferenceAsyncTaskStatusResponse struct {
	Code int                          `json:"code"`
	Msg  string                       `json:"msg"`
	Data InferenceAsyncTaskStatusData `json:"data"`
}

type InferenceAsyncTaskStatusData struct {
	TaskID   string                       `json:"task_id"`
	Status   string                       `json:"status"`             // PENDING/RUNNING/SUBMITTING/COMPLETED/FAILED
	Progress string                       `json:"progress,omitempty"` // 进度描述
	JobSetID string                       `json:"job_set_id,omitempty"`
	Error    string                       `json:"error,omitempty"`
	Result   *InferenceSubmitTaskResponse `json:"result,omitempty"` // 任务完成后的结果
}

// StopInferenceTaskRequest 停止推理任务请求
type StopInferenceTaskRequest struct {
	LocalJobID string `json:"localJobID" binding:"required"` // 本地任务ID
	JobSetID   string `json:"jobSetID" binding:"required"`   // 任务集ID
}

// StopInferenceTaskResponse 停止推理任务响应
type StopInferenceTaskResponse struct {
	Code    int    `json:"code"`    // 状态码
	Message string `json:"message"` // 消息
}

// ReSubmitTaskResponse 重启推理任务响应
type ReSubmitTaskResponse struct {
	Code    int              `json:"code"`    // 状态码
	Message string           `json:"message"` // 消息
	Data    ReSubmitTaskData `json:"data"`    // 响应数据
}

// ReSubmitTaskData 重启推理任务响应数据
type ReSubmitTaskData struct {
	JobSetID   string `json:"jobSetID"`   // 任务集ID（来自第三方接口返回）
	LocalJobID string `json:"localJobID"` // 本地任务ID（来自入参 jobs）
}

// StopInferenceSubmitRequest 停止推理任务提交给第三方的请求
type StopInferenceSubmitRequest struct {
	UserID     int        `json:"userID"`     // 用户ID
	JobSetInfo JobSetInfo `json:"jobSetInfo"` // 任务集信息
}

// StopInferenceSubmitResponse 停止推理任务第三方接口响应
type StopInferenceSubmitResponse struct {
	Code    string                  `json:"code"`    // 状态码
	Message string                  `json:"message"` // 消息
	Data    StopInferenceSubmitData `json:"data"`    // 数据
}

// StopInferenceSubmitData 停止推理任务响应数据
type StopInferenceSubmitData struct {
	JobSetID          string            `json:"jobSetID"`          // 任务集ID
	FilesUploadScheme FilesUploadScheme `json:"filesUploadScheme"` // 文件上传方案
	Message           string            `json:"message"`           // 消息
}

// FilesUploadScheme 文件上传方案
type FilesUploadScheme struct {
	LocalFileUploadSchemes []interface{} `json:"localFileUploadSchemes"` // 本地文件上传方案
}

// InferenceTaskDetailResponse 推理任务详情响应（第三方接口返回的原始格式）
type InferenceTaskDetailResponse struct {
	Code    string                  `json:"code"`    // 状态码
	Message string                  `json:"message"` // 消息
	Data    InferenceTaskDetailData `json:"data"`    // 任务详情数据
}

// InferenceTaskDetailData 推理任务详情数据（第三方接口返回的原始格式）
type InferenceTaskDetailData struct {
	Type     string            `json:"type"`     // 类型
	Instance InferenceInstance `json:"instance"` // 实例信息
}

// InferenceInstance 推理实例信息
type InferenceInstance struct {
	InstanceName string `json:"InstanceName"` // 实例名称
	InstanceId   string `json:"InstanceId"`   // 实例ID
	ModelName    string `json:"ModelName"`    // 模型名称
	ModelType    string `json:"ModelType"`    // 模型类型
	InferCard    string `json:"InferCard"`    // 推理卡类型
	InferUrl     string `json:"InferUrl"`     // 推理URL
	ClusterName  string `json:"ClusterName"`  // 集群名称
	ClusterType  string `json:"ClusterType"`  // 集群类型
	Status       string `json:"Status"`       // 状态
	CreatedTime  string `json:"CreatedTime"`  // 创建时间
}

// InferenceSubmitTaskResponse 推理任务提交响应
type InferenceSubmitTaskResponse struct {
	Code int                             `json:"code"` // 状态码
	Msg  string                          `json:"msg"`  // 消息
	Data InferenceSubmitTaskResponseData `json:"data"` // 响应数据
}

// InferenceSubmitTaskResponseData 推理任务提交响应数据
type InferenceSubmitTaskResponseData struct {
	TaskInfo   SubtaskRequest      `json:"taskInfo"`   // 任务信息
	ResultInfo InferenceResultInfo `json:"resultInfo"` // 结果信息
}

// InferenceTaskInfo 推理任务信息
type InferenceTaskInfo struct {
	UserID     int                 `json:"userID"`     // 用户ID
	JobSetInfo InferenceJobSetInfo `json:"jobSetInfo"` // 任务集信息
}

// InferenceJobSetInfo 推理任务集信息
type InferenceJobSetInfo struct {
	Jobs []InferenceJob `json:"jobs"` // 任务列表
}

// InferenceJob 推理任务
type InferenceJob struct {
	LocalJobID   string                 `json:"localJobID"`   // 本地任务ID
	Name         string                 `json:"name"`         // 任务名称
	Description  string                 `json:"description"`  // 任务描述
	Type         string                 `json:"type"`         // 任务类型
	Files        *InferenceJobFiles     `json:"files"`        // 文件信息
	JobResources *InferenceJobResources `json:"jobResources"` // 资源信息
}

// InferenceJobFiles 推理任务文件
type InferenceJobFiles struct {
	Model    FileBinding  `json:"model"`               // 模型绑定
	SubModel *FileBinding `json:"sub_model,omitempty"` // 增量模型绑定（可选）
	Image    ImageBinding `json:"image"`               // 镜像绑定
}

// InferenceJobResources 推理任务资源
type InferenceJobResources struct {
	ScheduleStrategy string             `json:"scheduleStrategy"` // 调度策略
	Clusters         []InferenceCluster `json:"clusters"`         // 集群列表
}

// InferenceCluster 推理集群信息
type InferenceCluster struct {
	ClusterID string          `json:"clusterID"` // 集群ID
	Runtime   Runtime         `json:"runtime"`   // 运行时配置
	Code      CodeInfo        `json:"code"`      // 代码信息
	Resources []ResourcesItem `json:"resources"` // 资源列表
}

// InferenceResultInfo 推理结果信息
type InferenceResultInfo struct {
	LocalJobID string `json:"localJobID"` // 本地任务ID
	JobSetID   string `json:"jobSetID"`   // 任务集ID
}