From e50484ad409248a756267d1c27850e8943cf9c54 Mon Sep 17 00:00:00 2001
From: qiwang <1364512070@qq.com>
Date: Thu, 18 May 2023 11:26:43 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E5=88=A0=E9=99=A4=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E9=9B=86=E6=8E=A5=E5=8F=A3=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Former-commit-id: 88d78d4dc3c127027fa4bf813f1228a73c30474b
---
 .../rpc/internal/logic/deletedatasetlogic.go  | 78 ++++++++++++++++---
 .../PCM-MODELARTS/rpc/pb/pcm-modelarts.proto  |  4 +-
 2 files changed, 70 insertions(+), 12 deletions(-)

diff --git a/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/logic/deletedatasetlogic.go b/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/logic/deletedatasetlogic.go
index 01068749..2647f9f4 100644
--- a/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/logic/deletedatasetlogic.go
+++ b/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/logic/deletedatasetlogic.go
@@ -1,10 +1,14 @@
 package logic
 
 import (
+	"APIGW-go-sdk/core"
 	"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/common"
 	"PCM/common/tool"
 	"context"
+	"fmt"
+	"io/ioutil"
 	"k8s.io/apimachinery/pkg/util/json"
+	"net/http"
 	"strings"
 
 	"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/svc"
@@ -31,17 +35,69 @@ func NewDeleteDataSetLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Del
 func (l *DeleteDataSetLogic) DeleteDataSet(in *modelarts.DeleteDataSetReq) (*modelarts.DeleteDataSetResq, error) {
 	// todo: add your logic here and delete this line
 	var resp modelarts.DeleteDataSetResq
-	url := "https://modelarts.cn-north-4.myhuaweicloud.com/v2/" + in.ProjectId + "/datasets/" + in.DatasetId
-	reqByte, err := json.Marshal(in)
-	if err != nil {
-		return nil, err
+	//根据智算类型判断走华为智算还是南京智算
+	modelArtsType := in.ModelArtsType
+	if modelArtsType == l.svcCtx.Config.HaweiModelArtsType {
+		modelArtsUrl := l.svcCtx.Config.ModelArtsUrl
+		url := modelArtsUrl + "v2/" + in.ProjectId + "/datasets/" + in.DatasetId
+		reqByte, err := json.Marshal(in)
+		if err != nil {
+			return nil, err
+		}
+		payload := strings.NewReader(string(reqByte))
+		token := common.GetToken()
+		statusCode, body, err := tool.HttpClientWithBodyAndCode(tool.DELETE, url, payload, token)
+		if err != nil {
+			return nil, err
+		}
+		json.Unmarshal(body, &resp)
+		if err != nil {
+			return nil, err
+		}
+		if statusCode == 202 {
+			json.Unmarshal(body, &resp)
+			resp.Code = 200
+			resp.Msg = "Success"
+		} else if statusCode != 202 {
+			json.Unmarshal(body, &resp)
+			resp.Code = 400
+			resp.Msg = "Failure"
+		}
+	} else if modelArtsType == l.svcCtx.Config.NanjingModelArtsType {
+		AK := l.svcCtx.Config.AK
+		SK := l.svcCtx.Config.SK
+		NanjingModelArtsUrl := l.svcCtx.Config.NanjingModelArtsUrl
+		XProjectId := l.svcCtx.Config.XProjectId
+		s := core.Signer{
+			Key:    AK,
+			Secret: SK,
+		}
+		r, err := http.NewRequest("DELETE", NanjingModelArtsUrl+"v2/"+in.ProjectId+"/datasets/"+in.DatasetId,
+			nil)
+		if err != nil {
+			fmt.Println(err)
+		}
+		r.Header.Add("content-type", "application/json;charset=UTF-8")
+		r.Header.Add("X-Project-Id", XProjectId)
+		r.Header.Add("x-stage", "RELEASE")
+		s.Sign(r)
+		client := http.DefaultClient
+		res, err := client.Do(r)
+		if err != nil {
+			fmt.Println(err)
+		}
+		defer res.Body.Close()
+		body, err := ioutil.ReadAll(res.Body)
+		statusCode := res.StatusCode
+		if statusCode == 202 {
+			json.Unmarshal(body, &resp)
+			resp.Code = 200
+			resp.Msg = "Success"
+		} else if statusCode != 202 {
+			json.Unmarshal(body, &resp)
+			resp.Code = 400
+			resp.Msg = "Failure"
+		}
 	}
-	payload := strings.NewReader(string(reqByte))
-	token := common.GetToken()
-	body, err := tool.HttpClient(tool.DELETE, url, payload, token)
-	if err != nil {
-		return nil, err
-	}
-	json.Unmarshal(body, &resp)
 	return &resp, nil
 }
diff --git a/adaptor/PCM-AI/PCM-MODELARTS/rpc/pb/pcm-modelarts.proto b/adaptor/PCM-AI/PCM-MODELARTS/rpc/pb/pcm-modelarts.proto
index 2f410a6e..521a8d02 100644
--- a/adaptor/PCM-AI/PCM-MODELARTS/rpc/pb/pcm-modelarts.proto
+++ b/adaptor/PCM-AI/PCM-MODELARTS/rpc/pb/pcm-modelarts.proto
@@ -1726,7 +1726,9 @@ message DeleteDataSetReq {
 }
 
 message DeleteDataSetResq{
-
+  int32 code =1;                  //@gotags: copier:"Code"
+  string msg =2;                  //@gotags: copier:"Msg"
+  string  error_msg =3;// @gotags: copier:"ErrorMsg"
 }
 /******************create dataset end*************************/