From 95349dc4575d7fc342314380e644d9ef58659462 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Thu, 4 Aug 2022 16:01:07 +0800 Subject: [PATCH] =?UTF-8?q?=E5=85=81=E8=AE=B8=E7=94=A8=E6=88=B7=E6=A0=87?= =?UTF-8?q?=E8=AE=B0=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6=E7=8A=B6=E6=80=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/rpc/services/service_file.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/internal/rpc/services/service_file.go b/internal/rpc/services/service_file.go index ce709915..6eff21c6 100644 --- a/internal/rpc/services/service_file.go +++ b/internal/rpc/services/service_file.go @@ -18,8 +18,6 @@ func (this *FileService) FindEnabledFile(ctx context.Context, req *pb.FindEnable return nil, err } - // TODO 检查用户权限 - var tx = this.NullTx() file, err := models.SharedFileDAO.FindEnabledFile(tx, req.FileId) if err != nil { @@ -67,13 +65,20 @@ func (this *FileService) CreateFile(ctx context.Context, req *pb.CreateFileReque // UpdateFileFinished 将文件置为已完成 func (this *FileService) UpdateFileFinished(ctx context.Context, req *pb.UpdateFileFinishedRequest) (*pb.RPCSuccess, error) { - _, err := this.ValidateAdmin(ctx) + _, userId, err := this.ValidateAdminAndUser(ctx) if err != nil { return nil, err } var tx = this.NullTx() + if userId > 0 { + err = models.SharedFileDAO.CheckUserFile(tx, userId, req.FileId) + if err != nil { + return nil, err + } + } + err = models.SharedFileDAO.UpdateFileIsFinished(tx, req.FileId) if err != nil { return nil, err