From fb5d0d131d024c484e487e11ebfa2537b759cfd4 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Wed, 29 Mar 2023 20:09:00 +0800 Subject: [PATCH] =?UTF-8?q?IP=E5=88=B6=E5=93=81=E5=88=97=E8=A1=A8=E4=B8=AD?= =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=96=87=E4=BB=B6=E5=B0=BA=E5=AF=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../rpc/services/service_ip_library_artifact.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/internal/rpc/services/service_ip_library_artifact.go b/internal/rpc/services/service_ip_library_artifact.go index 2cfa7f38..2ace69a8 100644 --- a/internal/rpc/services/service_ip_library_artifact.go +++ b/internal/rpc/services/service_ip_library_artifact.go @@ -73,6 +73,20 @@ func (this *IPLibraryArtifactService) FindAllIPLibraryArtifacts(ctx context.Cont var pbArtifacts = []*pb.IPLibraryArtifact{} for _, artifact := range artifacts { + var pbFile *pb.File + if artifact.FileId > 0 { + fileInfo, err := models.SharedFileDAO.FindEnabledFile(tx, int64(artifact.FileId)) + if err != nil { + return nil, err + } + if fileInfo != nil { + pbFile = &pb.File{ + Id: int64(fileInfo.Id), + Size: int64(fileInfo.Size), + } + } + } + pbArtifacts = append(pbArtifacts, &pb.IPLibraryArtifact{ Id: int64(artifact.Id), Name: artifact.Name, @@ -81,6 +95,7 @@ func (this *IPLibraryArtifactService) FindAllIPLibraryArtifacts(ctx context.Cont MetaJSON: artifact.Meta, IsPublic: artifact.IsPublic, Code: artifact.Code, + File: pbFile, }) } return &pb.FindAllIPLibraryArtifactsResponse{