From e888412c58a6d363397cd6e17ee4c3ceb56ef9b0 Mon Sep 17 00:00:00 2001
From: JeshuaRen <270813223@qq.com>
Date: Tue, 4 Mar 2025 11:16:30 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Etoken?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 sdks/pcmscheduler/jobset.go | 10 +++++++--
 sdks/pcmscheduler/models.go | 42 +++++++++++++++++++------------------
 sdks/uploader/models.go     |  1 +
 3 files changed, 31 insertions(+), 22 deletions(-)

diff --git a/sdks/pcmscheduler/jobset.go b/sdks/pcmscheduler/jobset.go
index 8bbdd4e..6f6a366 100644
--- a/sdks/pcmscheduler/jobset.go
+++ b/sdks/pcmscheduler/jobset.go
@@ -120,7 +120,7 @@ type ScheduleData struct {
 	ClusterIDs  []schsdk.ClusterID `json:"clusterIDs"`
 }
 
-func (c *Client) CreateJob(req CreateJobReq) (*CreateJobResp, error) {
+func (c *Client) CreateJob(req CreateJobReq, token string) (*CreateJobResp, error) {
 	targetUrl, err := url.JoinPath(c.baseURL, "/createTask")
 	if err != nil {
 		return nil, err
@@ -135,6 +135,9 @@ func (c *Client) CreateJob(req CreateJobReq) (*CreateJobResp, error) {
 
 	resp, err := http2.PostJSON(targetUrl, http2.RequestParam{
 		Body: req,
+		Header: map[string]string{
+			"Authorization": token,
+		},
 	})
 	if err != nil {
 		return nil, err
@@ -176,7 +179,7 @@ type DataScheduleResults struct {
 	Results  []DataScheduleResult `json:"results"`
 }
 
-func (c *Client) RunJob(req RunJobReq) error {
+func (c *Client) RunJob(req RunJobReq, token string) error {
 	targetUrl, err := url.JoinPath(c.baseURL, "runTask")
 	if err != nil {
 		return err
@@ -184,6 +187,9 @@ func (c *Client) RunJob(req RunJobReq) error {
 
 	resp, err := http2.PostJSON(targetUrl, http2.RequestParam{
 		Body: req,
+		Header: map[string]string{
+			"Authorization": token,
+		},
 	})
 	if err != nil {
 		return err
diff --git a/sdks/pcmscheduler/models.go b/sdks/pcmscheduler/models.go
index 32c55f7..4c4a490 100644
--- a/sdks/pcmscheduler/models.go
+++ b/sdks/pcmscheduler/models.go
@@ -376,40 +376,42 @@ func (d *DataBindingBase) Noop() {}
 type DatasetBinding struct {
 	serder.Metadata `union:"dataset"`
 	DataBindingBase
-	Type        string             `json:"type"`
-	Name        string             `json:"name"`
-	ClusterIDs  []schsdk.ClusterID `json:"clusterIDs"`
-	Description string             `json:"description"`
-	Category    string             `json:"category"`
-	PackageID   cdssdk.PackageID   `json:"packageID"`
+	Type           string             `json:"type"`
+	Name           string             `json:"name"`
+	ClusterIDs     []schsdk.ClusterID `json:"clusterIDs"`
+	Description    string             `json:"description"`
+	Category       string             `json:"category"`
+	PackageID      cdssdk.PackageID   `json:"packageID"`
+	RepositoryName string             `json:"repositoryName"`
 }
 
 type ModelBinding struct {
 	serder.Metadata `union:"model"`
 	DataBindingBase
-	Type        string             `json:"type"`
-	Name        string             `json:"name"`
-	ClusterIDs  []schsdk.ClusterID `json:"clusterIDs"`
-	Description string             `json:"description"`
-	Category    string             `json:"category"`
-	ModelType   string             `json:"modelType"`
-	Env         string             `json:"env"`
-	Version     string             `json:"version"`
-	PackageID   cdssdk.PackageID   `json:"packageID"`
+	Type           string             `json:"type"`
+	Name           string             `json:"name"`
+	ClusterIDs     []schsdk.ClusterID `json:"clusterIDs"`
+	Description    string             `json:"description"`
+	Category       string             `json:"category"`
+	ModelType      string             `json:"modelType"`
+	Env            string             `json:"env"`
+	Version        string             `json:"version"`
+	PackageID      cdssdk.PackageID   `json:"packageID"`
+	RepositoryName string             `json:"repositoryName"`
 }
 
 type CodeBinding struct {
 	serder.Metadata `union:"code"`
 	DataBindingBase
-	Type string `json:"type"`
-	Name string `json:"name"`
-	//ClusterIDs        []schsdk.ClusterID `json:"clusterIDs"`
+	Type              string           `json:"type"`
+	Name              string           `json:"name"`
 	ClusterID         schsdk.ClusterID `json:"clusterID"`
 	Description       string           `json:"description"`
 	ImageID           schsdk.ImageID   `json:"imageID"`
 	BootstrapObjectID cdssdk.ObjectID  `json:"bootstrapObjectID"`
-	//FilePath    string             `json:"filePath"`
-	PackageID cdssdk.PackageID `json:"packageID"`
+	PackageID         cdssdk.PackageID `json:"packageID"`
+	// 当集群为openi的时候，需要传入分支
+	Branch string `json:"branch"`
 }
 
 //type ImageBinding struct {
diff --git a/sdks/uploader/models.go b/sdks/uploader/models.go
index c4126bc..7a9879a 100644
--- a/sdks/uploader/models.go
+++ b/sdks/uploader/models.go
@@ -80,6 +80,7 @@ type BindingCluster struct {
 	BindingID DataID    `gorm:"column:binding_id" json:"bindingID"`
 	ClusterID ClusterID `gorm:"column:cluster_id" json:"clusterID"`
 	Status    string    `gorm:"column:status" json:"status"`
+	Param     string    `gorm:"column:param" json:"Param"`
 	JsonData  string    `gorm:"column:json_data" json:"jsonData"`
 }