删除多余的属性

Signed-off-by: zouap <zouap@pcl.ac.cn>

modules/aisafety/resty.go

@@ -16,14 +16,11 @@ import (
 var (
 	restyClient *resty.Client
 	HOST        string
-	TOKEN       string
+	KEY         string
 )
 
 const (
-	JobHasBeenStopped   = "S410"
-	errInvalidToken     = "S401"
-	Public              = "public"
-	Custom              = "custom"
+	APPID               = "1"
 	LogPageSize         = 500
 	LogPageTokenExpired = "5m"
 	pageSize            = 15
@@ -38,7 +35,7 @@ func getRestyClient() *resty.Client {
 }
 
 func checkSetting() {
-	if len(HOST) != 0 && len(TOKEN) != 0 && restyClient != nil {
+	if len(HOST) != 0 && len(KEY) != 0 && restyClient != nil {
 		return
 	}
 	_ = loginCloudbrain()
@@ -62,6 +59,7 @@ func createSign(params map[string]interface{}, signKey string) string {
 		}
 	}
 	re += "key=" + signKey
+	log.Info("sign key:" + re)
 	h := md5.New()
 	h.Write([]byte(re))
 	return strings.ToUpper(hex.EncodeToString(h.Sum(nil)))
@@ -75,8 +73,12 @@ func CreateSafetyTask(jobName string, body interface{}) (string, error) {
 	reqPara, _ := json.Marshal(body)
 	log.Warn("job req:", string(reqPara[:]))
 
+	params := make(map[string]interface{})
+	params["appId"] = APPID
+
 	res, err := client.R().
 		SetHeader("Content-Type", "application/json").
+		SetHeader("sign", createSign(params, KEY)).
 		//SetAuthToken(TOKEN).
 		SetBody(body).
 		SetResult(&jobResult).
@@ -99,9 +101,12 @@ func GetAlgorithmList() error {
 	checkSetting()
 	client := getRestyClient()
 
+	params := make(map[string]interface{})
+	params["appId"] = APPID
+
 	res, err := client.R().
 		SetHeader("Content-Type", "application/json").
-		SetAuthToken(TOKEN).
+		SetHeader("sign", createSign(params, KEY)).
 		Get(HOST + "/v1/external/eval-standard/algorithmList")
 
 	if err != nil {
@@ -118,9 +123,12 @@ func GetTaskStatus(jobID string) error {
 	checkSetting()
 	client := getRestyClient()
 
+	params := make(map[string]interface{})
+	params["appId"] = APPID
+
 	res, err := client.R().
 		SetHeader("Content-Type", "application/json").
-		SetAuthToken(TOKEN).
+		SetHeader("sign", createSign(params, KEY)).
 		Get(HOST + "/v1/external/eval-standard/query")
 
 	if err != nil {