Browse Source

新增任务相关接口

pull/53/head
JeshuaRen 4 months ago
parent
commit
c49c82d970
3 changed files with 25 additions and 19 deletions
  1. +1
    -0
      sdks/pcmscheduler/job.go
  2. +23
    -19
      sdks/pcmscheduler/jobmgr.go
  3. +1
    -0
      utils/http2/http.go

+ 1
- 0
sdks/pcmscheduler/job.go View File

@@ -13,6 +13,7 @@ type PCMJob struct {
JobSetID schsdk.JobSetID `gorm:"column:jobset_id" json:"jobSetID"`
LocalJobID string `gorm:"column:local_job_id" json:"localJobID"`
Param string `gorm:"column:param" json:"param"`
Output string `gorm:"column:run_output" json:"runOutput"`
Token string `gorm:"column:token" json:"token"`
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"`
}


+ 23
- 19
sdks/pcmscheduler/jobmgr.go View File

@@ -5,13 +5,14 @@ import (
"gitlink.org.cn/cloudream/common/utils/http2"
"gitlink.org.cn/cloudream/common/utils/serder"
"net/url"
"path/filepath"
"strings"
)

type QueryTasksReq struct {
PageNum int64 `json:"pageNum"`
PageSize int64 `json:"pageSize"`
Type int64 `json:"type"`
PageNum int64 `form:"pageNum"`
PageSize int64 `form:"pageSize"`
Type int64 `form:"type"`
}

type QueryTasksResp struct {
@@ -45,13 +46,13 @@ type TaskModel struct {
}

func (c *Client) QueryTasks(req QueryTasksReq, token string) (*QueryTasksResp, error) {
targetUrl, err := url.JoinPath(c.baseURL, "/pcm/v1/core/task/list")
targetUrl, err := url.JoinPath(c.baseURL, "/core/task/list")
if err != nil {
return nil, err
}

resp, err := http2.GetJSON(targetUrl, http2.RequestParam{
Body: req,
Query: req,
Header: map[string]string{
"Authorization": token,
},
@@ -78,7 +79,7 @@ func (c *Client) QueryTasks(req QueryTasksReq, token string) (*QueryTasksResp, e
}

type QueryDetailsReq struct {
TaskID string `json:"id"`
TaskID string `form:"id"`
}

type QueryDetailsResp struct {
@@ -139,13 +140,13 @@ type ClusterInfo struct {
}

func (c *Client) QueryDetails(req QueryDetailsReq, token string) (*QueryDetailsResp, error) {
targetUrl, err := url.JoinPath(c.baseURL, "/pcm/v1/core/task/details")
targetUrl, err := url.JoinPath(c.baseURL, "/core/task/details")
if err != nil {
return nil, err
}

resp, err := http2.GetJSON(targetUrl, http2.RequestParam{
Body: req,
Query: req,
Header: map[string]string{
"Authorization": token,
},
@@ -172,7 +173,7 @@ func (c *Client) QueryDetails(req QueryDetailsReq, token string) (*QueryDetailsR
}

type QueryHPCTrainReq struct {
TaskID string `json:"taskId"`
TaskID string `form:"taskId"`
}

type QueryHPCTrainResp struct {
@@ -183,13 +184,13 @@ type QueryHPCTrainResp struct {
}

func (c *Client) QueryHPCTrainLog(req QueryHPCTrainReq, token string) (*QueryHPCTrainResp, error) {
targetUrl, err := url.JoinPath(c.baseURL, "/pcm/v1/hpc/jobLogs")
targetUrl, err := url.JoinPath(c.baseURL, "/hpc/jobLogs")
if err != nil {
return nil, err
}

resp, err := http2.GetJSON(targetUrl, http2.RequestParam{
Body: req,
Query: req,
Header: map[string]string{
"Authorization": token,
},
@@ -216,10 +217,10 @@ func (c *Client) QueryHPCTrainLog(req QueryHPCTrainReq, token string) (*QueryHPC
}

type QueryAITrainLogReq struct {
AdapterID string `json:"adapterId"`
ClusterID string `json:"clusterId"`
TaskID string `json:"taskId"`
InstanceNum string `json:"instanceNum"`
AdapterID string `form:"adapterId"`
ClusterID string `form:"clusterId"`
TaskID string `form:"taskId"`
InstanceNum string `form:"instanceNum"`
}

type QueryAITrainLogResp struct {
@@ -227,16 +228,19 @@ type QueryAITrainLogResp struct {
}

func (c *Client) QueryAITrainLog(req QueryAITrainLogReq, token string) (*QueryAITrainLogResp, error) {
targetUrl, err := url.JoinPath(c.baseURL, "/pcm/v1/schedule/ai/getJobLog")
urlPath := filepath.Join("/schedule/ai/getJobLog", "1777144940459986944", req.ClusterID, req.TaskID, "0")
urlPath = filepath.ToSlash(urlPath)

targetUrl, err := url.JoinPath(c.baseURL, urlPath)
if err != nil {
return nil, err
}

req.AdapterID = "1777144940459986944"
req.InstanceNum = "0"
//req.AdapterID = "1777144940459986944"
//req.InstanceNum = "0"

resp, err := http2.GetJSON(targetUrl, http2.RequestParam{
Body: req,
Query: req,
Header: map[string]string{
"Authorization": token,
},


+ 1
- 0
utils/http2/http.go View File

@@ -25,6 +25,7 @@ const (
ContentTypeMultiPart = "multipart/form-data"
ContentTypeOctetStream = "application/octet-stream"
ContentTypeEventStream = "text/event-stream"
ContentTypeTextPlain = "text/plain"
)

var defaultClient = http.DefaultClient


Loading…
Cancel
Save