From 4a82cfa159517a82a7ba778b20f68832d02b965d Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Fri, 15 Sep 2023 17:56:24 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AD=98=E7=AE=97=E8=81=94=E5=8A=A8=E7=B1=BB?=
 =?UTF-8?q?=E5=9E=8B=E8=BD=AC=E6=8D=A2=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Former-commit-id: 07fc2e447b89c8a45e779a24d5d7bdc3a77e9450
---
 api/internal/pkg/storeLink/storeLink.go | 59 +++++++++++++++++++++++--
 1 file changed, 56 insertions(+), 3 deletions(-)

diff --git a/api/internal/pkg/storeLink/storeLink.go b/api/internal/pkg/storeLink/storeLink.go
index 24906091..a39babb9 100644
--- a/api/internal/pkg/storeLink/storeLink.go
+++ b/api/internal/pkg/storeLink/storeLink.go
@@ -41,15 +41,27 @@ func NewStoreLink(ctx context.Context, svcCtx *svc.ServiceContext, partId int64)
 func ConvertType[T any](in *T) (interface{}, error) {
 
 	switch (interface{})(in).(type) {
+	case *octopus.UploadImageResp:
+		var resp types.UploadLinkImageResp
+		inresp := (interface{})(in).(*octopus.UploadImageResp)
+		resp.Success = inresp.Success
+		if !resp.Success {
+			resp.ErrorMsg = inresp.Error.Message
+			return resp, nil
+		}
+
+		return resp, nil
+
 	case *octopus.GetUserImageListResp:
 		var resp types.GetLinkImageListResp
 		inresp := (interface{})(in).(*octopus.GetUserImageListResp)
 		resp.Success = inresp.Success
-		if inresp.Error == nil {
-			resp.ErrorMsg = ""
-		} else {
+		if !resp.Success {
 			resp.ErrorMsg = inresp.Error.Message
+			resp.Images = nil
+			return resp, nil
 		}
+
 		for _, v := range inresp.Payload.Images {
 			var image types.ImageSl
 			image.ImageId = v.Image.Id
@@ -58,18 +70,59 @@ func ConvertType[T any](in *T) (interface{}, error) {
 			resp.Images = append(resp.Images, image)
 		}
 		return resp, nil
+
 	case *octopus.DeleteImageResp:
 		var resp types.DeleteLinkImageResp
+		inresp := (interface{})(in).(*octopus.DeleteImageResp)
+		resp.Success = inresp.Success
+		if !resp.Success {
+			resp.ErrorMsg = inresp.Error.Message
+			return resp, nil
+		}
+
 		return resp, nil
+
 	case *octopus.CreateTrainJobResp:
 		var resp types.SubmitLinkTaskResp
+		inresp := (interface{})(in).(*octopus.CreateTrainJobResp)
+		resp.Success = inresp.Success
+		if !resp.Success {
+			resp.ErrorMsg = inresp.Error.Message
+			return resp, nil
+		}
+
+		resp.TaskId = inresp.Payload.JobId
+
 		return resp, nil
+
 	case *octopus.GetTrainJobResp:
 		var resp types.GetLinkTaskResp
+		inresp := (interface{})(in).(*octopus.GetTrainJobResp)
+		resp.Success = inresp.Success
+		if !resp.Success {
+			resp.ErrorMsg = inresp.Error.Message
+			return resp, nil
+		}
+
+		resp.Task.TaskId = inresp.Payload.TrainJob.Id
+		resp.Task.TaskName = inresp.Payload.TrainJob.Name
+		resp.Task.StartedAt = inresp.Payload.TrainJob.StartedAt
+		resp.Task.CompletedAt = inresp.Payload.TrainJob.CompletedAt
+		resp.Task.TaskStatus = inresp.Payload.TrainJob.Status
+
 		return resp, nil
+
 	case *octopus.DeleteTrainJobResp:
 		var resp types.DeleteLinkTaskResp
+		inresp := (interface{})(in).(*octopus.DeleteTrainJobResp)
+		resp.Success = inresp.Success
+		if !resp.Success {
+			resp.ErrorMsg = inresp.Error.Message
+			return resp, nil
+		}
+
 		return resp, nil
+
 	default:
 		return nil, errors.New("type convert fail")
 	}