From c49c82d97012bf01e0926e1469f7f08d6c5a4e42 Mon Sep 17 00:00:00 2001 From: JeshuaRen <270813223@qq.com> Date: Thu, 26 Jun 2025 09:19:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E4=BB=BB=E5=8A=A1=E7=9B=B8?= =?UTF-8?q?=E5=85=B3=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sdks/pcmscheduler/job.go | 1 + sdks/pcmscheduler/jobmgr.go | 42 ++++++++++++++++++++----------------- utils/http2/http.go | 1 + 3 files changed, 25 insertions(+), 19 deletions(-) diff --git a/sdks/pcmscheduler/job.go b/sdks/pcmscheduler/job.go index 898a3fa..02ca9eb 100644 --- a/sdks/pcmscheduler/job.go +++ b/sdks/pcmscheduler/job.go @@ -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"` } diff --git a/sdks/pcmscheduler/jobmgr.go b/sdks/pcmscheduler/jobmgr.go index 38d6d1e..1deb8d3 100644 --- a/sdks/pcmscheduler/jobmgr.go +++ b/sdks/pcmscheduler/jobmgr.go @@ -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, }, diff --git a/utils/http2/http.go b/utils/http2/http.go index f57b06f..b425c8d 100644 --- a/utils/http2/http.go +++ b/utils/http2/http.go @@ -25,6 +25,7 @@ const ( ContentTypeMultiPart = "multipart/form-data" ContentTypeOctetStream = "application/octet-stream" ContentTypeEventStream = "text/event-stream" + ContentTypeTextPlain = "text/plain" ) var defaultClient = http.DefaultClient