129 lines
3.6 KiB
Go
129 lines
3.6 KiB
Go
package manager
|
||
|
||
import (
|
||
"gitlink.org.cn/cloudream/common/pkgs/mq"
|
||
schsdk "gitlink.org.cn/cloudream/common/sdks/scheduler"
|
||
jobmod "gitlink.org.cn/cloudream/scheduler/common/models/job"
|
||
)
|
||
|
||
type JobService interface {
|
||
SubmitJobSet(msg *SubmitJobSet) (*SubmitJobSetResp, *mq.CodeMessage)
|
||
|
||
JobSetLocalFileUploaded(msg *JobSetLocalFileUploaded) (*JobSetLocalFileUploadedResp, *mq.CodeMessage)
|
||
|
||
GetJob(msg *GetJob) (*GetJobResp, *mq.CodeMessage)
|
||
|
||
// GetJobSetJobs(msg *GetJobSetJobs) (*GetJobSetJobsResp, *mq.CodeMessage)
|
||
}
|
||
|
||
// 提交任务集
|
||
var _ = Register(Service.SubmitJobSet)
|
||
|
||
type SubmitJobSet struct {
|
||
mq.MessageBodyBase
|
||
JobSet schsdk.JobSetInfo `json:"jobSet"`
|
||
PreScheduleScheme jobmod.JobSetPreScheduleScheme `json:"preScheduleScheme"`
|
||
}
|
||
type SubmitJobSetResp struct {
|
||
mq.MessageBodyBase
|
||
JobSetID schsdk.JobSetID `json:"jobSetID"`
|
||
}
|
||
|
||
func NewSubmitJobSet(jobSet schsdk.JobSetInfo, preScheduleScheme jobmod.JobSetPreScheduleScheme) *SubmitJobSet {
|
||
return &SubmitJobSet{
|
||
JobSet: jobSet,
|
||
PreScheduleScheme: preScheduleScheme,
|
||
}
|
||
}
|
||
func NewSubmitJobSetResp(jobSetID schsdk.JobSetID) *SubmitJobSetResp {
|
||
return &SubmitJobSetResp{
|
||
JobSetID: jobSetID,
|
||
}
|
||
}
|
||
func (c *Client) SubmitJobSet(msg *SubmitJobSet, opts ...mq.RequestOption) (*SubmitJobSetResp, error) {
|
||
return mq.Request(Service.SubmitJobSet, c.roundTripper, msg, opts...)
|
||
}
|
||
|
||
// JobSet中需要使用的一个文件上传完成
|
||
var _ = Register(Service.JobSetLocalFileUploaded)
|
||
|
||
type JobSetLocalFileUploaded struct {
|
||
mq.MessageBodyBase
|
||
JobSetID schsdk.JobSetID `json:"jobSetID"`
|
||
LocalPath string `json:"localPath"`
|
||
Error string `json:"error"` // 如果上传文件失败，那么这个字段说明了失败原因
|
||
PackageID int64 `json:"packageID"` // 如果上传文件成功，那么这个字段是上传之后得到的PackageID
|
||
}
|
||
type JobSetLocalFileUploadedResp struct {
|
||
mq.MessageBodyBase
|
||
}
|
||
|
||
func NewJobSetLocalFileUploaded(jobSetID schsdk.JobSetID, localPath string, err string, packageID int64) *JobSetLocalFileUploaded {
|
||
return &JobSetLocalFileUploaded{
|
||
JobSetID: jobSetID,
|
||
LocalPath: localPath,
|
||
Error: err,
|
||
PackageID: packageID,
|
||
}
|
||
}
|
||
func NewJobSetLocalFileUploadedResp() *JobSetLocalFileUploadedResp {
|
||
return &JobSetLocalFileUploadedResp{}
|
||
}
|
||
func (c *Client) JobSetLocalFileUploaded(msg *JobSetLocalFileUploaded, opts ...mq.RequestOption) (*JobSetLocalFileUploadedResp, error) {
|
||
return mq.Request(Service.JobSetLocalFileUploaded, c.roundTripper, msg, opts...)
|
||
}
|
||
|
||
// 获取任务数据
|
||
type GetJob struct {
|
||
mq.MessageBodyBase
|
||
JobID schsdk.JobID `json:"jobID"`
|
||
}
|
||
type GetJobResp struct {
|
||
mq.MessageBodyBase
|
||
Job jobmod.Job `json:"job"`
|
||
}
|
||
|
||
func NewGetJob(jobID schsdk.JobID) *GetJob {
|
||
return &GetJob{
|
||
JobID: jobID,
|
||
}
|
||
}
|
||
func NewGetJobResp(job jobmod.Job) *GetJobResp {
|
||
return &GetJobResp{
|
||
Job: job,
|
||
}
|
||
}
|
||
func (c *Client) GetJob(msg *GetJob, opts ...mq.RequestOption) (*GetJobResp, error) {
|
||
return mq.Request(Service.GetJob, c.roundTripper, msg, opts...)
|
||
}
|
||
|
||
/*
|
||
// 获取指定任务集中的所有任务数据
|
||
type GetJobSetJobs struct {
|
||
mq.MessageBodyBase
|
||
JobSetID string `json:"jobSetID"`
|
||
}
|
||
type GetJobSetJobsResp struct {
|
||
mq.MessageBodyBase
|
||
Jobs []jobmod.Job `json:"jobs"`
|
||
}
|
||
|
||
func NewGetJobSetJobs(jobSetID string) *GetJobSetJobs {
|
||
return &GetJobSetJobs{
|
||
JobSetID: jobSetID,
|
||
}
|
||
}
|
||
func NewGetJobSetJobsResp(jobs []jobmod.Job) *GetJobSetJobsResp {
|
||
return &GetJobSetJobsResp{
|
||
Jobs: jobs,
|
||
}
|
||
}
|
||
func (c *Client) GetJobSetJobs(msg *GetJobSetJobs, opts ...mq.RequestOption) (*GetJobSetJobsResp, error) {
|
||
return mq.Request(Service.GetJobSetJobs, c.rabbitCli, msg, opts...)
|
||
}
|
||
*/
|
||
|
||
func init() {
|
||
mq.RegisterUnionType(jobmod.JobTypeUnion)
|
||
}
|