fix: api descript
This commit is contained in:
@@ -2,10 +2,13 @@ package logic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"juwan-backend/app/objectstory/rpc/internal/svc"
|
||||
"juwan-backend/app/objectstory/rpc/pb"
|
||||
|
||||
"github.com/aws/aws-sdk-go-v2/service/s3"
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
)
|
||||
|
||||
@@ -25,7 +28,26 @@ func NewGetFileUrlLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetFil
|
||||
|
||||
// 获取文件访问链接(处理私有文件的鉴权)
|
||||
func (l *GetFileUrlLogic) GetFileUrl(in *pb.GetFileUrlReq) (*pb.GetFileUrlResp, error) {
|
||||
// todo: add your logic here and delete this line
|
||||
if in == nil || in.GetFileId() == "" {
|
||||
return nil, errors.New("file id is required")
|
||||
}
|
||||
|
||||
return &pb.GetFileUrlResp{}, nil
|
||||
_, err := l.svcCtx.S3.HeadObject(l.ctx, &s3.HeadObjectInput{
|
||||
Bucket: &l.svcCtx.Config.S3Conf.Bucket,
|
||||
Key: &in.FileId,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
presignClient := s3.NewPresignClient(l.svcCtx.S3)
|
||||
presigned, err := presignClient.PresignGetObject(l.ctx, &s3.GetObjectInput{
|
||||
Bucket: &l.svcCtx.Config.S3Conf.Bucket,
|
||||
Key: &in.FileId,
|
||||
}, s3.WithPresignExpires(15*time.Minute))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &pb.GetFileUrlResp{Url: presigned.URL}, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user