Merge branch 'develop' of https://gitee.com/fusenpack/fusenapi into develop

server/upload/internal/logic/uploadfilebackendlogic.go

@@ -0,0 +1,83 @@
+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(),
+	})
+
+}

server/upload/internal/logic/uploadfilefrontendlogic.go

@@ -0,0 +1,88 @@
+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 UploadFileFrontendLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewUploadFileFrontendLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadFileFrontendLogic {
+	return &UploadFileFrontendLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *UploadFileFrontendLogic) UploadFileFrontend(req *types.RequestUploadFileFrontend, 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 req.FileSize > 1024*1024*500 {
+		return resp.SetStatus(basic.CodeS3PutSizeLimitErr)
+	}
+
+	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.FileName,
+			)),
+			ContentLength: aws.Int64(req.FileSize),
+		},
+	)
+
+	uri, err := s3req.Presign(time.Minute * 5)
+	if err != nil {
+		return resp.SetStatus(basic.CodeS3PutObjectRequestErr)
+	}
+	// log.Println(uri)
+
+	return resp.SetStatus(basic.CodeOK, map[string]interface{}{
+		"upload_url": uri,
+	})
+}

server/upload/internal/logic/uploadupfilelogic.go

@@ -0,0 +1,34 @@
+package logic
+
+import (
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+
+	"context"
+
+	"fusenapi/server/upload/internal/svc"
+	"fusenapi/server/upload/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type UploadUpFileLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewUploadUpFileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadUpFileLogic {
+	return &UploadUpFileLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *UploadUpFileLogic) UploadUpFile(req *types.RequestUpFile, userinfo *auth.UserInfo) (resp *basic.Response) {
+	// 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data)
+	// userinfo 传入值时, 一定不为null
+
+	return resp.SetStatus(basic.CodeOK)
+}