feat:图片缩放

server/resource/internal/handler/logoresizehandler.go

@@ -0,0 +1,35 @@
+package handler
+
+import (
+	"net/http"
+	"reflect"
+
+	"fusenapi/utils/basic"
+
+	"fusenapi/server/resource/internal/logic"
+	"fusenapi/server/resource/internal/svc"
+	"fusenapi/server/resource/internal/types"
+)
+
+func LogoResizeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+
+		var req types.LogoResizeReq
+		userinfo, err := basic.RequestParse(w, r, svcCtx, &req)
+		if err != nil {
+			return
+		}
+
+		// 创建一个业务逻辑层实例
+		l := logic.NewLogoResizeLogic(r.Context(), svcCtx)
+
+		rl := reflect.ValueOf(l)
+		basic.BeforeLogic(w, r, rl)
+
+		resp := l.LogoResize(&req, userinfo)
+
+		if !basic.AfterLogic(w, r, rl, resp) {
+			basic.NormalAfterLogic(w, r, resp)
+		}
+	}
+}

server/resource/internal/handler/routes.go

@@ -27,6 +27,11 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 				Path:    "/api/resource/info",
 				Handler: ResourceInfoHandler(serverCtx),
 			},
+			{
+				Method:  http.MethodPost,
+				Path:    "/api/resource/logo-resize",
+				Handler: LogoResizeHandler(serverCtx),
+			},
 		},
 	)
 }

server/resource/internal/logic/logoresizelogic.go

@@ -0,0 +1,132 @@
+package logic
+
+import (
+	"bytes"
+	"encoding/json"
+	"fusenapi/model/gmodel"
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+	"fusenapi/utils/file"
+	"fusenapi/utils/hash"
+	"image"
+	"image/jpeg"
+	"io"
+	"net/http"
+
+	"context"
+
+	"fusenapi/server/resource/internal/svc"
+	"fusenapi/server/resource/internal/types"
+
+	"github.com/disintegration/imaging"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type LogoResizeLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewLogoResizeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LogoResizeLogic {
+	return &LogoResizeLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+// 处理进入前逻辑w,r
+// func (l *LogoResizeLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) {
+// }
+
+func (l *LogoResizeLogic) LogoResize(req *types.LogoResizeReq, userinfo *auth.UserInfo) (resp *basic.Response) {
+	// 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data)
+	// userinfo 传入值时, 一定不为null
+
+	var userId int64
+	var guestId int64
+
+	// 检查用户是否是游客
+	if userinfo.IsGuest() {
+		// 如果是，使用游客ID和游客键名格式
+		guestId = userinfo.GuestId
+	} else {
+		// 否则，使用用户ID和用户键名格式
+		userId = userinfo.UserId
+	}
+
+	resourceModel := gmodel.NewFsResourceModel(l.svcCtx.MysqlConn)
+	resourceInfo, err := resourceModel.FindOneById(l.ctx, req.ResourceId)
+	if err != nil {
+		logx.Errorf("find resource err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+	response, err := http.Get(*resourceInfo.ResourceUrl)
+	if err != nil {
+		logx.Errorf("http get err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+	defer response.Body.Close()
+
+	imgByte, err := io.ReadAll(response.Body)
+	if err != nil {
+		logx.Errorf("io readall err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+
+	src, _, err := image.Decode(bytes.NewReader(imgByte))
+	if err != nil {
+		logx.Errorf("image decode err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+
+	// Resize the cropped image to width = 200px preserving the aspect ratio.
+	src = imaging.Resize(src, int(req.Width), int(req.Height), imaging.Lanczos)
+	if err != nil {
+		logx.Errorf("imaging esize err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+
+	emptyBuff := bytes.NewBuffer(nil) //开辟一个新的空buff
+	jpeg.Encode(emptyBuff, src, nil)  //img写入到buff
+
+	var hashKeyDataMap = make(map[string]interface{})
+
+	hashKeyDataB, _ := json.Marshal(req)
+	json.Unmarshal(hashKeyDataB, &hashKeyDataMap)
+	var resourceId string = hash.JsonHashKey(hashKeyDataMap)
+
+	// 上传文件
+	var upload = file.Upload{
+		Ctx:        l.ctx,
+		MysqlConn:  l.svcCtx.MysqlConn,
+		AwsSession: l.svcCtx.AwsSession,
+	}
+	uploadRes, err := upload.UploadFileByByte(&file.UploadBaseReq{
+		FileHash:     resourceId,
+		FileByte:     emptyBuff.Bytes(),
+		UploadBucket: 1,
+		ApiType:      2,
+		UserId:       userId,
+		GuestId:      guestId,
+		Source:       "logo-resize",
+	})
+	if err != nil {
+		logx.Errorf("upload UploadFileByByte err: %v", err)
+		return resp.SetStatus(basic.CodeFileNoFoundErr)
+	}
+
+	// 原图metadata 更新
+
+	// 返回成功的响应和上传URL
+	return resp.SetStatus(basic.CodeOK, map[string]interface{}{
+		"resource_id":  uploadRes.ResourceId,
+		"resource_url": uploadRes.ResourceUrl,
+	})
+}
+
+// 处理逻辑后 w,r 如:重定向, resp 必须重新处理
+// func (l *LogoResizeLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) {
+// // httpx.OkJsonCtx(r.Context(), w, resp)
+// }

server/resource/internal/types/types.go

@@ -5,6 +5,15 @@ import (
 	"fusenapi/utils/basic"
 )
 
+type LogoResizeReq struct {
+	ResourceId string `form:"resource_id"`
+	Width      int64  `form:"width"`
+	Height     int64  `form:"height"`
+}
+
+type LogoResizeRes struct {
+}
+
 type LogoRemovebgReq struct {
 	IsRemoveBg string `form:"is_remove_bg"`
 	LogoFile   string `form:"logo_file"`