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"` JobSetID schsdk.JobSetID `gorm:"column:jobset_id" json:"jobSetID"`
LocalJobID string `gorm:"column:local_job_id" json:"localJobID"` LocalJobID string `gorm:"column:local_job_id" json:"localJobID"`
Param string `gorm:"column:param" json:"param"` Param string `gorm:"column:param" json:"param"`
Output string `gorm:"column:run_output" json:"runOutput"`
Token string `gorm:"column:token" json:"token"` Token string `gorm:"column:token" json:"token"`
CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"` 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/http2"
"gitlink.org.cn/cloudream/common/utils/serder" "gitlink.org.cn/cloudream/common/utils/serder"
"net/url" "net/url"
"path/filepath"
"strings" "strings"
) )


type QueryTasksReq struct { 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 { type QueryTasksResp struct {
@@ -45,13 +46,13 @@ type TaskModel struct {
} }


func (c *Client) QueryTasks(req QueryTasksReq, token string) (*QueryTasksResp, error) { 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 { if err != nil {
return nil, err return nil, err
} }


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


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


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


func (c *Client) QueryDetails(req QueryDetailsReq, token string) (*QueryDetailsResp, error) { 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 { if err != nil {
return nil, err return nil, err
} }


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


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


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


func (c *Client) QueryHPCTrainLog(req QueryHPCTrainReq, token string) (*QueryHPCTrainResp, error) { 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 { if err != nil {
return nil, err return nil, err
} }


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


type QueryAITrainLogReq struct { 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 { type QueryAITrainLogResp struct {
@@ -227,16 +228,19 @@ type QueryAITrainLogResp struct {
} }


func (c *Client) QueryAITrainLog(req QueryAITrainLogReq, token string) (*QueryAITrainLogResp, error) { 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 { if err != nil {
return nil, err return nil, err
} }


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


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


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

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


var defaultClient = http.DefaultClient var defaultClient = http.DefaultClient


Loading…
Cancel
Save