84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package logic
|
|
|
|
import (
|
|
"fusenapi/utils/auth"
|
|
"fusenapi/utils/basic"
|
|
"fusenapi/utils/check"
|
|
"fusenapi/utils/format"
|
|
"time"
|
|
|
|
"context"
|
|
|
|
"fusenapi/server/upload/internal/svc"
|
|
"fusenapi/server/upload/internal/types"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type UploadFileBackendLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewUploadFileBackendLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadFileBackendLogic {
|
|
return &UploadFileBackendLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *UploadFileBackendLogic) UploadFileBackend(req *types.RequestUploadFileBackend, userinfo *auth.UserInfo) (resp *basic.Response) {
|
|
// 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data)
|
|
// userinfo 传入值时, 一定不为null
|
|
if userinfo.IsOnlooker() {
|
|
return resp.SetStatus(basic.CodeUnAuth)
|
|
}
|
|
|
|
var uid int64
|
|
var keytype format.TypeFormatS3KeyName
|
|
if userinfo.IsGuest() {
|
|
uid = userinfo.GuestId
|
|
keytype = format.TypeS3KeyGuest
|
|
} else {
|
|
uid = userinfo.UserId
|
|
keytype = format.TypeS3KeyUser
|
|
}
|
|
|
|
l.svcCtx.AwsSession.Config.Region = aws.String("us-west-1")
|
|
svc := s3.New(l.svcCtx.AwsSession)
|
|
|
|
if !check.CheckCategory(req.Category) {
|
|
return resp.SetStatus(basic.CodeS3CategoryErr)
|
|
}
|
|
|
|
now := time.Now()
|
|
s3req, _ := svc.PutObjectRequest(
|
|
&s3.PutObjectInput{
|
|
Bucket: aws.String("storage.fusenpack.com"),
|
|
Key: aws.String(format.FormatS3KeyName(
|
|
keytype,
|
|
uid,
|
|
now,
|
|
l.svcCtx.Config.Env,
|
|
req.Category,
|
|
req.File.Filename,
|
|
)),
|
|
},
|
|
)
|
|
|
|
s3req.SetBufferBody(req.File.Data)
|
|
err := s3req.Send()
|
|
if err != nil {
|
|
return resp.SetStatus(basic.CodeS3PutObjectRequestErr)
|
|
}
|
|
|
|
return resp.SetStatus(basic.CodeOK, map[string]interface{}{
|
|
"upload_url": s3req.HTTPRequest.URL.String(),
|
|
})
|
|
|
|
}
|